Update app.py

app.py

@@ -75,15 +75,15 @@ async def generate(input_text: str):
         yield "🔍 Thinking...", ""
 
         config = types.GenerateContentConfig(
-            tools=[train_tool],
+            tools=[train_tool, types.Tool(google_search=types.GoogleSearch())],
             temperature=0.3,
         )
 
-        # Create async chat session (must await)
-        chat = await client.aio.chats.create(model=model_id, config=config)
+        # Create async chat session (DO NOT await here)
+        chat = client.aio.chats.create(model=model_id, config=config)
 
-        # Send the user input to the model
-        response = await chat.send_message(input_text)
+        # Send the user input to the model (await is correct here)
+        response =  chat.send_message(input_text)
 
         # --- Manual Tool Loop ---
         max_turns = 5
@@ -97,7 +97,7 @@ async def generate(input_text: str):
             if not parts:
                 break
 
-            # Collect function calls from parts
+            # Collect function calls from parts (correct attribute is function_call)
             tool_calls = []
             for p in parts:
                 call = getattr(p, "function_call", None)
@@ -140,7 +140,7 @@ async def generate(input_text: str):
             if tool_responses:
                 yield "📝 Finalizing response...", ""
                 # Provide tool results back to the model so it can compose the final answer
-                response = await chat.send_message(tool_responses)
+                response =  chat.send_message(tool_responses)
             else:
                 break