Update app.py

app.py

@@ -1,18 +1,25 @@
 import gradio as gr
 from google import genai
 import os
+import time
 
-def ground_truth_engine(past_img, present_img):
+# We add 'progress=gr.Progress()' as an argument to enable the status bar
+def ground_truth_engine(past_img, present_img, progress=gr.Progress()):
+    # 1. Immediate Visual Feedback
+    progress(0, desc="Initializing Spatial Engine...")
+    
     api_key = os.environ.get("GOOGLE_API_KEY")
     if not api_key:
         return "ERROR: GOOGLE_API_KEY not found in Space Secrets."
 
-    # Initialize the new SDK client
     client = genai.Client(api_key=api_key)
     
     if past_img is None or present_img is None:
         return "Please upload both images."
 
+    # 2. Update status before the heavy lifting
+    progress(0.3, desc="Uploading Images to Gemini...")
+    
     prompt = """
     Perform a high-fidelity structural audit comparing these two images.
     Focus on changes in roofing, landscaping, and exterior condition.
@@ -20,18 +27,22 @@ def ground_truth_engine(past_img, present_img):
     """
 
     try:
+        # 3. Setting expectation for the 'Thinking' phase
+        progress(0.6, desc="Reasoning Spatially (This takes ~20-30s)...")
+        
         response = client.models.generate_content(
             model="gemini-robotics-er-1.5-preview",
             contents=[prompt, past_img, present_img]
         )
+        
+        progress(1.0, desc="Audit Complete!")
         return response.text
     except Exception as e:
         return f"Analysis Failed: {str(e)}"
 
-# Define the UI using Gradio 4 syntax
+# --- UI Setup ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🏠 GroundTruth AI (Stable Build)")
-    gr.Markdown("Temporal Property Sentinel powered by Gemini Robotics-ER 1.5")
     
     with gr.Row():
         with gr.Column():
@@ -42,8 +53,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         with gr.Column():
             output = gr.Markdown(label="Audit Report")
 
-    submit.click(fn=ground_truth_engine, inputs=[p_img, c_img], outputs=output)
+    # Important: show_progress="full" ensures the spinner stays visible
+    submit.click(
+        fn=ground_truth_engine, 
+        inputs=[p_img, c_img], 
+        outputs=output,
+        show_progress="full"
+    )
 
 if __name__ == "__main__":
-    # The server_name binding is required for Docker containers on HF
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    # Ensure queue is enabled for progress bar to update
+    demo.queue().launch(server_name="0.0.0.0", server_port=7860)