add logic to wait 60s after calling API 10 times

app.py

@@ -1,14 +1,21 @@
 import os
+import time
 import gradio as gr
+import litellm
 import requests
 import inspect
 import pandas as pd
 from smolagents import CodeAgent, LiteLLMModel
+from api_call_counter import APICallCounter
 
 # (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]
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
@@ -89,6 +96,10 @@ 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_count() >= 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}"})