Update app.py

app.py

@@ -1,95 +1,37 @@
-import gradio as gr
-from google import genai
-import os
-
-# 1. Custom CSS for Visual Trajectory Badges
-custom_css = """
-.improving { background-color: #dcf2dc; color: #155724; padding: 12px; border-radius: 8px; font-weight: bold; font-size: 1.2em; text-align: center; border: 1px solid #c3e6cb;}
-.declining { background-color: #f8d7da; color: #721c24; padding: 12px; border-radius: 8px; font-weight: bold; font-size: 1.2em; text-align: center; border: 1px solid #f5c6cb;}
-.stable { background-color: #fff3cd; color: #856404; padding: 12px; border-radius: 8px; font-weight: bold; font-size: 1.2em; text-align: center; border: 1px solid #ffeeba;}
-"""
-
-def ground_truth_engine(past_img, present_img, audit_level, progress=gr.Progress()):
-    progress(0, desc="Waking Spatial Sentinel...")
-    
-    api_key = os.environ.get("GOOGLE_API_KEY")
-    if not api_key:
-        return None, "<div class='declining'>ERROR: API Key Missing</div>", "Set GOOGLE_API_KEY in Secrets."
-
-    client = genai.Client(api_key=api_key)
-    if past_img is None or present_img is None:
-        return None, "<div class='stable'>Missing Images</div>", "Please upload both images."
-
-    audit_directives = {
-        "Standard Audit": "Perform a structural comparison of roofing, siding, and landscaping.",
-        "Deep Forensic": """Perform an exhaustive spatial audit. 
-        Identify subtle wear (shingles, foundation, paint) and biological encroachment. 
-        MANDATORY: Return center points for top 3 changes as [y, x] normalized coordinates (0-1000)."""
-    }
-    
-    directive = audit_directives.get(audit_level)
-    progress(0.2, desc=f"Running {audit_level}...")
-
-    prompt = f"""
-    SYSTEM INSTRUCTION: You are a structural forensic agent.
-    DIRECTIVE: {directive}
-    
-    TASK: Compare 'Past Condition' vs 'Current Condition'. 
-    Return a structured report. YOU MUST CONCLUDE with exactly one of these phrases: 
-    'Maintenance Trajectory: IMPROVING', 'Maintenance Trajectory: STABLE', or 'Maintenance Trajectory: DECLINING'.
-    """
-
-    try:
-        progress(0.5, desc="Robotics-ER is scanning for spatial anomalies...")
-        
-        response = client.models.generate_content(
-            model="gemini-robotics-er-1.5-preview",
-            contents=[prompt, past_img, present_img]
-        )
-        
-        text_out = response.text
-        
-        # 2. Logic to trigger the correct CSS Badge
-        badge_html = "<div class='stable'>➖ Trajectory: STABLE</div>" # Default fallback
-        if "IMPROVING" in text_out.upper():
-            badge_html = "<div class='improving'>📈 Trajectory: IMPROVING</div>"
-        elif "DECLINING" in text_out.upper():
-            badge_html = "<div class='declining'>📉 Trajectory: DECLINING</div>"
-            
-        progress(1.0, desc="Audit Finalized.")
-        
-        # 3. Return a tuple of images to automatically populate the Slider
-        return (past_img, present_img), badge_html, text_out
-        
-    except Exception as e:
-        return None, "<div class='declining'>Analysis Failed</div>", str(e)
-
 # Inject the CSS block into the main Gradio application 
 with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
     gr.Markdown("# 🏠 GroundTruth AI: Spatial Property Sentinel")
     
+    # --- TOP ROW: The "Input Console" ---
+    # Breaking this into 3 columns forces the images to be smaller and sit side-by-side
     with gr.Row():
-        # Left Side: The "Agent Console"
         with gr.Column(scale=1):
             p_img = gr.Image(label="Past Condition", type="pil")
+        with gr.Column(scale=1):
             c_img = gr.Image(label="Current Condition", type="pil")
+        with gr.Column(scale=1):
             audit_mode = gr.Radio(
                 choices=["Standard Audit", "Deep Forensic"],
                 value="Standard Audit",
                 label="Select Audit Precision"
             )
-            submit = gr.Button("Analyze Property Trajectory", variant="primary")
-        
-        # Right Side: The "Executive Summary"
+            # Added some margin-top to push the button down slightly for better alignment
+            submit = gr.Button("Analyze Property Trajectory", variant="primary", size="lg")
+            
+    # --- MIDDLE ROW: The "Hero Output" ---
+    # This row takes up the full width of the screen to highlight the visual evidence
+    with gr.Row():
         with gr.Column(scale=1):
-           # The CSS Badge renders immediately at the top
+            # The CSS Badge renders immediately above the slider
             status_badge = gr.HTML()
             
-            # USE THE NATIVE GRADIO 5 COMPONENT
-            slider_out = gr.ImageSlider(label="Spatial Delta Comparison", type="pil")
-            
+            # Using height=600 ensures the slider is large and premium-feeling
+            slider_out = gr.ImageSlider(label="Spatial Delta Comparison", type="pil", height=600)
             
-            # The Accordion keeps the raw LLM output hidden until needed 
+    # --- BOTTOM ROW: The "Forensic Data" ---
+    with gr.Row():
+        with gr.Column(scale=1):
+            # The Accordion spans the bottom, keeping the technical text neatly tucked away
             with gr.Accordion("View Detailed Forensic Data", open=False):
                 report_out = gr.Markdown()
 
@@ -98,7 +40,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
         fn=ground_truth_engine, 
         inputs=[p_img, c_img, audit_mode], 
         outputs=[slider_out, status_badge, report_out],
-        show_progress="full"
+        show_progress="minimal"
     )
 
 if __name__ == "__main__":