feat: Add Gradio UI for interactive query and expiry sweep agents, integrated with FastAPI and uvicorn.

app.py

@@ -1,24 +1,36 @@
-import sys
 import os
+import sys
 import asyncio
+from fastapi import FastAPI
+import gradio as gr
+import uvicorn
 
 # Add src to path so we can import the package
 sys.path.append(os.path.join(os.path.dirname(__file__), "src"))
 
 from credentialwatch_agent.main import demo, mcp_client
 
+# Create FastAPI app
+app = FastAPI()
+
+@app.on_event("startup")
+async def startup_event():
+    """Connect to MCP servers on startup."""
+    print("Connecting to MCP servers...")
+    # We can't use run_until_complete here because the loop is already running
+    await mcp_client.connect()
+
+@app.on_event("shutdown")
+async def shutdown_event():
+    """Close connections on shutdown."""
+    print("Closing MCP connections...")
+    await mcp_client.close()
+
+# Mount Gradio app
+# path="/" mounts it at the root
+app = gr.mount_gradio_app(app, demo, path="/")
+
 if __name__ == "__main__":
-    # Simple async wrapper to run the app
-    loop = asyncio.new_event_loop()
-    asyncio.set_event_loop(loop)
-    
-    try:
-        print("Connecting to MCP servers...")
-        loop.run_until_complete(mcp_client.connect())
-        print("Starting Gradio app...")
-        demo.launch()
-    except KeyboardInterrupt:
-        pass
-    finally:
-        print("Closing MCP connections...")
-        loop.run_until_complete(mcp_client.close())
+    # Run with uvicorn
+    # This ensures everything runs on the same async loop
+    uvicorn.run(app, host="0.0.0.0", port=7860)

pyproject.toml

@@ -11,7 +11,9 @@ dependencies = [
     "mcp>=0.1.0",
     "gradio[mcp]>=6.0.1",
     "python-dotenv>=1.0.0",
-    "httpx>=0.25.0" 
+    "httpx>=0.25.0",
+    "fastapi>=0.100.0",
+    "uvicorn>=0.20.0"
 ]
 
 [build-system]

src/credentialwatch_agent/main.py

@@ -47,9 +47,15 @@ async def run_chat_turn(message: str, history: List[List[str]]) -> str:
     """
     # Convert history to LangChain format
     messages = []
-    for human, ai in history:
-        messages.append(HumanMessage(content=human))
-        messages.append(AIMessage(content=ai))
+    for item in history:
+        if isinstance(item, (list, tuple)) and len(item) >= 2:
+            human = item[0]
+            ai = item[1]
+            messages.append(HumanMessage(content=str(human)))
+            messages.append(AIMessage(content=str(ai)))
+        else:
+            # Fallback for unexpected format
+            print(f"Warning: Skipping malformed history item: {item}")
     messages.append(HumanMessage(content=message))
     
     initial_state = {"messages": messages}