Update app.py

app.py

@@ -46,9 +46,21 @@ async def chat(request: ChatRequest):
             return {"response": response.choices[0].message.content}
 
         if request.model_choice == "google":
-            client = genai.Client(api_key=google_api_key)
-    
-            response = client.models.generate_content(model="gemini-2.0-flash", contents=request.message)
+            messages = [
+                {"role": "user", "parts": [{"text": "I have 10 dogs"}]},
+                {"role": "model", "parts": [{"text": "Great! Dogs are fun pets."}]},
+                {"role": "user", "parts": [{"text": "How many dogs do I have?"}]},
+            ]
+            
+            response = client.models.generate_content(
+                model="gemini-2.0-flash",
+                contents=messages,
+                config=GenerateContentConfig(
+                    system_instruction=[
+                        "Respond like you are a pirate.",
+                    ]
+                ),
+            )
             return {"response": response.text}
             
     except Exception as e: