Update app.py

app.py

@@ -132,7 +132,7 @@ if img_file:
     if not realtime_update:
         st.write("Double click to save crop")
 
-    col1, col2, col3 = st.columns(3)
+    col1, col2 = st.columns(2)
     with col1:
         st.subheader("Input: Upload and Crop Your Image")
     # Get a cropped image from the frontend
@@ -151,24 +151,23 @@ if img_file:
         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)
-                elif Lng == "English":
-                    ocr_text = predict_nougat(cropped_img)
-                elif Lng == "French":
-                    ocr_text = predict_tesseract(cropped_img)
-                elif Lng == "Korean":
-                    ocr_text = predict_english(cropped_img)
-                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')
+    if button:
+        with st.spinner('Running OCR...'):
+            if Lng == "Arabic":
+                ocr_text = predict_arabic(cropped_img)
+            elif Lng == "English":
+                ocr_text = predict_nougat(cropped_img)
+            elif Lng == "French":
+                ocr_text = predict_tesseract(cropped_img)
+            elif Lng == "Korean":
+                ocr_text = predict_english(cropped_img)
+            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')
 
     openai.api_key = ""
     
@@ -183,7 +182,7 @@ if img_file:
             st.markdown(message["content"])
     
     if prompt := st.chat_input("How can I help?"):
-        st.session_state.messages.append({"role": "user", "content": ocr_text+ prompt})
+        st.session_state.messages.append({"role": "user", "content": ocr_text + prompt})
         with st.chat_message("user"):
             st.markdown(prompt)