async changes

app.py

@@ -6,10 +6,8 @@ os.environ['HF_HOME'] = '/tmp/huggingface_cache'
 os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
 os.environ['HF_DATASETS_CACHE'] = '/tmp/datasets_cache'
 
-# Suppress warnings
 warnings.filterwarnings("ignore")
 
-# Now import everything else
 import json
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
@@ -19,6 +17,7 @@ from google.adk.sessions import InMemorySessionService
 from google.adk.runners import Runner
 from google.genai import types
 import re
+import asyncio
 from tools import (
     db_tool,
     tavily_tool,
@@ -163,6 +162,7 @@ INSTITUTE_MAPPING = {
     "gkv": ["gurukula kangri vishwavidyalaya"],
     "GKV": ["gurukula kangri vishwavidyalaya"]
 }
+
 BRANCH_MAPPING = {
     "cse": "computer science and engineering",
     "CSE": "computer science and engineering",
@@ -178,8 +178,6 @@ BRANCH_MAPPING = {
     "CE": "civil engineering",
     "che": "chemical engineering",
     "CHE": "chemical engineering",
-    # "it": "information technology",
-    # "IT": "information technology",
     "aero": "aerospace engineering",
     "AERO": "aerospace engineering",
     "bio": "biotechnology",
@@ -198,8 +196,9 @@ def preprocess_query(query: str) -> str:
         query = re.sub(pattern, full_name, query, flags=re.IGNORECASE)
 
     return query
+
 # === AGENT SETUP ===
-def create_agent_runner(user_id: str, session_id: str):
+async def create_agent_runner(user_id: str, session_id: str):
     instruction_text="""
 You are a highly experienced college counselor specializing in helping high school students choose the right engineering colleges. You have access to several tools to help answer student queries.
 
@@ -270,12 +269,36 @@ IMPORTANT RULES:
             temperature=0.1,
         ),
     )
+    
     session_svc = InMemorySessionService()
-    session = session_svc.create_session(app_name="college_agent_app", user_id=user_id, session_id=session_id)
+    
+    # Handle potential async session creation
+    try:
+        session = session_svc.create_session(
+            app_name="college_agent_app", 
+            user_id=user_id, 
+            session_id=session_id
+        )
+        # If it's a coroutine, await it
+        if asyncio.iscoroutine(session):
+            session = await session
+    except Exception as e:
+        print(f"Session creation error: {e}")
+        # Fallback: create a simple session object
+        class SimpleSession:
+            def __init__(self, session_id):
+                self.id = session_id
+        session = SimpleSession(session_id)
+    
     runner = Runner(agent=agent, app_name="college_agent_app", session_service=session_svc)
     return runner, session
+
 # === FASTAPI SETUP ===
-app = FastAPI()
+app = FastAPI(
+    title="College Counselor Agent",
+    description="AI Agent for college counseling and selection",
+    version="1.0.0"
+)
 
 class ChatRequest(BaseModel):
     user_id: str
@@ -285,42 +308,88 @@ class ChatRequest(BaseModel):
 class ChatResponse(BaseModel):
     session_id: str
     answer: str
+
+@app.get("/")
+async def root():
+    return {
+        "message": "College Counselor Agent is running!",
+        "docs": "/docs",
+        "health": "/healthz"
+    }
+
 @app.post("/chat", response_model=ChatResponse)
 async def chat_endpoint(req: ChatRequest):
     try:
-        runner, session = create_agent_runner(req.user_id, req.session_id)
+        runner, session = await create_agent_runner(req.user_id, req.session_id)
         processed_query = preprocess_query(req.question)
-        # processed_query = req.question
+        
         user_msg = types.Content(role="user", parts=[types.Part(text=processed_query)])
         reply_text = ""
 
-        events = runner.run(user_id=req.user_id, session_id=session.id, new_message=user_msg)
+        # Handle potential async runner execution
+        try:
+            events = runner.run(user_id=req.user_id, session_id=session.id, new_message=user_msg)
+            
+            # If events is a coroutine, await it
+            if asyncio.iscoroutine(events):
+                events = await events
+        except Exception as runner_error:
+            print(f"Runner error: {runner_error}")
+            return ChatResponse(
+                session_id=session.id, 
+                answer="Sorry, I encountered an error processing your request. Please try again."
+            )
 
-        for ev in events:
-            if ev.is_final_response():
-                try:
-                    for part in ev.content.parts:
-                        if part.text:
-                            reply_text = part.text
-                            break
-                except Exception:
+        # Process events
+        try:
+            for ev in events:
+                if hasattr(ev, 'is_final_response') and ev.is_final_response():
                     try:
-                        reply_text = ev.text
+                        for part in ev.content.parts:
+                            if hasattr(part, 'text') and part.text:
+                                reply_text = part.text
+                                break
                     except Exception:
                         try:
-                            reply_text = str(ev.message)
+                            reply_text = getattr(ev, 'text', '')
                         except Exception:
-                            reply_text = "Sorry, I couldn't understand the response."
-                break
+                            try:
+                                reply_text = str(getattr(ev, 'message', ''))
+                            except Exception:
+                                reply_text = "Sorry, I couldn't process the response properly."
+                    break
+        except Exception as event_error:
+            print(f"Event processing error: {event_error}")
+            reply_text = "Sorry, I encountered an error while processing the response."
+
+        # Fallback if no reply_text
+        if not reply_text:
+            reply_text = "I'm sorry, I couldn't generate a proper response. Please try rephrasing your question."
+
+        # Log to sheet (handle potential errors)
+        try:
+            add_query_to_sheet(req.user_id, processed_query, reply_text)
+        except Exception as sheet_error:
+            print(f"Sheet logging error: {sheet_error}")
+            # Don't fail the request if logging fails
 
-        add_query_to_sheet(req.user_id, processed_query, reply_text)
         return ChatResponse(session_id=session.id, answer=reply_text)
 
     except Exception as e:
+        print(f"Chat endpoint error: {str(e)}")
         raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
+
 @app.get("/healthz")
 def health_check():
-    return {"status": "ok"}
-@app.get("/")
-async def root():
-    return {"message": "Hello world"}
+    return {"status": "ok", "service": "college-counselor-agent"}
+
+# Add a simple test endpoint
+@app.get("/test")
+async def test_endpoint():
+    return {
+        "message": "Test endpoint working",
+        "env_vars": {
+            "HF_HOME": os.environ.get('HF_HOME'),
+            "TRANSFORMERS_CACHE": os.environ.get('TRANSFORMERS_CACHE')
+        }
+    }