Update ZIP input feature

app.py

@@ -115,38 +115,63 @@ def gemini_analysis(images):
     return response.text
 
 
-# full flow process
+def extract_zip_and_collect_files(zip_file_path):
+    """
+    Extract zip file to a temp directory and return list of pdf/image file paths inside.
+    """
+    temp_dir = tempfile.mkdtemp()
+    with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
+        zip_ref.extractall(temp_dir)
+
+    # Collect all pdf and image files in extracted folder recursively
+    collected_files = []
+    for root, _, files in os.walk(temp_dir):
+        for f in files:
+            if f.lower().endswith(('.pdf', '.jpg', '.jpeg', '.png')):
+                collected_files.append(os.path.join(root, f))
+    return collected_files
+
+
 def main_process(files):
     all_images = []
-    
-    for file in files:
-        file_path = file.name if hasattr(file, 'name') else file  # handle Gradio file objects
 
-        if file_path.lower().endswith('.pdf'):
-            images = process_pdfs([file_path])  # Pass as list
+    for file in files:
+        file_path = file.name if hasattr(file, 'name') else file
+
+        if file_path.lower().endswith('.zip'):
+            # Extract zip and process contained files
+            extracted_files = extract_zip_and_collect_files(file_path)
+            for extracted_file in extracted_files:
+                if extracted_file.lower().endswith('.pdf'):
+                    images = process_pdfs([extracted_file])
+                    all_images.extend(images)
+                elif extracted_file.lower().endswith(('.jpg', '.jpeg', '.png')):
+                    image = Image.open(extracted_file)
+                    all_images.append(image)
+        elif file_path.lower().endswith('.pdf'):
+            images = process_pdfs([file_path])
             all_images.extend(images)
         elif file_path.lower().endswith(('.jpg', '.jpeg', '.png')):
             image = Image.open(file_path)
             all_images.append(image)
         else:
-            raise ValueError(f"File {file_path} is not a valid image or PDF.")
-    
-    # Perform analysis using Gemini AI
+            raise ValueError(f"File {file_path} is not a valid image, PDF, or ZIP.")
+
     summary = gemini_analysis(all_images)
     return summary
 
 
-# Gradio UI
+# Gradio UI update: add ".zip" to accepted file types
 with gr.Blocks() as demo:
     gr.Markdown("## 🧠 Summarize Documents with Gemini AI (Image Analysis Only)")
     with gr.Row():
         file_input = gr.File(
-            label="Upload PDFs or Images (Multiple Supported)",
-            file_types=[".pdf", ".jpg", ".jpeg", ".png"],
+            label="Upload PDFs, Images or ZIP files (Multiple Supported)",
+            file_types=[".pdf", ".jpg", ".jpeg", ".png", ".zip"],
             file_count="multiple"
         )
     output = gr.Textbox(label="📝 Summary Result", lines=20)
     run_btn = gr.Button("🔍 Run Analysis")
     run_btn.click(fn=main_process, inputs=[file_input], outputs=output)
 
-demo.launch()
+demo.launch()