Update app.py

app.py

@@ -51,20 +51,18 @@ async def chat(request: ChatRequest):
             return {"response": response.choices[0].message.content}
 
         if request.model_choice == "google":
-            genai.configure(api_key=google_api_key)
-            model = genai.GenerativeModel("gemini-2.0-flash")
+            client = genai.Client(api_key=google_api_key)
     
             messages = [
                 {"role": "system", "parts": [request.system_message]},
                 {"role": "user", "parts": [request.message]},
             ]
     
-            response = model.generate_content(messages)
-    
+           response = client.models.generate_content(
+                model="gemini-2.0-flash", contents=request.message
+            )
             if response and hasattr(response, 'text'):
-                return response.text
-            else:
-                return "No response text received from the model."
+                return response.text  # Return text if it's present
             
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))