Update app.py

app.py

@@ -18,14 +18,30 @@ class ChatRequest(BaseModel):
     emphasizes deep scouting, strategic access, and minimalist setups. Through The Hunting Beast, you teach hunters how to kill big bucks 
     using terrain, wind, and thermals. You speak from firsthand experience, keeping your advice practical and to the point. Provide detailed 
     yet concise responses, with a maximum of 150 words"""
-    max_tokens: int = 512
     temperature: float = 0.7
-    top_p: float = 0.95
-    model_choice: str = "HF"
+    model_choice: str = "google"
 
 @app.post("/chat")
 async def chat(request: ChatRequest):
     try:
+        if request.model_choice == "google":
+            client = genai.Client(api_key=google_api_key)
+
+            messages = [
+                {"role": "user", "parts": [{"text": request.message}]},
+                # {"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=[system_message]
+                ),
+            )
+            return {"response": response.text}
+
         if request.model_choice == "HF":
             if hf_token:
                 client = InferenceClient("meta-llama/Llama-3.2-3B-Instruct", token=hf_token)
@@ -45,26 +61,6 @@ 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)
-
-            messages = [
-                {"role": "user", "parts": [{"text": request.message}]},
-                # {"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:
         raise HTTPException(status_code=500, detail=str(e))