Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from transformers import pipeline
+from PIL import Image
 import requests
 
 # Load OCR model for extracting text from images
@@ -41,24 +42,29 @@ def main():
     uploaded_file = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])
 
     if uploaded_file is not None:
-        st.image(uploaded_file, caption="Uploaded Image", use_column_width=True)
-        st.write("Processing...")
+        # Convert uploaded file to a PIL image
+        try:
+            image = Image.open(uploaded_file)
+            st.image(image, caption="Uploaded Image", use_column_width=True)
+            st.write("Processing...")
 
-        # Extract text from image
-        ocr_model = load_ocr_model()
-        extracted_text = ocr_model(uploaded_file)["generated_text"]
+            # Extract text from image
+            ocr_model = load_ocr_model()
+            extracted_text = ocr_model(image)[0]["generated_text"]
 
-        if extracted_text:
-            st.write("### Extracted Text:")
-            st.write(extracted_text)
+            if extracted_text:
+                st.write("### Extracted Text:")
+                st.write(extracted_text)
 
-            # Send text to ChatGPT
-            st.write("### ChatGPT Explanation:")
-            explanation = chat_with_gpt(extracted_text)
-            if explanation:
-                st.write(explanation)
-        else:
-            st.error("Could not extract text. Please try again with another image.")
+                # Send text to ChatGPT
+                st.write("### ChatGPT Explanation:")
+                explanation = chat_with_gpt(extracted_text)
+                if explanation:
+                    st.write(explanation)
+            else:
+                st.error("Could not extract text. Please try again with another image.")
+        except Exception as e:
+            st.error(f"Error processing the image: {e}")
 
 if __name__ == "__main__":
     main()