Update app.py

app.py

@@ -2,16 +2,22 @@ import gradio as gr
 from google import genai
 import os
 
-def audit_engine(past_img, present_img):
+def ground_truth_engine(past_img, present_img):
     api_key = os.environ.get("GOOGLE_API_KEY")
     if not api_key:
-        return "ERROR: Set GOOGLE_API_KEY in Space Secrets."
+        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 "Upload both images."
+        return "Please upload both images."
 
-    prompt = "Perform a structural audit. Compare roofing, landscaping, and exterior condition. Output a 'Maintenance Trajectory': IMPROVING, STABLE, or DECLINING."
+    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.
+    """
 
     try:
         response = client.models.generate_content(
@@ -22,15 +28,22 @@ def audit_engine(past_img, present_img):
     except Exception as e:
         return f"Analysis Failed: {str(e)}"
 
+# Define the UI using Gradio 4 syntax
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🏠 GroundTruth AI (Docker Stable)")
+    gr.Markdown("# 🏠 GroundTruth AI (Stable Build)")
+    gr.Markdown("Temporal Property Sentinel powered by Gemini Robotics-ER 1.5")
+    
     with gr.Row():
-        p_img = gr.Image(label="Past", type="pil")
-        c_img = gr.Image(label="Present", type="pil")
-    output = gr.Markdown(label="Audit Report")
-    submit = gr.Button("Analyze", variant="primary")
-    submit.click(fn=audit_engine, inputs=[p_img, c_img], outputs=output)
+        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")
+        
+        with gr.Column():
+            output = gr.Markdown(label="Audit Report")
+
+    submit.click(fn=ground_truth_engine, inputs=[p_img, c_img], outputs=output)
 
 if __name__ == "__main__":
-    # In Docker, you MUST bind to 0.0.0.0 and port 7860
+    # Standard Docker bindings for Hugging Face Spaces
     demo.queue().launch(server_name="0.0.0.0", server_port=7860)