Update app.py

app.py

@@ -3,24 +3,24 @@ from google import genai
 import os
 
 def ground_truth_audit(past_img, present_img):
+    """Explicitly mapped backend function for GroundTruth."""
     api_key = os.environ.get("GOOGLE_API_KEY")
     if not api_key:
-        return "## Error: GOOGLE_API_KEY not found in Space Secrets."
+        return "## Error\nAPI Key not found. Please add GOOGLE_API_KEY to Space Secrets."
 
     client = genai.Client(api_key=api_key)
     
-    # Robotics-ER 1.5 requires precise prompts for spatial reasoning
+    if past_img is None or present_img is None:
+        return "Please upload both images."
+
     prompt = """
     Perform a structural audit comparing these two images.
-    1. Roofing & Gutters: Look for shingle wear or replacement.
-    2. Landscaping: Identify maturity or neglect.
-    3. Exterior: Check paint and siding condition.
-    
-    Conclude with a 'Maintenance Trajectory': IMPROVING, STABLE, or DECLINING.
+    Look for changes in roofing, landscaping, and exterior condition.
+    Provide a 'Maintenance Trajectory': IMPROVING, STABLE, or DECLINING.
     """
 
     try:
-        # Call the Robotics-ER 1.5 preview model
+        # Utilizing Robotics-ER 1.5 for spatial/temporal analysis
         response = client.models.generate_content(
             model="gemini-robotics-er-1.5-preview",
             contents=[prompt, past_img, present_img]
@@ -29,18 +29,24 @@ def ground_truth_audit(past_img, present_img):
     except Exception as e:
         return f"## Analysis Failed\n{str(e)}"
 
-# The Interface class is the 'Architectural Safe Mode' for Gradio
-demo = gr.Interface(
-    fn=ground_truth_audit,
-    inputs=[
-        gr.Image(label="Past Condition", type="pil"),
-        gr.Image(label="Present Condition", type="pil")
-    ],
-    outputs=gr.Markdown(),
-    title="GroundTruth AI",
-    description="Spatial Property Analysis via Gemini Robotics-ER 1.5",
-    theme="default"
-)
+# Explicitly defining the UI blocks to fix the 'No API found' error
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🏠 GroundTruth AI")
+    gr.Markdown("Temporal Property Sentinel powered by Gemini Robotics-ER 1.5")
+    
+    with gr.Row():
+        with gr.Column():
+            past = gr.Image(label="Past Condition", type="pil")
+            present = gr.Image(label="Present Condition", type="pil")
+            # Explicitly naming the trigger button
+            btn = gr.Button("Analyze Structural Trajectory", variant="primary")
+        
+        with gr.Column():
+            # Explicitly naming the output container
+            out = gr.Markdown(label="Audit Report")
+
+    # Manually mapping the click to the API function
+    btn.click(fn=ground_truth_audit, inputs=[past, present], outputs=out)
 
-if __name__ == "__main__":
-    demo.launch()
+# Mandatory for Gradio 5+ to handle background API processing
+demo.queue().launch()