Update app.py

app.py

@@ -130,7 +130,7 @@ if img_file:
     if not realtime_update:
         st.write("Double click to save crop")
 
-    col1, col2 = st.columns(2)
+    col1, col2, col3 = st.columns(3)
     with col1:
         st.subheader("Input: Upload and Crop Your Image")
     # Get a cropped image from the frontend
@@ -148,38 +148,27 @@ if img_file:
         # _ = cropped_img.thumbnail((150, 150))
         st.image(cropped_img)
         button = st.button("Run OCR")
+
+    with col3:
         if button:
             with st.spinner('Running OCR...'):
                 if Lng == "Arabic":
                     ocr_text = predict_arabic(cropped_img)
-                    st.subheader(f"OCR Results for {Lng}")
-                    st.write(ocr_text)
-                    text_file = BytesIO(ocr_text.encode())
-                    st.download_button('Download Text', text_file, file_name='ocr_text.txt')
                 elif Lng == "English":
                     ocr_text = predict_nougat(cropped_img)
-                    st.subheader(f"OCR Results for {Lng}")
-                    st.write(ocr_text)
-                    text_file = BytesIO(ocr_text.encode())
-                    st.download_button('Download Text', text_file, file_name='ocr_text.txt')
                 elif Lng == "French":
                     ocr_text = predict_tesseract(cropped_img)
-                    st.subheader(f"OCR Results for {Lng}")
-                    st.write(ocr_text)
-                    text_file = BytesIO(ocr_text.encode())
-                    st.download_button('Download Text', text_file, file_name='ocr_text.txt')
                 elif Lng == "Korean":
                     ocr_text = predict_english(cropped_img)
-                    st.subheader(f"OCR Results for {Lng}")
-                    st.write(ocr_text)
-                    text_file = BytesIO(ocr_text.encode())
-                    st.download_button('Download Text', text_file, file_name='ocr_text.txt')
                 elif Lng == "Chinese":
                     ocr_text = predict_english(cropped_img)
-                    st.subheader(f"OCR Results for {Lng}")
-                    st.write(ocr_text)
-                    text_file = BytesIO(ocr_text.encode())
-                    st.download_button('Download Text', text_file, file_name='ocr_text.txt')
+
+            st.subheader(f"OCR Results for {Lng}")
+            st.write(ocr_text)
+            text_file = BytesIO(ocr_text.encode())
+            st.download_button('Download Text', text_file, file_name='ocr_text.txt')
+
+