Update app.py

app.py

@@ -1,61 +1,64 @@
 import gradio as gr
 import google.generativeai as genai
 import os
-from PIL import Image
+import time
 
-# --- Configuration ---
-# You'll set this 'GOOGLE_API_KEY' secret in the Settings tab later
-api_key = os.environ.get("GOOGLE_API_KEY")
-if api_key:
-    genai.configure(api_key=api_key)
-
-def analyze_property(past_img, present_img):
+def analyze_property(past_img, present_img, progress=gr.Progress()):
+    # 1. Immediate Visual Feedback
+    gr.Info("GroundTruth is initializing the Robotics-ER engine...")
+    progress(0, desc="Authenticating API...")
+    
+    api_key = os.environ.get("GOOGLE_API_KEY")
     if not api_key:
-        return "## Error\nAPI Key not configured. Please add GOOGLE_API_KEY to your Space Secrets."
+        raise gr.Error("API Key missing! Add it to 'Secrets' in Settings.")
+
+    genai.configure(api_key=api_key)
     
     if past_img is None or present_img is None:
-        return "Please upload both images to begin the analysis."
+        gr.Warning("Both images are required for temporal analysis.")
+        return "Please upload both images."
 
-    # Initialize the Robotics-ER 1.5 model for spatial/temporal reasoning
+    # 2. Update Progress Bar
+    progress(0.2, desc="Uploading images to Google Robotics-ER 1.5...")
     model = genai.GenerativeModel("gemini-robotics-er-1.5-preview")
     
     prompt = """
-    You are an expert structural analyst. Compare these two images of the same property.
-    Focus on:
-    1. Roofing & Gutters (wear, replacement, sagging)
-    2. Landscaping (maturity, maintenance, overgrowth)
-    3. Exterior Envelope (paint condition, siding, updated fixtures)
-    
-    Determine the Maintenance Trajectory: IMPROVING, DECLINING, or STABLE.
+    Perform a high-fidelity structural audit comparing these two images.
+    Identify changes in roofing, landscaping, and exterior condition.
+    Provide a 'Maintenance Trajectory' (IMPROVING/STABLE/DECLINING).
     """
-    
-    # Send both images to the VLM for temporal reasoning
-    response = model.generate_content([prompt, past_img, present_img])
-    return response.text
 
-# --- Custom UI Layout ---
+    try:
+        progress(0.5, desc="AI is reasoning spatially... (10-20 seconds typical)")
+        # The model call is synchronous and waits here
+        response = model.generate_content([prompt, past_img, present_img])
+        
+        progress(1.0, desc="Analysis complete!")
+        return response.text
+        
+    except Exception as e:
+        raise gr.Error(f"Analysis failed: {str(e)}")
+
+# --- UI Setup ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🏠 GroundTruth: Temporal Property Sentinel")
-    gr.Markdown("### Powered by Google Gemini Robotics-ER 1.5")
-    gr.Markdown("Identify 'Pride of Ownership' and structural changes across time using advanced spatial AI.")
     
     with gr.Row():
         with gr.Column():
-            past_input = gr.Image(label="Past Condition (e.g., 2020)", type="pil")
-            present_input = gr.Image(label="Present Condition (e.g., 2026)", type="pil")
+            past_input = gr.Image(label="Past Condition", type="pil")
+            present_input = gr.Image(label="Present Condition", type="pil")
+            # show_progress="full" ensures the spinner stays visible
             submit_btn = gr.Button("Analyze Structural Trajectory", variant="primary")
         
         with gr.Column():
-            output_text = gr.Markdown(label="Analysis Results")
+            output_text = gr.Markdown(label="Structural Audit Report")
 
+    # Important: .queue() is required for gr.Progress to work
     submit_btn.click(
         fn=analyze_property,
         inputs=[past_input, present_input],
-        outputs=output_text
+        outputs=output_text,
+        show_progress="full"
     )
 
-    gr.Markdown("---")
-    gr.Markdown("*Licensed under Apache 2.0. Built for Real Estate Professionals.*")
-
-if __name__ == "__main__":
-    demo.launch()
+demo.queue().launch()