Update app.py

app.py

@@ -1,13 +1,13 @@
 import streamlit as st
-
-# ✅ Fix: set_page_config() must be the first command
-st.set_page_config(page_title="Multimodal AI Assistant", layout="wide")
-
 from PIL import Image
 import torch
 import easyocr
+import io
 from transformers import CLIPModel, CLIPImageProcessor
 
+# ✅ Fix: set_page_config() must be the first command
+st.set_page_config(page_title="Multimodal AI Assistant", layout="wide")
+
 # ---- Load CLIP Model (Vision Only) ---- #
 @st.cache_resource
 def load_clip_model():
@@ -35,14 +35,20 @@ st.write("Upload an image and ask a question about it!")
 uploaded_file = st.file_uploader("📤 Upload an image", type=["jpg", "png", "jpeg"])
 
 if uploaded_file is not None:
-    # Display Image
+    # Convert file to image format
     image = Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image", use_column_width=True)
+
+    # ✅ Fix: use `use_container_width` instead of `use_column_width`
+    st.image(image, caption="Uploaded Image", use_container_width=True)
+
+    # Convert uploaded file to bytes for EasyOCR
+    file_bytes = uploaded_file.read()  # ✅ Fix: Convert file to bytes
+    image_bytes = io.BytesIO(file_bytes)
 
     # Extract Text using OCR
     with st.spinner("🔍 Extracting text from image..."):
-        extracted_text = reader.readtext(uploaded_file, detail=0)
-    
+        extracted_text = reader.readtext(image_bytes, detail=0)
+
     st.write("### 📝 Extracted Text:")
     if extracted_text:
         st.success(extracted_text)