Upload app.py

app.py

@@ -49,12 +49,13 @@ def convert_to_images(file_path):
 # -----------------------------------------------------------------------------
 def scan_slides(client, image_paths):
     inventory = []
+    total = len(image_paths)
     
-    # Process in batches to speed up? No, 1-by-1 is safer for context.
-    # Gemini 2.0 is fast enough.
-    
+    # Process 1-by-1 to preserve context and allow progress updates
     for i, img_path in enumerate(image_paths):
         slide_num = i + 1
+        yield f"Reading Slide {slide_num}/{total}...", None
+        
         print(f"Scanning Slide {slide_num}...")
         
         with open(img_path, "rb") as f:
@@ -88,14 +89,18 @@ def scan_slides(client, image_paths):
         except Exception as e:
             print(f"Error scanning slide {slide_num}: {e}")
             
-    return inventory
+    yield "Scan Complete", inventory
 
 # -----------------------------------------------------------------------------
 # LOGIC: PASS 2 (COACH CRITIQUE)
 # -----------------------------------------------------------------------------
 def generate_critique(client, inventory):
     # Filter Appendix
-    active = [s for s in inventory if "appendix" not in s.get("title", "").lower()]
+    def get_title(slide):
+        t = slide.get("title")
+        return t if t else ""
+        
+    active = [s for s in inventory if "appendix" not in get_title(s).lower()]
     
     # Script
     script = []
@@ -171,29 +176,44 @@ def generate_critique(client, inventory):
 # -----------------------------------------------------------------------------
 def process_presentation(file_obj, password):
     print("--- NEW JOB STARTED ---")
+    if file_obj is None:
+        return "❌ Error: No file uploaded", "", "", None, None, ""
+
     if password != ACCESS_PASSWORD:
-        return "❌ Incorrect Password", "", ""
+        return "❌ Incorrect Password", "", "", None, None, ""
         
     if not API_KEY:
-        return "❌ Server Error: API Key missing", "", ""
+        return "❌ Server Error: API Key missing", "", "", None, None, ""
         
     client = genai.Client(api_key=API_KEY)
     
     try:
         # 1. Convert
         print("Step 1: Converting PDF...")
-        yield "Converting PDF...", "", "", None, None
+        yield "Converting PDF...", "", "", None, None, "Converting PDF..."
         images = convert_to_images(file_obj.name)
         print(f"   Converted {len(images)} slides.")
         
         # 2. Scan
-        yield f"Scanning {len(images)} slides... (This takes ~30s)", "", "", None, None
+        yield f"Scanning {len(images)} slides... (This takes ~30s)", "", "", None, None, "Starting Scan..."
         print("Step 2: Scanning Slides (Pass 1)...")
-        inventory = scan_slides(client, images)
+        
+        # Iterate through the generator from scan_slides
+        scanner = scan_slides(client, images)
+        inventory = []
+        
+        for msg, result in scanner:
+            if result is None:
+                # Still scanning, update status
+                yield msg, "", "", None, None, msg
+            else:
+                # Scan complete, result is the inventory
+                inventory = result
+                
         print("   Scan Complete.")
         
         # 3. Coach
-        yield "Dr. Jones is thinking (Pass 2)...", "", "", None, None
+        yield "Dr. Jones is thinking (Pass 2)...", "", "", None, None, "Analyzing Story Arc..."
         print("Step 3: Calling Gemini Coach (Pass 2)...")
         critique = generate_critique(client, inventory)
         print("   Critique Received.")
@@ -220,12 +240,12 @@ def process_presentation(file_obj, password):
             f.write("## Story Roadmap\n")
             f.write(table_md)
             
-        # Return: Status, Summary, Table, Preview Image, Report File
-        yield "✅ Done!", summary, table_md, images[0], report_filename
+        # Return: Status, Summary, Table, Preview Image, Report File, Small Status
+        yield "✅ Done!", summary, table_md, images[0], report_filename, ""
         
     except Exception as e:
         print(f"CRITICAL ERROR: {e}")
-        yield f"❌ Error: {str(e)}", "", "", None, None
+        yield f"❌ Error: {str(e)}", "", "", None, None, "Error"
 
 with gr.Blocks(title="Dr. Jones AI Coach") as demo:
     gr.Markdown("# 🎓 Dr. Jones: AI Presentation Coach")
@@ -245,7 +265,10 @@ with gr.Blocks(title="Dr. Jones AI Coach") as demo:
     with gr.Row():
         with gr.Column(scale=1):
             preview_img = gr.Image(label="Presentation Title Slide", interactive=False)
-            download_btn = gr.File(label="Download Full Report")
+            with gr.Row():
+                download_btn = gr.File(label="Download Full Report", scale=2)
+                with gr.Column(scale=1):
+                    progress_status = gr.Markdown(value="") # Temporary status next to download
         
         with gr.Column(scale=2):
             gr.Markdown("### 👨‍🏫 Coach Summary")
@@ -258,7 +281,7 @@ with gr.Blocks(title="Dr. Jones AI Coach") as demo:
     btn.click(
         fn=process_presentation,
         inputs=[file_input, pass_input],
-        outputs=[status, summary_display, roadmap_display, preview_img, download_btn]
+        outputs=[status, summary_display, roadmap_display, preview_img, download_btn, progress_status]
     )
 
 if __name__ == "__main__":