Update app.py

app.py

@@ -6,6 +6,8 @@ import pandas as pd
 from smolagents import OpenAIServerModel, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool
 from smolagents.tools import Tool
 import time
+import openai
+from tenacity import retry, stop_after_attempt, wait_exponential, retry_if_exception_type
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -13,6 +15,26 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+
+class RateLimitException(Exception):
+    pass
+
+@retry(
+    reraise=True,
+    stop=stop_after_attempt(10),  
+    wait=wait_exponential(multiplier=20, min=20, max=120),  
+    retry=retry_if_exception_type(RateLimitException),  
+)
+def call_openai_with_retry(**kwargs):
+    try:
+        return openai.ChatCompletion.create(**kwargs)
+    except openai.error.RateLimitError as e:
+        print("Rate limit error detected. Will retry...")
+        raise RateLimitException from e  
+    except Exception as e:
+        print(f"Non-rate-limit error occurred: {e}")
+        raise  
+
 class BasicAgent:
     def __init__(self):
         self.agent = CodeAgent(
@@ -24,13 +46,25 @@ class BasicAgent:
             add_base_tools=True,
         )
         print("BasicAgent initialized.")
+        
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = self.agent.run(question)
+        try:
+            fixed_answer = call_openai_with_retry(
+                model="gpt-4o",
+                messages=[
+                    {"role": "user", "content": question}
+                ],
+                temperature=0.2,
+            )['choices'][0]['message']['content']
+        except Exception as e:
+            print(f"Agent error: {e}")
+            raise
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer
 
 
+
 def safe_agent_call(agent, question, retries=5, wait_time=20):
     """
     Helper function to safely call the agent with retry on rate limit errors (HTTP 429).