Update app.py

app.py

@@ -78,42 +78,38 @@ def load_and_extract_text(file_path):
     except Exception as e:
         return f"Error processing file: {e}"
 
-# File uploader
-uploaded_file = st.file_uploader(
-    "Upload a document or image",
-    type=["docx", "xlsx", "pdf", "csv", "txt", "pptx", "tex", "html", "htm", "jpg", "jpeg", "png"]
+# Multi-file uploader
+uploaded_files = st.file_uploader(
+    "Upload one or more documents or images",
+    type=["docx", "xlsx", "pdf", "csv", "txt", "pptx", "tex", "html", "htm", "jpg", "jpeg", "png"],
+    accept_multiple_files=True
 )
 
 # Question input
 question = st.text_input("💬 Enter your question:")
 
 # Processing
-if uploaded_file and question:
-    file_ext = uploaded_file.name.split('.')[-1].lower()
-
-    if file_ext in ['jpg', 'jpeg', 'png']:
-        # Image Q&A
-        image_bytes = uploaded_file.read()
-        encoded_image = base64.b64encode(image_bytes).decode("utf-8")
-        mime_type = uploaded_file.type
-
-        st.image(image_bytes, caption="Uploaded Image", use_container_width=True)
-
-        with st.spinner("Processing image..."):
-            try:
-                content = [
-                    {"inline_data": {"mime_type": mime_type, "data": encoded_image}},
-                    {"text": question}
-                ]
-                response = model.generate_content(contents=content)
-                st.success("💡 Answer:")
-                st.write(response.text)
-
-            except Exception as e:
-                st.error(f"Error processing image: {e}")
-
-    else:
-        # Document Q&A
+if uploaded_files and question:
+    combined_context = ""
+    image_contents = []
+
+    for uploaded_file in uploaded_files:
+        file_ext = uploaded_file.name.split('.')[-1].lower()
+
+        if file_ext in ['jpg', 'jpeg', 'png']:
+            # Store image data for later processing
+            image_bytes = uploaded_file.read()
+            encoded_image = base64.b64encode(image_bytes).decode("utf-8")
+            image_contents.append({
+                "inline_data": {
+                    "mime_type": uploaded_file.type,
+                    "data": encoded_image
+                }
+            })
+            st.image(image_bytes, caption=uploaded_file.name, use_container_width=True)
+            continue
+
+        # Process document files
         temp_file_path = "temp." + file_ext
         with open(temp_file_path, "wb") as temp_file:
             temp_file.write(uploaded_file.read())
@@ -122,13 +118,31 @@ if uploaded_file and question:
         os.remove(temp_file_path)
 
         if "Error" in extracted_text or "Unsupported" in extracted_text:
-            st.error(extracted_text)
+            st.error(f"{uploaded_file.name}: {extracted_text}")
         else:
-            with st.spinner("Processing document..."):
-                try:
-                    prompt = f"Context: {extracted_text}\n\nQuestion: {question}\n\nAnswer:"
-                    response = model.generate_content(prompt)
-                    st.success("💡 Answer:")
-                    st.write(response.text)
-                except Exception as e:
-                    st.error(f"Error generating answer: {e}")
+            combined_context += f"\n\n---\nDocument: {uploaded_file.name}\n\n{extracted_text}"
+
+    # Process if there's valid content
+    if combined_context.strip() or image_contents:
+        with st.spinner("Generating answer..."):
+            try:
+                content_blocks = []
+
+                # Add text context
+                if combined_context.strip():
+                    content_blocks.append({"text": f"Context:\n{combined_context}"})
+
+                # Add image blocks
+                content_blocks.extend(image_contents)
+
+                # Add user question
+                content_blocks.append({"text": f"Question: {question}"})
+
+                response = model.generate_content(contents=content_blocks)
+                st.success("💡 Answer:")
+                st.write(response.text)
+
+            except Exception as e:
+                st.error(f"Error generating answer: {e}")
+    else:
+        st.warning("No valid documents or images processed.")