reflect code changes

app.py

@@ -12,15 +12,15 @@ from smolagents import (
     DuckDuckGoSearchTool,
     VisitWebpageTool
 )
-from api_call_counter import APICallCounter
+from rate_limit import RateLimiter
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-call_counter = APICallCounter()
-litellm.success_callback = [call_counter.count_successful_call]
-litellm.failure_callback = [call_counter.count_failed_call]
+rate_limiter = RateLimiter()
+litellm.success_callback = [rate_limiter.increment_successful_calls]
+litellm.failure_callback = [rate_limiter.increment_failed_calls]
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -109,10 +109,6 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             submitted_answer = agent(question_text)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
-
-            if call_counter.get_total_calls() >= 10:
-                time.sleep(60)
-                call_counter.reset()
         except Exception as e:
              print(f"Error running agent on task {task_id}: {e}")
              results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})