Implement question counter in BasicAgent and modify response logic in app.py. Update main function in main.py to streamline task execution and improve logging. Prepare for potential rate limiting with a placeholder for wait time.

app.py

@@ -1,9 +1,15 @@
 import os
+import time
 import gradio as gr
 import requests
 import inspect
 import pandas as pd
 
+from main import main
+
+global question_counter
+question_counter = 0
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -15,9 +21,29 @@ class BasicAgent:
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+
+        global question_counter
+        question_counter += 1
+
+        if question_counter > 1:
+            return "This is a default answer."
+
+        # fixed_answer = "This is a default answer."
+        # print(f"Agent returning fixed answer: {fixed_answer}")
+        # return fixed_answer
+
+        # TODO: Set a wait time between each question to avoid rate limiting
+        # print("Sleeping for 60 seconds...")
+        # time.sleep(60)
+
+        final_answer = str(main(task=question))
+
+        print("--------------------------------")
+        print(f"Question: {question}")
+        print(f"Final Answer: {final_answer}")
+        print("--------------------------------")
+
+        return final_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """

main.py

@@ -1,3 +1,4 @@
+import asyncio
 import logging
 import os
 import uuid  # for generating thread IDs for checkpointer
@@ -231,17 +232,11 @@ async def run_with_streaming(task: str, thread_id: str) -> dict:
         return {"steps": steps, "final_answer": final_answer_text, "error": error_msg}
 
 
-if __name__ == "__main__":
-    import asyncio
-    import uuid
-
-    # Example Usage
-    task_to_run = "What is the capital of France?"
-    thread_id = str(uuid.uuid4())  # Generate a unique thread ID for this run
+def main(task: str, thread_id: str = str(uuid.uuid4())):
     logger.info(
-        f"Starting agent run from __main__ for task: '{task_to_run}' with thread_id: {thread_id}"
+        f"Starting agent run from __main__ for task: '{task}' with thread_id: {thread_id}"
     )
-    result = asyncio.run(run_with_streaming(task_to_run, thread_id))
+    result = asyncio.run(run_with_streaming(task, thread_id))
     logger.info("Agent run finished.")
 
     # Print final results
@@ -253,3 +248,14 @@ if __name__ == "__main__":
     print(f"Final Answer: {result.get('final_answer') or 'Not found'}")
     if err := result.get("error"):
         print(f"Error: {err}")
+
+    return result.get("final_answer")
+
+
+if __name__ == "__main__":
+    # Example Usage
+    task_to_run = "What is the capital of France?"
+    thread_id = str(uuid.uuid4())  # Generate a unique thread ID for this run
+
+    final_answer = main(task_to_run, thread_id)
+    print(f"Final Answer: {final_answer}")