Update app.py

app.py

@@ -19,9 +19,12 @@ from huggingface_hub import login
 HF_API_KEY = os.getenv('HF_API_KEY')
 login(HF_API_KEY)
 
+# Set API key for OPENAI
+os.environ['OPENAI_API_KEY"'] = os.getenv('OPENAI_API_KEY')
+
 # Create model
-# model = OpenAIServerModel(model_id="gpt-4o")
-model = HfApiModel()
+model = OpenAIServerModel(model_id="gpt-4o")
+# model = HfApiModel()
 
 # Create final answer instance
 final_answer = SimplifyAnswerTool(model)
@@ -68,12 +71,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     try:
         agent = CodeAgent(model=model,
                           tools=[DuckDuckGoSearchTool(), VisitWebpageTool(), final_answer], 
-                          max_steps=6,
-                          verbosity_level=1,
-                          grammar=None,
-                          planning_interval=None,
-                          name=None,
-                          description=None,
+                          max_steps=20,
+                          verbosity_level=2,
                           prompt_templates=prompt_templates)
     except Exception as e:
         print(f"Error instantiating agent: {e}")