Update app.py

app.py

@@ -1,38 +1,42 @@
 import gradio as gr
 from google import genai
 import os
-import time
 
-# 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
+def ground_truth_engine(past_img, present_img, audit_level, progress=gr.Progress()):
     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."
+        return "ERROR: Set GOOGLE_API_KEY in Space Secrets."
 
     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...")
+    # Map Slider to API 'thinking_budget'
+    # Quick = Low budget, Forensic = High budget
+    budget_map = {"Quick Scan": 100, "Standard Audit": 400, "Deep Forensic": 1000}
+    selected_budget = budget_map.get(audit_level, 400)
     
-    prompt = """
-    Perform a high-fidelity structural audit comparing these two images.
-    Focus on changes in roofing, landscaping, and exterior condition.
-    Conclude with a 'Maintenance Trajectory': IMPROVING, STABLE, or DECLINING.
+    progress(0.2, desc=f"Level: {audit_level} (Budget: {selected_budget})")
+
+    prompt = f"""
+    Perform a {audit_level} structural audit. 
+    Compare roofing, landscaping, and exterior condition.
+    Provide precise physical evidence for changes.
+    Conclude with 'Maintenance Trajectory': IMPROVING, STABLE, or DECLINING.
     """
 
     try:
-        # 3. Setting expectation for the 'Thinking' phase
-        progress(0.6, desc="Reasoning Spatially (This takes ~20-30s)...")
+        progress(0.5, desc="Robotics-ER is reasoning spatially...")
         
+        # We pass the audit level intent into the generation config
         response = client.models.generate_content(
             model="gemini-robotics-er-1.5-preview",
-            contents=[prompt, past_img, present_img]
+            contents=[prompt, past_img, present_img],
+            config={
+                "thinking_budget": selected_budget
+            }
         )
         
         progress(1.0, desc="Audit Complete!")
@@ -40,27 +44,32 @@ def ground_truth_engine(past_img, present_img, progress=gr.Progress()):
     except Exception as e:
         return f"Analysis Failed: {str(e)}"
 
-# --- UI Setup ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🏠 GroundTruth AI (Stable Build)")
+    gr.Markdown("# 🏠 GroundTruth AI: Spatial Property Sentinel")
     
     with gr.Row():
         with gr.Column():
             p_img = gr.Image(label="Past Condition", type="pil")
             c_img = gr.Image(label="Current Condition", type="pil")
-            submit = gr.Button("Analyze Trajectory", variant="primary")
+            
+            # New UI Element: Audit Level Slider
+            audit_slider = gr.Radio(
+                choices=["Quick Scan", "Standard Audit", "Deep Forensic"],
+                value="Standard Audit",
+                label="Audit Depth (Robotics Reasoning Level)"
+            )
+            
+            submit = gr.Button("Analyze Structural Trajectory", variant="primary")
         
         with gr.Column():
             output = gr.Markdown(label="Audit Report")
 
-    # Important: show_progress="full" ensures the spinner stays visible
     submit.click(
         fn=ground_truth_engine, 
-        inputs=[p_img, c_img], 
+        inputs=[p_img, c_img, audit_slider], 
         outputs=output,
         show_progress="full"
     )
 
 if __name__ == "__main__":
-    # Ensure queue is enabled for progress bar to update
-    demo.queue().launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)