Upload app.py

app.py

@@ -0,0 +1,400 @@
+"""
+KhidmatAI — AI Service Orchestrator for Informal Economy
+Streamlit interface for HuggingFace Spaces (free tier)
+"""
+
+import streamlit as st
+import json, time, os, sys
+from datetime import datetime
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).parent))
+from agents import intent as intent_agent
+from agents import discovery as discovery_agent
+from agents import recommendation as recommendation_agent
+from agents import booking as booking_agent
+
+# ════════════════════════════════════════════════════════════════════
+# PAGE CONFIG
+# ════════════════════════════════════════════════════════════════════
+st.set_page_config(page_title="KhidmatAI", page_icon="🔧", layout="centered")
+
+st.markdown("""
+<style>
+@import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600;700&display=swap');
+html, body, [class*="css"] { font-family: 'Inter', sans-serif; }
+
+.hero {
+    background: linear-gradient(135deg, #1a1a2e 0%, #16213e 50%, #0f3460 100%);
+    border-radius: 16px; padding: 32px 28px 24px;
+    text-align: center; margin-bottom: 24px;
+    border: 1px solid rgba(255,255,255,0.08);
+}
+.hero h1 { color:#fff; font-size:2rem; font-weight:700; margin:0; }
+.hero p  { color:rgba(255,255,255,0.65); font-size:0.95rem; margin:8px 0 0; }
+.hero-badge {
+    display:inline-block; background:rgba(99,179,237,0.15);
+    border:1px solid rgba(99,179,237,0.3); color:#90cdf4;
+    font-size:0.75rem; padding:3px 10px; border-radius:20px;
+    margin-bottom:12px; font-weight:500;
+}
+.key-box {
+    background:#fffbeb; border:1.5px solid #f6ad55;
+    border-radius:12px; padding:16px 18px; margin-bottom:18px;
+}
+.key-box h4 { color:#c05621; margin:0 0 8px; font-size:0.95rem; }
+.key-box p  { color:#744210; font-size:0.83rem; margin:4px 0; line-height:1.6; }
+.key-box code { background:#feebc8; padding:1px 5px; border-radius:4px; font-size:0.82rem; }
+.mode-pill {
+    display:inline-block; font-size:0.72rem; font-weight:600;
+    padding:2px 9px; border-radius:10px; margin-left:6px;
+}
+.mode-gemini  { background:#ebf8ff; color:#2b6cb0; }
+.mode-rules   { background:#fefcbf; color:#744210; }
+.agent-card {
+    background:#f8fafc; border:1px solid #e2e8f0;
+    border-left:4px solid #4299e1; border-radius:10px;
+    padding:14px 16px; margin-bottom:10px;
+}
+.agent-card.done  { border-left-color:#48bb78; background:#f0fff4; }
+.agent-card.error { border-left-color:#fc8181; background:#fff5f5; }
+.agent-title { font-weight:600; font-size:0.9rem; color:#2d3748; margin-bottom:4px; }
+.agent-body  { font-size:0.82rem; color:#4a5568; line-height:1.5; }
+.provider-card {
+    background:#fff; border:1.5px solid #e2e8f0;
+    border-radius:12px; padding:16px 18px; margin-bottom:10px;
+}
+.provider-card.best { border-color:#48bb78; background:#f0fff4; }
+.provider-name { font-size:1rem; font-weight:600; color:#1a202c; }
+.provider-meta { font-size:0.8rem; color:#718096; margin-top:4px; }
+.badge-best {
+    background:#c6f6d5; color:#276749; font-size:0.72rem;
+    padding:2px 8px; border-radius:8px; font-weight:600; margin-left:8px;
+}
+.score-bar  { height:5px; background:#e2e8f0; border-radius:3px; margin-top:8px; }
+.score-fill { height:5px; background:#48bb78; border-radius:3px; }
+.receipt {
+    background:linear-gradient(135deg,#f0fff4,#e6fffa);
+    border:2px solid #9ae6b4; border-radius:16px;
+    padding:24px; text-align:center;
+}
+.receipt-id   { font-size:0.78rem; color:#68d391; letter-spacing:2px; font-weight:600; }
+.receipt-name { font-size:1.4rem; font-weight:700; color:#1a202c; margin:8px 0 4px; }
+.receipt-row  {
+    display:flex; justify-content:space-between; padding:8px 0;
+    border-bottom:1px solid rgba(0,0,0,0.06); font-size:0.88rem;
+}
+.receipt-row:last-child { border-bottom:none; }
+.receipt-label { color:#718096; }
+.receipt-value { color:#1a202c; font-weight:500; }
+.reminder-box {
+    background:#ebf8ff; border:1px solid #90cdf4;
+    border-radius:10px; padding:12px 16px; margin-top:14px;
+    font-size:0.85rem; color:#2b6cb0;
+}
+.trace-header {
+    font-size:0.75rem; font-weight:600; color:#a0aec0;
+    text-transform:uppercase; letter-spacing:1px; margin-bottom:6px;
+}
+</style>
+""", unsafe_allow_html=True)
+
+# ════════════════════════════════════════════════════════════════════
+# SESSION STATE
+# ════════════════════════════════════════════════════════════════════
+for k, v in [("result", None), ("trace_log", None)]:
+    if k not in st.session_state:
+        st.session_state[k] = v
+
+# ═══���════════════════════════════════════════════════════════════════
+# HERO
+# ════════════════════════════════════════════════════════════════════
+st.markdown("""
+<div class="hero">
+    <div class="hero-badge">🤖 Google Antigravity · Gemini API · Streamlit</div>
+    <h1>🔧 KhidmatAI</h1>
+    <p>AI Service Orchestrator for Pakistan's Informal Economy<br>
+    Type your request in <strong>Urdu · Roman Urdu · English</strong></p>
+</div>
+""", unsafe_allow_html=True)
+
+# ════════════════════════════════════════════════════════════════════
+# API KEY STATUS BANNER
+# ════════════════════════════════════════════════════════════════════
+api_key = os.environ.get("GEMINI_API_KEY", "").strip()
+key_valid = bool(api_key and api_key != "your_gemini_api_key_here" and len(api_key) > 20)
+
+if not key_valid:
+    st.markdown("""
+    <div class="key-box">
+        <h4>⚠️ Gemini API Key Not Set — Running in Demo Mode</h4>
+        <p>The app still works using a <strong>rule-based parser</strong>, but Gemini AI gives better results.</p>
+        <p>To enable full AI mode:</p>
+        <p>
+        1️⃣ Get a <strong>free</strong> key at <code>aistudio.google.com/apikey</code><br>
+        2️⃣ In your HuggingFace Space → <strong>Settings → Variables and Secrets</strong><br>
+        3️⃣ Click <strong>New secret</strong> → Name: <code>GEMINI_API_KEY</code> → paste your key → Save<br>
+        4️⃣ Restart the Space (Settings → Factory reboot)
+        </p>
+    </div>
+    """, unsafe_allow_html=True)
+else:
+    st.success("✅ Gemini API key detected — AI mode active")
+
+# ════════════════════════════════════════════════════════════════════
+# EXAMPLE BUTTONS
+# ════════════════════════════════════════════════════════════════════
+st.markdown("**Try an example:**")
+examples = [
+    "Mujhe kal subah G-13 mein AC technician chahiye",
+    "I need a plumber in F-10 today",
+    "G-9 mein electrician chahiye aaj evening",
+    "Maths tutor chahiye G-11 mein kal",
+    "Home maid service chahiye F-7 mein",
+    "Driver chahiye kal airport ke liye G-10 se",
+]
+cols = st.columns(3)
+selected_example = None
+for i, ex in enumerate(examples):
+    if cols[i % 3].button(f"💬 {ex[:28]}…", key=f"ex_{i}", use_container_width=True):
+        selected_example = ex
+
+# ════════════════════════════════════════════════════════════════════
+# INPUT
+# ════════════════════════════════════════════════════════════════════
+user_input = st.text_area(
+    "Request:",
+    value=selected_example or st.session_state.get("last_input", ""),
+    height=90,
+    placeholder="e.g. Mujhe kal subah G-13 mein AC technician chahiye",
+    label_visibility="collapsed",
+)
+c1, c2 = st.columns([3, 1])
+run_btn   = c1.button("🚀 Find & Book Service", type="primary", use_container_width=True)
+clear_btn = c2.button("🔄 Clear", use_container_width=True)
+
+if clear_btn:
+    st.session_state.result = None
+    st.session_state.trace_log = None
+    st.rerun()
+
+# ════════════════════════════════════════════════════════════════════
+# PIPELINE
+# ════════════════════════════════════════════════════════════════════
+def agent_card_running(title, desc):
+    st.markdown(f"""<div class="agent-card">
+        <div class="agent-title">⏳ {title}</div>
+        <div class="agent-body">{desc}</div>
+    </div>""", unsafe_allow_html=True)
+
+def agent_card_done(title, body, ms, mode=None):
+    mode_pill = ""
+    if mode == "gemini":
+        mode_pill = '<span class="mode-pill mode-gemini">Gemini AI</span>'
+    elif mode in ("rule_based_fallback", "rules"):
+        mode_pill = '<span class="mode-pill mode-rules">Rule-based</span>'
+    st.markdown(f"""<div class="agent-card done">
+        <div class="agent-title">✅ {title}{mode_pill} <small style="color:#718096">({ms}ms)</small></div>
+        <div class="agent-body">{body}</div>
+    </div>""", unsafe_allow_html=True)
+
+def agent_card_error(title, err):
+    st.markdown(f"""<div class="agent-card error">
+        <div class="agent-title">❌ {title} Failed</div>
+        <div class="agent-body"><code>{err}</code></div>
+    </div>""", unsafe_allow_html=True)
+
+def run_pipeline(message: str):
+    trace = []
+    st.markdown("---")
+    st.markdown("### 🤖 Agent Pipeline")
+
+    # Agent 1 — Intent
+    ph1 = st.empty()
+    with ph1:
+        agent_card_running("Agent 1 — Intent Agent", "Parsing your request...")
+    t0 = time.time()
+    try:
+        intent = intent_agent.run(message)
+        ms = int((time.time()-t0)*1000)
+        warning = intent.pop("_warning", None)
+        mode    = intent.get("mode", "rules")
+        trace.append({"agent":"IntentAgent","step":1,
+                       "input":{"message":message},"output":intent,
+                       "duration_ms":ms,
+                       "reasoning":f"Language={intent.get('language')}, service={intent.get('service_type')}, "
+                                   f"location={intent.get('location')}, time={intent.get('time')} [mode={mode}]"})
+        ph1.empty()
+        agent_card_done(
+            "Agent 1 — Intent Agent",
+            f"🏷️ <b>{intent.get('service_type','?')}</b> &nbsp;|&nbsp; "
+            f"📍 <b>{intent.get('location','?')}</b> &nbsp;|&nbsp; "
+            f"🕐 <b>{intent.get('time','?')}</b> &nbsp;|&nbsp; "
+            f"🌐 <b>{intent.get('language','?')}</b>",
+            ms, mode
+        )
+        if warning:
+            st.caption(f"ℹ️ {warning}")
+    except Exception as e:
+        ph1.empty()
+        agent_card_error("Agent 1 — Intent Agent", str(e))
+        return None, trace
+
+    # Agent 2 — Discovery
+    ph2 = st.empty()
+    with ph2:
+        agent_card_running("Agent 2 — Discovery Agent", "Searching nearby providers...")
+    t0 = time.time()
+    try:
+        disc = discovery_agent.run(intent)
+        ms = int((time.time()-t0)*1000)
+        trace.append({"agent":"DiscoveryAgent","step":2,
+                       "input":intent,"output":{"total_found":disc["total_found"]},
+                       "duration_ms":ms,
+                       "reasoning":f"Filtered providers.json for '{intent.get('service_type')}' near {disc['user_location']}. "
+                                   f"Found {disc['total_found']} provider(s). Distance via haversine formula."})
+        ph2.empty()
+        agent_card_done(
+            "Agent 2 — Discovery Agent",
+            f"Found <b>{disc['total_found']}</b> provider(s) for "
+            f"<b>{intent.get('service_type')}</b> near <b>{disc['user_location']}</b>.",
+            ms
+        )
+    except Exception as e:
+        ph2.empty()
+        agent_card_error("Agent 2 — Discovery Agent", str(e))
+        return None, trace
+
+    if disc["total_found"] == 0:
+        st.warning("😕 No providers found. Try a different service or location.")
+        return None, trace
+
+    # Agent 3 — Recommendation
+    ph3 = st.empty()
+    with ph3:
+        agent_card_running("Agent 3 — Recommendation Agent", "Ranking providers...")
+    t0 = time.time()
+    try:
+        rec = recommendation_agent.run(disc)
+        ms = int((time.time()-t0)*1000)
+        trace.append({"agent":"RecommendationAgent","step":3,
+                       "input":{"providers_found":disc["total_found"]},
+                       "output":{"selected":rec["best_provider"]["name"],"score":rec["best_provider"]["score"]},
+                       "duration_ms":ms, "reasoning":rec["reasoning"]})
+        ph3.empty()
+        agent_card_done("Agent 3 — Recommendation Agent", rec["reasoning"], ms)
+    except Exception as e:
+        ph3.empty()
+        agent_card_error("Agent 3 — Recommendation Agent", str(e))
+        return None, trace
+
+    # Agent 4 — Booking
+    ph4 = st.empty()
+    with ph4:
+        agent_card_running("Agent 4 — Booking Agent", "Confirming your booking...")
+    t0 = time.time()
+    try:
+        bk = booking_agent.run(rec, intent)
+        ms = int((time.time()-t0)*1000)
+        trace.append({"agent":"BookingAgent","step":4,
+                       "input":{"provider":rec["best_provider"]["name"]},
+                       "output":{"booking_id":bk["booking_id"],"slot":bk["confirmed_slot"]},
+                       "duration_ms":ms,
+                       "reasoning":f"Slot {bk['confirmed_slot']} selected. Booking ID: {bk['booking_id']}. Reminder: {bk['reminder_time']}."})
+        ph4.empty()
+        agent_card_done(
+            "Agent 4 — Booking Agent",
+            f"Booking <b>{bk['booking_id']}</b> confirmed · Slot <b>{bk['confirmed_slot']}</b> · "
+            f"Reminder at <b>{bk['reminder_time']}</b>.",
+            ms
+        )
+    except Exception as e:
+        ph4.empty()
+        agent_card_error("Agent 4 — Booking Agent", str(e))
+        return None, trace
+
+    return {"intent":intent, "discovery":disc, "recommendation":rec, "booking":bk}, trace
+
+# ════════════════════════════════════════════════════════════════════
+# TRIGGER
+# ════════════════════════════════════════════════════════════════════
+if run_btn:
+    if not user_input.strip():
+        st.warning("Please type a service request first.")
+    else:
+        st.session_state["last_input"] = user_input.strip()
+        st.session_state.result = None
+        st.session_state.trace_log = None
+        result, trace = run_pipeline(user_input.strip())
+        st.session_state.result = result
+        st.session_state.trace_log = trace
+
+# ════════════════════════════════════════════════════════════════════
+# RESULTS
+# ════════════════════════════════════════════════════════════════════
+if st.session_state.result:
+    result = st.session_state.result
+    rec  = result["recommendation"]
+    bk   = result["booking"]
+    disc = result["discovery"]
+
+    st.markdown("---")
+    tab1, tab2, tab3 = st.tabs(["📋 Providers", "✅ Booking Receipt", "🔍 Agent Trace"])
+
+    with tab1:
+        st.markdown(f"**{disc['total_found']} provider(s) found** — ranked by distance · rating · availability")
+        for i, p in enumerate(rec["all_ranked"]):
+            is_best = i == 0
+            badge   = '<span class="badge-best">⭐ TOP PICK</span>' if is_best else ""
+            avail   = "🟢 Available" if p.get("available") else "🔴 Unavailable"
+            sw      = int(p["score"]*100)
+            st.markdown(f"""
+            <div class="provider-card {'best' if is_best else ''}">
+                <div class="provider-name">#{i+1} {p['name']}{badge}</div>
+                <div class="provider-meta">
+                    📍 {p['location']['area']} &nbsp;·&nbsp;
+                    📏 {p['distance_km']} km &nbsp;·&nbsp;
+                    ⭐ {p['rating']} ({p['total_reviews']} reviews) &nbsp;·&nbsp;
+                    {avail} &nbsp;·&nbsp;
+                    💰 {p.get('price_range','N/A')}
+                </div>
+                <div class="score-bar"><div class="score-fill" style="width:{sw}%"></div></div>
+                <div style="font-size:0.73rem;color:#a0aec0;margin-top:3px;">Score: {p['score']}</div>
+            </div>""", unsafe_allow_html=True)
+
+    with tab2:
+        st.markdown(f"""
+        <div class="receipt">
+            <div class="receipt-id">BOOKING ID: {bk['booking_id']}</div>
+            <div class="receipt-name">🎉 {bk['provider_name']}</div>
+            <div style="color:#48bb78;font-size:0.85rem;margin-bottom:16px;">Booking Confirmed ✓</div>
+            <div class="receipt-row"><span class="receipt-label">🔧 Service</span><span class="receipt-value">{bk['service']}</span></div>
+            <div class="receipt-row"><span class="receipt-label">📍 Location</span><span class="receipt-value">{bk['location']}</span></div>
+            <div class="receipt-row"><span class="receipt-label">📅 Date</span><span class="receipt-value">{bk['booked_date']}</span></div>
+            <div class="receipt-row"><span class="receipt-label">⏰ Slot</span><span class="receipt-value">{bk['confirmed_slot']}</span></div>
+            <div class="receipt-row"><span class="receipt-label">📞 Contact</span><span class="receipt-value">{bk['provider_phone']}</span></div>
+            <div class="reminder-box">🔔 <b>Reminder</b> set for <b>{bk['reminder_time']}</b> — 1 hour before appointment</div>
+        </div>""", unsafe_allow_html=True)
+
+    with tab3:
+        st.markdown('<div class="trace-header">Full Agent Decision Log</div>', unsafe_allow_html=True)
+        for step in st.session_state.trace_log:
+            with st.expander(f"Step {step['step']} — {step['agent']}  ({step['duration_ms']}ms)"):
+                st.markdown(f"**Reasoning:** {step['reasoning']}")
+                ca, cb = st.columns(2)
+                with ca:
+                    st.markdown("**Input**"); st.json(step["input"])
+                with cb:
+                    st.markdown("**Output**"); st.json(step["output"])
+        st.markdown("**Full JSON:**")
+        st.json(st.session_state.trace_log)
+
+# ════════════════════════════════════════════════════════════════════
+# FOOTER
+# ═══════════════════════════════════════��════════════════════════════
+st.markdown("---")
+st.markdown(
+    "<div style='text-align:center;color:#a0aec0;font-size:0.78rem'>"
+    "KhidmatAI · Google Antigravity · Gemini API · Streamlit · Hackathon 2025"
+    "</div>", unsafe_allow_html=True
+)