Update app.py

app.py

@@ -5,32 +5,25 @@ import pandas as pd
 import time
 import traceback
 
-from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, VisitWebpageTool
+from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, VisitWebpageTool
 
 # --- Constantes ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- HACK DE RATE LIMIT: Modelo Throttled ---
-# Sobrescrevemos a classe base para injetar uma pausa forçada ANTES de cada requisição
-class ThrottledGemini(LiteLLMModel):
-    def generate(self, *args, **kwargs):
-        print("⏳ [Rate Limit Hack] Pausando 15s antes da chamada da API para respeitar os 5 RPM do Gemini 2.5...")
-        time.sleep(15)
-        return super().generate(*args, **kwargs)
-
-# --- Definição do Agente com Gemini ---
+# --- Definição do Agente com OpenAI ---
 class BasicAgent:
     def __init__(self):
-        print("Inicializando CodeAgent de produção com Gemini 2.5...")
+        print("Inicializando CodeAgent de produção com OpenAI (GPT-4o)...")
         
-        gemini_key = os.environ.get("GEMINI_API_KEY")
-        if not gemini_key:
-            raise ValueError("⚠️ GEMINI_API_KEY não encontrada! Verifique os Secrets do Space.")
+        openai_key = os.environ.get("OPENAI_API_KEY")
+        if not openai_key:
+            raise ValueError("⚠️ OPENAI_API_KEY não encontrada! Verifique os Secrets do Space.")
         
-        # Usamos nossa classe customizada que faz as pausas automáticas!
-        self.model = ThrottledGemini(
-            model_id="gemini/gemini-2.5-flash",
-            api_key=gemini_key
+        # Inicializa o modelo GPT-4o
+        # Nota: Se quiser economizar créditos, você pode trocar "gpt-4o" por "gpt-4o-mini"
+        self.model = OpenAIServerModel(
+            model_id="gpt-4o",
+            api_key=openai_key
         )
         
         self.agent = CodeAgent(
@@ -39,7 +32,7 @@ class BasicAgent:
             add_base_tools=True,
             max_steps=12 
         )
-        print("Agente inicializado com sucesso usando Gemini 2.5 Flash.")
+        print("Agente inicializado com sucesso usando OpenAI.")
 
     def __call__(self, question: str) -> str:
         print(f"Agente recebeu a pergunta (primeiros 50 chars): {question[:50]}...")
@@ -142,7 +135,8 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
              traceback.print_exc() 
              results_log.append({"Task ID": task_id, "Question": question_text[:120] + "...", "Submitted Answer": f"AGENT ERROR: {e}"})
 
-        print("Pausando por 10 segundos extras entre as perguntas...")
+        # Uma pausa leve (10s) apenas por boas práticas de rede
+        print("Pausando brevemente entre as perguntas...")
         time.sleep(10)
 
     if not answers_payload:
@@ -168,7 +162,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         return status_message, pd.DataFrame(results_log)
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Runner de Avaliação do Agente (Powered by Gemini 2.5 Flash)")
+    gr.Markdown("# Runner de Avaliação do Agente (Powered by GPT-4o)")
     gr.Markdown(
         """
         **Instruções:**