Update app.py

app.py

@@ -22,12 +22,7 @@ class ChatRequest(BaseModel):
     top_p: float = 0.95
     model_choice: str = "HF"
 
-class ChatResponse(BaseModel):
-    response: str
-
-prompt_template = f""""""
-
-@app.post("/chat", response_model=ChatResponse)
+@app.post("/chat")
 async def chat(request: ChatRequest):
     try:
         if request.model_choice == "HF":
@@ -53,13 +48,8 @@ async def chat(request: ChatRequest):
         if request.model_choice == "google":
             client = genai.Client(api_key=google_api_key)
     
-            messages = [
-                {"role": "system", "parts": [request.system_message]},
-                {"role": "user", "parts": [request.message]},
-            ]
-    
             response = client.models.generate_content(model="gemini-2.0-flash", contents=request.message)
-            return response.text
+            return {"response": google_api_key}
             
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))