from fastapi import FastAPI, Request
from fastapi.responses import HTMLResponse, StreamingResponse, FileResponse
from google.adk.runners import Runner
from google.adk.sessions import InMemorySessionService
from google.genai.types import Content, Part
from dotenv import load_dotenv
import asyncio
# --- Configuration ---
load_dotenv()
from aida.agent import root_agent # Import your agent
APP_NAME = "aida"
# --- ADK & FastAPI Setup ---
session_service = InMemorySessionService()
runner = Runner(app_name=APP_NAME, agent=root_agent, session_service=session_service)
app = FastAPI()
# --- HTML Frontend ---
HTML_CONTENT = """
AIDA Terminal
SYSTEM> AIDA Interface Loaded.
AIDA-8000
"""
# --- Routes ---
@app.get("/", response_class=HTMLResponse)
async def root():
return HTML_CONTENT
@app.get("/idle")
async def get_idle_image():
return FileResponse("static/assets/idle.png")
@app.get("/talk")
async def get_talk_image():
return FileResponse("static/assets/talk.png")
@app.post("/chat")
async def chat_endpoint(request: Request):
data = await request.json()
user_query = data.get("query")
# Hardcoded for demo simplicity
user_id = "demo_user"
session_id = "demo_session"
# Ensure session exists
if not await session_service.get_session(APP_NAME, user_id, session_id):
await session_service.create_session(APP_NAME, user_id, session_id)
async def response_stream():
"""Generates text chunks from the agent's events."""
async for event in runner.run_async(
user_id=user_id,
session_id=session_id,
new_message=Content(role="user", parts=[Part.from_text(text=user_query)]),
):
# We only want the final text response for this simple UI
if event.is_final_response() and event.content and event.content.parts:
for part in event.content.parts:
if hasattr(part, "text") and part.text:
yield part.text
return StreamingResponse(response_stream(), media_type="text/plain")