Update app.py

app.py

@@ -1,9 +1,104 @@
-# Inject the CSS block into the main Gradio application 
+import gradio as gr
+from google import genai
+import os
+import re
+from PIL import ImageDraw
+
+# 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 draw_forensic_targets(image, report_text):
+    """Parses [y, x] coordinates and draws red circles on the image."""
+    annotated_img = image.copy()
+    draw = ImageDraw.Draw(annotated_img)
+    width, height = annotated_img.size
+    
+    # Find all instances of [y, x] in the text
+    coords = re.findall(r'\[(\d+),\s*(\d+)\]', report_text)
+    
+    for y_str, x_str in coords:
+        # Denormalize (0-1000 scale) to actual image pixels
+        y = int((int(y_str) / 1000) * height)
+        x = int((int(x_str) / 1000) * width)
+        
+        # Draw a red circle around the target point
+        r = 15 
+        draw.ellipse((x - r, y - r, x + r, y + r), outline="red", width=5)
+        
+    return annotated_img
+
+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
+        
+        # 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(0.8, desc="Annotating image...")
+        
+        # Determine which image to show on the right side of the slider
+        final_present_img = present_img
+        if "Deep Forensic" in audit_level:
+            final_present_img = draw_forensic_targets(present_img, text_out)
+            
+        progress(1.0, desc="Audit Finalized.")
+        
+        # Return a tuple of images to automatically populate the Slider
+        return (past_img, final_present_img), badge_html, text_out
+            
+    except Exception as e:
+        return None, "<div class='declining'>Analysis Failed</div>", str(e)
+
+
+# --- UI LAYOUT ---
 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():
         with gr.Column(scale=1):
             p_img = gr.Image(label="Past Condition", type="pil")
@@ -15,23 +110,17 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
                 value="Standard Audit",
                 label="Select Audit Precision"
             )
-            # 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 above the slider
             status_badge = gr.HTML()
-            
-            # Using height=600 ensures the slider is large and premium-feeling
             slider_out = gr.ImageSlider(label="Spatial Delta Comparison", type="pil", height=600)
             
     # --- 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()