Update app.py

app.py

@@ -170,30 +170,30 @@ Models = {
 st.sidebar.markdown(f"### Selected Model: {Models[Lng]}")
 
 if img_file:
-    if not img_file.name.endswith(".pdf"):
-        cropped_img = Image.open(img_file)
+    if not img_file.type == ".pdf":
+        img = Image.open(img_file)
         if not realtime_update:
             st.write("Double click to save crop")
     
-        # col1, col2 = st.columns(2)
-        # with col1:
-        #     st.subheader("Input: Upload and Crop Your Image")
-        # # Get a cropped image from the frontend
-        #     cropped_img = st_cropper(
-        #         img,
-        #         realtime_update=realtime_update,
-        #         box_color="#FF0000",
-        #         aspect_ratio=aspect_ratio,
-        #         should_resize_image=True,
-        #     )
+        col1, col2 = st.columns(2)
+        with col1:
+            st.subheader("Input: Upload and Crop Your Image")
+        # Get a cropped image from the frontend
+            cropped_img = st_cropper(
+                img,
+                realtime_update=realtime_update,
+                box_color="#FF0000",
+                aspect_ratio=aspect_ratio,
+                should_resize_image=True,
+            )
     
-        # with col2:
-        # # Manipulate cropped image at will
-        #     st.subheader("Output: Preview and Analyze")
-        #     # _ = cropped_img.thumbnail((150, 150))
-        #     st.image(cropped_img)
+        with col2:
+        # Manipulate cropped image at will
+            st.subheader("Output: Preview and Analyze")
+            # _ = cropped_img.thumbnail((150, 150))
+            st.image(cropped_img)
     
-        button = st.sidebar.button("Run OCR")
+        button = st.button("Run OCR")
     
         if button:
             with st.spinner('Running OCR...'):
@@ -214,7 +214,7 @@ if img_file:
             st.download_button('Download Text', text_file,
                                file_name='ocr_text.txt')
     
-    elif img_file.name.endswith(".pdf"):
+    elif img_file.type == ".pdf":
         button = st.sidebar.button("Run OCR")
     
         if button: