Update app.py

app.py

@@ -1,251 +1,42 @@
-from shiny import App, ui, render, reactive
-import json
-import datetime as dt
+from fastapi import FastAPI
+from fastapi.responses import HTMLResponse, JSONResponse
+from pydantic import BaseModel
+from typing import Dict
+import time
+
+app = FastAPI(title="Tessai LLM Bridge", version="0.1.0")
 
 # -----------------------------
-# UI
+# Models
 # -----------------------------
 
-app_ui = ui.page_fluid(
-    ui.h2("Tessai Link: Phase 0"),
-    ui.p(
-        "This form helps us understand your current environment and what you "
-        "are hoping to change. Fill out what you can -- we will review the rest together."
-    ),
-
-    ui.layout_columns(
-        # Left column: basic company + contacts
-        ui.card(
-            ui.card_header("1. Company & Contact"),
-            ui.input_text("company_name", "Company name"),
-            ui.input_text("contact_name", "Primary contact name"),
-            ui.input_text("contact_role", "Contact role / title"),
-            ui.input_text("contact_email", "Contact email"),
-            ui.input_text("contact_phone", "Contact phone (optional)"),
-        ),
-
-        # Right column: project trigger + goals
-        ui.card(
-            ui.card_header("2. Why now?"),
-            ui.input_text_area(
-                "project_trigger",
-                "What triggered this project now?",
-                placeholder="Example: Our current system is slowing us down, we are growing, a new compliance requirement, etc.",
-                rows=4,
-            ),
-            ui.input_text_area(
-                "primary_outcomes",
-                "What outcomes would feel like a clear win?",
-                placeholder="Example: Fewer dropped handoffs, clearer ownership, faster quoting, better visibility across teams...",
-                rows=4,
-            ),
-        ),
-    ),
 
-    ui.layout_columns(
-        # Current tools / systems / data
-        ui.card(
-            ui.card_header("3. Current tools & data"),
-            ui.input_text_area(
-                "current_tools",
-                "Current tools / systems in use",
-                placeholder="Example: Excel, email, QuickBooks, Odoo, custom access DB, paper forms...",
-                rows=4,
-            ),
-            ui.input_text_area(
-                "data_sources",
-                "Key data sources we should know about",
-                placeholder="Example: Shared drives, vendor portals, ERP, CRM, PDF drawings, email inboxes...",
-                rows=4,
-            ),
-        ),
+class ChatRequest(BaseModel):
+    session_id: str
+    message: str
+    context: Dict[str, str] | None = None
 
-        # Scope, change profile, constraints
-        ui.card(
-            ui.card_header("4. Scope & constraints"),
-            ui.input_selectize(
-                "areas_of_interest",
-                "Primary areas you want to improve (choose all that apply)",
-                choices=[
-                    "Lead / opportunity tracking",
-                    "Quoting / estimating",
-                    "Operations / workflows",
-                    "Vendor integration",
-                    "Reporting / dashboards",
-                    "Document / email intake",
-                    "Change management / approvals",
-                    "Other",
-                ],
-                multiple=True,
-            ),
-            ui.input_radio_buttons(
-                "change_profile",
-                "How would you describe this change?",
-                choices=[
-                    "Exploring options -- nothing urgent yet",
-                    "We have a clearly broken process to fix",
-                    "We are replacing or consolidating existing systems",
-                    "We have a hard deadline / compliance date",
-                ],
-            ),
-            ui.input_select(
-                "team_size",
-                "Roughly how many people are affected?",
-                choices=[
-                    "1–5",
-                    "6–15",
-                    "16–50",
-                    "51–150",
-                    "150+",
-                ],
-            ),
-            ui.input_select(
-                "timeline",
-                "Ideal timeline to see first meaningful results",
-                choices=[
-                    "0–6 weeks",
-                    "6–12 weeks",
-                    "3–6 months",
-                    "6–12 months",
-                    "Just exploring for now",
-                ],
-            ),
-        ),
-    ),
 
-    ui.layout_columns(
-        ui.card(
-            ui.card_header("5. Risks, blockers, and notes"),
-            ui.input_text_area(
-                "risks_blockers",
-                "Known risks or blockers",
-                placeholder="Example: Limited internal capacity, key person risk, uncertain budget, complex approvals...",
-                rows=3,
-            ),
-            ui.input_text_area(
-                "other_notes",
-                "Anything else we should know?",
-                placeholder="Free space for context, history, politics, or 'things that have been tried before'.",
-                rows=3,
-            ),
-        ),
-    ),
+class ChatResponse(BaseModel):
+    session_id: str
+    reply: str
+    tokens_used: int
 
-    ui.hr(),
-    ui.input_action_button("submit", "Submit intake", class_="btn-primary"),
-
-    ui.hr(),
-    ui.layout_columns(
-        ui.card(
-            ui.card_header("Intake summary (human-readable)"),
-            ui.output_text_verbatim("intake_summary"),
-        ),
-        ui.card(
-            ui.card_header("Raw payload (for systems)"),
-            ui.output_text_verbatim("intake_payload"),
-        ),
-    ),
-)
 
 # -----------------------------
-# Server
+# Simple in-memory metrics
 # -----------------------------
 
+metrics = {
+    "total_requests": 0,
+    "total_tokens": 0,
+    "sessions": {}  # session_id -> {"last_seen": float, "requests": int, "tokens": int}
+}
 
-def server(input, output, session):
-    # Build a structured dict from the current inputs
-    def build_payload():
-        return {
-            "submitted_at": dt.datetime.now().isoformat(timespec="seconds"),
-            "company": {
-                "name": input.company_name(),
-                "contact_name": input.contact_name(),
-                "contact_role": input.contact_role(),
-                "contact_email": input.contact_email(),
-                "contact_phone": input.contact_phone(),
-            },
-            "project": {
-                "trigger": input.project_trigger(),
-                "primary_outcomes": input.primary_outcomes(),
-            },
-            "environment": {
-                "current_tools": input.current_tools(),
-                "data_sources": input.data_sources(),
-            },
-            "scope": {
-                "areas_of_interest": input.areas_of_interest(),
-                "change_profile": input.change_profile(),
-                "team_size": input.team_size(),
-                "timeline": input.timeline(),
-            },
-            "risks_and_notes": {
-                "risks_blockers": input.risks_blockers(),
-                "other_notes": input.other_notes(),
-            },
-        }
-
-    # Reactive value that updates only when "Submit intake" is clicked
-    @reactive.calc
-    def submitted_payload():
-        # Wait for the button click event
-        input.submit()
-        return build_payload()
-
-    @output
-    @render.text
-    def intake_summary():
-        d = submitted_payload()
-
-        company = d["company"]
-        project = d["project"]
-        env = d["environment"]
-        scope = d["scope"]
-        risks = d["risks_and_notes"]
-
-        areas = scope["areas_of_interest"] or []
-        areas_str = ", ".join(areas) if areas else "Not specified"
-
-        lines = [
-            f"Submitted at: {d['submitted_at']}",
-            "",
-            f"Company: {company['name'] or 'Unknown'}",
-            f"Contact: {company['contact_name'] or 'Unknown'} "
-            f"({company['contact_role'] or 'role not specified'})",
-            f"Email: {company['contact_email'] or 'not provided'}",
-            f"Phone: {company['contact_phone'] or 'not provided'}",
-            "",
-            "Why now:",
-            project["trigger"] or "(no description provided)",
-            "",
-            "Primary outcomes:",
-            project["primary_outcomes"] or "(no outcomes described yet)",
-            "",
-            "Current tools / systems:",
-            env["current_tools"] or "(not listed)",
-            "",
-            "Key data sources:",
-            env["data_sources"] or "(not listed)",
-            "",
-            "Scope snapshot:",
-            f"  Areas of interest: {areas_str}",
-            f"  Change profile: {scope['change_profile'] or 'Not specified'}",
-            f"  Team size affected: {scope['team_size'] or 'Not specified'}",
-            f"  Timeline: {scope['timeline'] or 'Not specified'}",
-            "",
-            "Risks / blockers:",
-            risks["risks_blockers"] or "(none listed)",
-            "",
-            "Other notes:",
-            risks["other_notes"] or "(none)",
-        ]
-
-        return "\n".join(lines)
-
-    @output
-    @render.text
-    def intake_payload():
-        d = submitted_payload()
-        return json.dumps(d, indent=2)
 
+def record_request(session_id: str, tokens_used: int) -> None:
+    now = time.time()
+    metrics["total_requests"] += 1
+    metrics["total_tokens"] += tokens_used
 
-app = App(app_ui, server)
+    if session_id not in metr_