Update app.py

app.py

@@ -11,7 +11,8 @@ import json
 import boto3
 from smolagents import FinalAnswerTool, Tool, tool, OpenAIServerModel, DuckDuckGoSearchTool, CodeAgent, VisitWebpageTool
 from nova_agent import NovaProAgent
-from gemini_agent import GeminiAgent
+
+import google.generativeai as genai
 
 from dotenv import load_dotenv
 
@@ -91,11 +92,10 @@ class SlpMultiAgent:
         short_question = question  # [:MAX_QUESTION_LENGTH]
         
         # Use cheaper, faster model
-        model = OpenAIServerModel(
-            model_id="gpt-3.5-turbo", 
-            temperature=0.0,  # Deterministic for consistency
-            max_tokens=400    # Reduced tokens for cost efficiency
-        )
+        api_key = os.getenv('GOOGLE_API_KEY')
+        genai.configure(api_key=api_key)
+        
+        model = genai.GenerativeModel('gemini-2.0-flash-exp')
         
         # Create only essential agents with reduced complexity
         research_agent = CodeAgent(
@@ -229,7 +229,7 @@ async def run_and_submit_all(profile):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = GeminiAgent()
+        agent = SlpMultiAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None