minor changes pt3!

app.py

@@ -1,326 +1,177 @@
 """
-app.py — GharScan HuggingFace Space (Improved UI)
+app.py — GharScan HuggingFace Space
+Uses gr.Blocks (ZeroGPU-compatible). Custom CSS for Off-Brand badge.
 """
+import os
 import spaces
 import gradio as gr
 from PIL import Image
+from pathlib import Path
+
 from inference import run_gharscan_pipeline
 from agent_trace import AgentTraceLogger
 
 trace_logger = AgentTraceLogger()
 
+# ── ZeroGPU-compatible inference ──────────────────────────────────────────────
 @spaces.GPU
-def analyze_image(image: Image.Image) -> str:
+def analyze_image(image: Image.Image, language: str) -> dict:
     if image is None:
-        return "<p style='color:#6b7280;padding:24px;text-align:center'>📷 Please upload or photograph a wall defect first.</p>"
+        return {}
     session = trace_logger.start_trace()
-    report  = run_gharscan_pipeline(image, language="en", trace_session=session)
+    report  = run_gharscan_pipeline(image, language=language, trace_session=session)
     trace_logger.save_trace(session)
-    return _render_report(report)
-
+    return report
 
-def _render_report(r: dict) -> str:
+def analyze_and_render(image, language):
+    if image is None:
+        return "<p style='color:#6b7280;padding:20px'>Please upload or take a photo first.</p>"
+    r = analyze_image(image, language)
     if not r.get("analysis_ok"):
-        desc = r.get("description", "Could not analyse the image.")
-        return f"""
-        <div style='background:#1a1d23;border:1px solid #2a2f38;border-radius:12px;
-                    padding:24px;text-align:center;color:#9ca3af;font-family:Inter,sans-serif'>
-            <div style='font-size:36px;margin-bottom:12px'>🔍</div>
-            <p style='font-weight:600;color:#e8eaed;margin-bottom:6px'>No defect detected</p>
-            <p style='font-size:13px'>{desc or 'Try photographing the defect area more closely with good lighting.'}</p>
+        return f"<p style='color:#ef4444;padding:20px'>⚠️ {r.get('description','Analysis failed.')}</p>"
+    
+    color  = r.get("severity_color", "#6b7280")
+    sev    = r.get("severity", 0)
+    pct    = sev * 20
+
+    struct_html = ""
+    if r.get("is_structural"):
+        struct_html = f"""<div style='background:rgba(127,29,29,0.25);border:1px solid #7f1d1d;border-radius:8px;padding:12px;margin-bottom:12px;color:#fca5a5'>
+        ⚠️ <strong>STRUCTURAL RISK</strong> — {r.get("structural_reasoning","")}
         </div>"""
+    else:
+        struct_html = "<div style='background:rgba(21,128,61,0.2);border:1px solid #166534;border-radius:8px;padding:10px;margin-bottom:12px;color:#86efac'>✅ Not Structural — No immediate safety risk</div>"
 
-    sev   = r.get("severity", 1)
-    color = r.get("severity_color", "#22c55e")
-    pct   = sev * 20
-
-    struct = (f"""<div style='background:rgba(127,29,29,0.3);border:1px solid #7f1d1d;
-              border-radius:10px;padding:13px 16px;margin-bottom:14px;
-              display:flex;gap:10px;align-items:flex-start;color:#fca5a5;font-size:13px'>
-              <span style='font-size:18px;flex-shrink:0'>⚠️</span>
-              <div><strong style='font-size:14px'>STRUCTURAL RISK</strong>
-              <br><span style='color:#f87171'>{r.get("structural_reasoning","")}</span></div></div>"""
-             if r.get("is_structural") else
-             """<div style='background:rgba(21,128,61,0.2);border:1px solid #166534;
-             border-radius:10px;padding:11px 16px;margin-bottom:14px;
-             color:#86efac;font-size:13px;display:flex;align-items:center;gap:8px'>
-             <span>✅</span><strong>Not Structural — No immediate safety risk</strong></div>""")
+    liability = ""
+    if r.get("show_liability_banner"):
+        liability = f"<div style='background:rgba(239,68,68,0.1);border:1px solid rgba(239,68,68,0.3);border-radius:8px;padding:12px;color:#fca5a5;font-size:12px;margin-top:10px'>⚠️ {r.get('liability_text','')}</div>"
 
-    monsoon = ("""<div style='background:rgba(234,179,8,0.12);border:1px solid rgba(234,179,8,0.35);
-               border-radius:8px;padding:10px 14px;color:#fde68a;font-size:13px;margin-top:10px'>
-               🌧️ <strong>Monsoon Risk</strong> — This defect worsens in heavy rain. Act before June.</div>"""
-               if r.get("monsoon_risk") else "")
+    disclaimer = ""
+    if r.get("disclaimer"):
+        disclaimer = f"<div style='border-left:3px solid #f59e0b;padding:10px 14px;font-size:12px;color:#9ca3af;margin-top:8px'>{r['disclaimer']}</div>"
 
-    liability = (f"""<div style='background:rgba(239,68,68,0.1);border:1px solid rgba(239,68,68,0.3);
-                 border-radius:10px;padding:12px 14px;font-size:12px;color:#fca5a5;
-                 margin-top:10px;display:flex;gap:8px'>
-                 <span style='flex-shrink:0'>⚠️</span><p>{r.get("liability_text","")}</p></div>"""
-                 if r.get("show_liability_banner") else "")
-
-    disclaimer = (f"""<div style='border-left:3px solid #f59e0b;padding:10px 14px;
-                  font-size:12px;color:#9ca3af;margin-top:8px;background:#1a1d23;
-                  border-radius:0 8px 8px 0'>{r["disclaimer"]}</div>"""
-                  if r.get("disclaimer") else "")
+    monsoon = ""
+    if r.get("monsoon_risk"):
+        monsoon = "<div style='background:rgba(234,179,8,0.1);border:1px solid rgba(234,179,8,0.3);border-radius:6px;padding:10px;color:#fde68a;font-size:13px;margin-top:10px'>🌧️ <strong>Monsoon Risk:</strong> This defect worsens during heavy rainfall. Address before June.</div>"
 
     return f"""
-<div style='font-family:Inter,sans-serif;color:#e8eaed;max-width:640px'>
-  {struct}
-
-  <div style='background:#181b20;border:1px solid #2a2f38;border-radius:14px;
-              padding:20px;box-shadow:0 4px 16px rgba(0,0,0,0.5)'>
-
-    <!-- Header row -->
-    <div style='display:flex;justify-content:space-between;align-items:flex-start;margin-bottom:16px;gap:12px'>
-      <div>
-        <div style='font-size:20px;font-weight:700;line-height:1.2'>{r.get("defect_display","")}</div>
-        <div style='font-size:11px;color:#6b7280;margin-top:4px;font-family:monospace;letter-spacing:.05em'>
-          {r.get("defect_type","").replace("_"," ").upper()}</div>
-      </div>
-      <div style='background:#20242b;border:2px solid {color};border-radius:12px;
-                  padding:8px 14px;text-align:center;flex-shrink:0;min-width:72px'>
-        <div style='font-size:26px;font-weight:700;color:{color};font-family:monospace;line-height:1'>{sev}</div>
-        <div style='font-size:10px;color:#8b9099;letter-spacing:.06em;margin-top:2px'>
-          {r.get("severity_label","").upper()}</div>
-      </div>
+    <div style='background:#181b20;border:1px solid #2a2f38;border-radius:14px;padding:20px;font-family:Inter,sans-serif;color:#e8eaed;max-width:600px'>
+        {struct_html}
+        <div style='display:flex;justify-content:space-between;align-items:flex-start;margin-bottom:14px'>
+            <div>
+                <div style='font-size:18px;font-weight:700'>{r.get("defect_display","")}</div>
+                <div style='font-size:11px;color:#8b9099;margin-top:3px'>{r.get("defect_type","").replace("_"," ").upper()}</div>
+            </div>
+            <div style='background:#20242b;border:2px solid {color};border-radius:10px;padding:8px 14px;text-align:center'>
+                <div style='font-size:22px;font-weight:700;color:{color}'>{sev}</div>
+                <div style='font-size:10px;color:#8b9099'>{r.get("severity_label","").upper()}</div>
+            </div>
+        </div>
+        <div style='background:#20242b;border-radius:4px;height:8px;margin-bottom:4px'>
+            <div style='width:{pct}%;height:8px;border-radius:4px;background:{color}'></div>
+        </div>
+        <div style='display:flex;justify-content:space-between;font-size:10px;color:#555d6b;margin-bottom:14px'>
+            <span>Cosmetic</span><span>Moderate</span><span>Critical</span>
+        </div>
+        <div style='height:1px;background:#1e2229;margin:12px 0'></div>
+        <div style='margin-bottom:10px'>
+            <div style='font-size:10px;color:#6fb3e0;font-family:monospace;letter-spacing:.06em'>WHAT IT IS</div>
+            <div style='font-size:14px;margin-top:4px'>{r.get("description","")}</div>
+        </div>
+        <div style='margin-bottom:10px'>
+            <div style='font-size:10px;color:#6fb3e0;font-family:monospace;letter-spacing:.06em'>WHY IT HAPPENS</div>
+            <div style='font-size:14px;margin-top:4px'>{r.get("primary_cause","")}</div>
+        </div>
+        <div style='height:1px;background:#1e2229;margin:12px 0'></div>
+        <div style='background:rgba(59,130,246,0.08);border:1px solid rgba(59,130,246,0.2);border-radius:10px;padding:12px;margin-bottom:10px'>
+            <div style='font-size:10px;color:#6fb3e0;font-family:monospace'>WHAT TO DO</div>
+            <div style='font-size:14px;font-weight:500;color:#93c5fd;margin-top:4px'>{r.get("immediate_action","")}</div>
+        </div>
+        <div style='margin-bottom:12px'>
+            <div style='font-size:10px;color:#6fb3e0;font-family:monospace'>WHEN TO ACT</div>
+            <div style='font-size:14px;font-weight:500;margin-top:4px'>{r.get("urgency_display","")}</div>
+        </div>
+        <div style='height:1px;background:#1e2229;margin:12px 0'></div>
+        <div style='background:#20242b;border-radius:10px;padding:14px'>
+            <div style='font-size:22px;font-weight:700;color:#22c55e'>{r.get("cost_range_inr","")}</div>
+            <div style='font-size:13px;color:#8b9099;margin-top:6px'>👷 {r.get("professional_display","")}</div>
+        </div>
+        {monsoon}
+        {liability}
+        {disclaimer}
     </div>
+    """
 
-    <!-- Severity bar -->
-    <div style='background:#1a1d23;border-radius:4px;height:8px;margin-bottom:5px;overflow:hidden'>
-      <div style='width:{pct}%;height:8px;border-radius:4px;background:{color};
-                  transition:width 0.6s ease'></div>
-    </div>
-    <div style='display:flex;justify-content:space-between;font-size:10px;
-                color:#4b5563;font-family:monospace;margin-bottom:18px'>
-      <span>Cosmetic</span><span>Moderate</span><span>Critical</span>
-    </div>
-
-    <div style='height:1px;background:#1e2229;margin-bottom:16px'></div>
-
-    <!-- Description rows -->
-    <div style='margin-bottom:12px'>
-      <div style='font-size:10px;color:#6fb3e0;font-family:monospace;letter-spacing:.08em;
-                  margin-bottom:4px'>WHAT IT IS</div>
-      <div style='font-size:14px;line-height:1.5'>{r.get("description","")}</div>
-    </div>
-    <div style='margin-bottom:12px'>
-      <div style='font-size:10px;color:#6fb3e0;font-family:monospace;letter-spacing:.08em;
-                  margin-bottom:4px'>WHY IT HAPPENS</div>
-      <div style='font-size:14px;line-height:1.5'>{r.get("primary_cause","")}</div>
-    </div>
-
-    <div style='height:1px;background:#1e2229;margin-bottom:16px'></div>
-
-    <!-- Action block -->
-    <div style='background:rgba(59,130,246,0.09);border:1px solid rgba(59,130,246,0.25);
-                border-radius:10px;padding:13px 15px;margin-bottom:12px'>
-      <div style='font-size:10px;color:#6fb3e0;font-family:monospace;
-                  letter-spacing:.08em;margin-bottom:5px'>WHAT TO DO</div>
-      <div style='font-size:14px;font-weight:500;color:#93c5fd;line-height:1.5'>
-        {r.get("immediate_action","Consult a professional.")}</div>
-    </div>
-    <div style='margin-bottom:16px'>
-      <div style='font-size:10px;color:#6fb3e0;font-family:monospace;
-                  letter-spacing:.08em;margin-bottom:4px'>WHEN TO ACT</div>
-      <div style='font-size:14px;font-weight:500'>{r.get("urgency_display","")}</div>
-    </div>
-
-    <div style='height:1px;background:#1e2229;margin-bottom:16px'></div>
-
-    <!-- Cost block -->
-    <div style='background:#20242b;border-radius:12px;padding:16px'>
-      <div style='font-size:26px;font-weight:700;color:#22c55e;
-                  font-family:monospace;margin-bottom:6px'>{r.get("cost_range_inr","")}</div>
-      <div style='font-size:13px;color:#8b9099'>
-        👷 {r.get("professional_display","")}</div>
-    </div>
-
-    {monsoon}
-  </div>
-  {liability}
-  {disclaimer}
-</div>"""
-
-
-# ── CSS ────────────────────────────────────────────────────────────────────────
+# ── Custom CSS ─────────────────────────────────────────────────────────────────
 CSS = """
-@import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600;700&display=swap');
-
-*, *::before, *::after { box-sizing: border-box; }
-body, .gradio-container {
-    background: #0c0e12 !important;
-    color: #e8eaed !important;
-    font-family: 'Inter', system-ui, sans-serif !important;
-}
-.gradio-container { max-width: 680px !important; margin: 0 auto !important; padding: 0 !important; }
-.main { padding: 0 16px 60px !important; }
-footer, .footer { display: none !important; }
-
-/* Image upload zone */
-.image-upload-zone {
-    border: 2px dashed #2a2f38 !important;
-    border-radius: 14px !important;
-    background: #181b20 !important;
-    min-height: 280px !important;
-    transition: border-color 0.2s !important;
-}
-.image-upload-zone:hover { border-color: #3b82f6 !important; }
-.image-upload-zone .upload-text { color: #6b7280 !important; }
-.image-upload-zone svg { color: #374151 !important; }
-
-/* Analyse button */
-button.primary {
-    background: linear-gradient(135deg, #2563eb, #1d4ed8) !important;
-    border: none !important;
-    border-radius: 10px !important;
-    font-size: 15px !important;
-    font-weight: 600 !important;
-    padding: 14px !important;
-    letter-spacing: 0.02em !important;
-    box-shadow: 0 4px 14px rgba(37,99,235,0.4) !important;
-    transition: opacity 0.15s, transform 0.1s !important;
-}
-button.primary:hover:not(:disabled) { opacity: 0.9 !important; transform: translateY(-1px) !important; }
-button.primary:disabled { opacity: 0.35 !important; box-shadow: none !important; }
-
-/* Output HTML area */
-.output-html { background: transparent !important; padding: 0 !important; }
-
-/* Examples */
-.examples-holder { margin-top: 4px !important; }
-.examples-holder table { background: #181b20 !important; border-radius: 10px !important; }
-.examples-holder td { padding: 6px !important; }
-.examples-holder img { border-radius: 6px !important; }
-
-/* Radio buttons hidden (language removed) */
-.gr-radio { display: none !important; }
-
-/* Labels */
-label.block, .block > label { color: #6b7280 !important; font-size: 12px !important; font-family: monospace !important; letter-spacing: .06em !important; }
-
-/* Scrollbar */
-::-webkit-scrollbar { width: 6px; } ::-webkit-scrollbar-track { background: #0c0e12; }
-::-webkit-scrollbar-thumb { background: #2a2f38; border-radius: 3px; }
-"""
-
-HEADER = """
-<div style='display:flex;align-items:center;justify-content:space-between;
-            padding:16px 0 18px;border-bottom:1px solid #1e2229;margin-bottom:20px'>
-  <div style='display:flex;align-items:center;gap:12px'>
-    <span style='font-size:30px'>🏗️</span>
-    <div>
-      <div style='font-size:20px;font-weight:700;color:#e8eaed;letter-spacing:-0.3px'>GharScan</div>
-      <div style='font-size:11px;color:#6b7280;margin-top:1px'>AI Building Defect Inspector · India</div>
-    </div>
-  </div>
-  <div style='display:flex;flex-direction:column;align-items:flex-end;gap:6px'>
-    <div style='background:#0d1f0d;border:1px solid #166534;border-radius:20px;
-                padding:4px 12px;font-size:11px;color:#86efac;font-family:monospace;
-                display:flex;align-items:center;gap:6px'>
-      <span style='width:6px;height:6px;border-radius:50%;background:#22c55e;
-                   animation:pulse 2s infinite;display:inline-block'></span>
-      ZeroGPU · Online
-    </div>
-    <div style='font-size:10px;color:#374151;font-family:monospace'>Qwen2-VL-2B · 2.07B params</div>
-  </div>
-</div>
-<style>@keyframes pulse{0%,100%{opacity:1}50%{opacity:.4}}</style>
-"""
-
-HOW_IT_WORKS = """
-<div style='display:grid;grid-template-columns:1fr 1fr 1fr;gap:8px;margin:16px 0 4px;'>
-  <div style='background:#181b20;border:1px solid #1e2229;border-radius:10px;
-              padding:12px;text-align:center;font-size:12px;color:#6b7280'>
-    <div style='font-size:20px;margin-bottom:6px'>📸</div>
-    <strong style='color:#9ca3af;display:block;margin-bottom:3px'>1. Photograph</strong>
-    Aim at any crack or stain
-  </div>
-  <div style='background:#181b20;border:1px solid #1e2229;border-radius:10px;
-              padding:12px;text-align:center;font-size:12px;color:#6b7280'>
-    <div style='font-size:20px;margin-bottom:6px'>🔍</div>
-    <strong style='color:#9ca3af;display:block;margin-bottom:3px'>2. Analyse</strong>
-    AI inspects in ~12 sec
-  </div>
-  <div style='background:#181b20;border:1px solid #1e2229;border-radius:10px;
-              padding:12px;text-align:center;font-size:12px;color:#6b7280'>
-    <div style='font-size:20px;margin-bottom:6px'>₹</div>
-    <strong style='color:#9ca3af;display:block;margin-bottom:3px'>3. Get Answer</strong>
-    Type · Severity · Cost
-  </div>
-</div>
+body, .gradio-container { background: #0f1114 !important; color: #e8eaed !important; }
+.gradio-container { max-width: 700px !important; margin: 0 auto !important; }
+.gr-button-primary { background: linear-gradient(135deg,#2563eb,#1d4ed8) !important; border: none !important; }
+.gr-button-primary:hover { opacity: 0.9 !important; }
+footer { display: none !important; }
+#component-0 { padding: 20px !important; }
+.dark { --background-fill-primary: #181b20; --background-fill-secondary: #20242b; --border-color-primary: #2a2f38; --color-text-body: #e8eaed; }
 """
 
-FOOTER_HTML = """
-<div style='text-align:center;padding:28px 0 8px;border-top:1px solid #1e2229;
-            font-size:11px;color:#374151;font-family:monospace;line-height:2'>
-  <p style='color:#4b5563'>Qwen2-VL-2B fine-tuned on 8,860 deduplicated defect images · No cloud APIs</p>
-  <p>
-    <a href='https://huggingface.co/ritvik360/gharscan-qwen2vl-lora' target='_blank' style='color:#4b5563;text-decoration:none'>Model</a>
-    · <a href='https://huggingface.co/datasets/ritvik360/gharscan-defect-dataset' target='_blank' style='color:#4b5563;text-decoration:none'>Dataset</a>
-    · <a href='https://huggingface.co/ritvik360/gharscan-qwen2vl-gguf' target='_blank' style='color:#4b5563;text-decoration:none'>GGUF</a>
-    · <a href='https://huggingface.co/blog/ritvik360/gharscan' target='_blank' style='color:#4b5563;text-decoration:none'>Blog</a>
-  </p>
-  <p style='color:#1f2937;margin-top:4px'>
-    🏗️ Build Small Hackathon 2026 · Backyard AI Track
-  </p>
-  <p style='color:#1f2937'>GharScan is a triage aid, not a substitute for professional structural assessment.</p>
-</div>
-"""
-
-# ── Build UI ───────────────────────────────────────────────────────────────────
+# ── UI ──────────────────────────────────────────────────────────────────────────
 with gr.Blocks(
     css=CSS,
-    title="GharScan — AI Building Defect Inspector",
+    title="GharScan — Building Defect Inspector",
     theme=gr.themes.Base(
         primary_hue="blue",
         neutral_hue="slate",
-        font=gr.themes.GoogleFont("Inter"),
     )
 ) as demo:
-
-    gr.HTML(HEADER)
-    gr.HTML(HOW_IT_WORKS)
-
-    image_input = gr.Image(
-        sources=["upload", "webcam"],
-        type="pil",
-        label="📷  DROP IMAGE · CLICK TO UPLOAD · USE WEBCAM",
-        height=300,
-        elem_classes=["image-upload-zone"],
-        show_download_button=False,
-        show_share_button=False,
-    )
-
-    analyse_btn = gr.Button(
-        "🔍  Analyse Defect",
-        variant="primary",
-        size="lg",
+    gr.HTML("""
+    <div style='display:flex;align-items:center;justify-content:space-between;
+                padding:14px 0;border-bottom:1px solid #2a2f38;margin-bottom:20px'>
+        <div style='display:flex;align-items:center;gap:10px'>
+            <span style='font-size:26px'>🏗️</span>
+            <div>
+                <div style='font-size:18px;font-weight:700;color:#e8eaed'>GharScan</div>
+                <div style='font-size:11px;color:#8b9099'>AI Building Defect Inspector · India</div>
+            </div>
+        </div>
+        <div style='background:#20242b;border:1px solid #2a2f38;border-radius:20px;padding:4px 12px;
+                    font-size:11px;color:#6fb3e0;font-family:monospace'>
+            ● Qwen2-VL-2B · 2.07B
+        </div>
+    </div>
+    """)
+
+    with gr.Row():
+        image_input = gr.Image(
+            sources=["upload", "webcam"],
+            type="pil",
+            label="📸 Take Photo or Upload",
+            height=300,
+        )
+
+    language = gr.Radio(
+        choices=["en", "hi"],
+        value="en",
+        label="Output language",
     )
 
-    gr.HTML("<div style='height:4px'></div>")
-
-    # Example images
-    gr.Examples(
-        examples=[
-            ["https://upload.wikimedia.org/wikipedia/commons/thumb/3/3e/Cracked_Wall.jpg/640px-Cracked_Wall.jpg"],
-        ],
-        inputs=image_input,
-        label="▸ Try an example",
-        examples_per_page=3,
-    )
+    analyze_btn = gr.Button("🔍 Analyse Defect", variant="primary", size="lg")
+    report_output = gr.HTML(label="Inspection Report")
 
-    report_output = gr.HTML(
-        value="",
-        elem_classes=["output-html"],
+    analyze_btn.click(
+        fn=analyze_and_render,
+        inputs=[image_input, language],
+        outputs=report_output,
+        api_name="analyze"
     )
 
-    gr.HTML(FOOTER_HTML)
-
-    analyse_btn.click(
-        fn=analyze_image,
-        inputs=[image_input],
-        outputs=[report_output],
-        api_name="analyze",
-    )
+    gr.HTML("""
+    <div style='text-align:center;padding:20px 0;font-size:11px;color:#555d6b;
+                border-top:1px solid #1e2229;margin-top:20px'>
+        <p>Qwen2-VL-2B fine-tuned on Indian building defects · No cloud APIs</p>
+        <p>🏗️ Built for <a href="https://huggingface.co/build-small-hackathon" 
+           style='color:#8b9099'>Build Small Hackathon 2026</a> · Backyard AI Track</p>
+        <p style='color:#374151'>GharScan is a triage aid, not a substitute for professional structural assessment.</p>
+    </div>
+    """)
 
-demo.queue(max_size=10)
+demo.queue()
 demo.launch()