Update app.py

app.py

@@ -3,6 +3,8 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+import re
+import time
 #from tavily import TavilyClient
 from langchain_tavily import TavilySearch
 from langgraph.prebuilt import create_react_agent
@@ -33,11 +35,22 @@ GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 ##HUGGINGFACEHUB_API_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 
 #PROMPTS
+#prompt_recomendado = """You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template:
+#                        FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
+#                        If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
+#                        If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
+#                        If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."""
+
 prompt_recomendado = """You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template:
                         FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
                         If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
                         If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
-                        If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."""
+                        If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+                        To assist in your task, you can supervise other agents who perform specific tasks that could not be handled by tools, since they require the processing of another LLM. Below, I will inform you about your assistants:
+                        - web_research_agent. Assign web research related tasks to this agent
+                        Assign work to one agent at a time, do not call agents in parallel.
+                        Priorize the use of tools and another agents to help in reasoning.
+                        When a file or URL is entered at the prompt, use it in tools or other agents, both are prepared to handle files and URLs."""
 
 prompt_search = """You are a web research agent.
                    INSTRUCTIONS:
@@ -113,21 +126,22 @@ class BasicAgent:
                         ]
                     }
 
-        events = supervisor.stream(
-            messages,
-            stream_mode="values",
-        )
+        #events = supervisor.stream(
+        #    messages,
+        #    stream_mode="values",
+        #)
         
-        for event in events:
-            messages.extend(event["messages"])
+        #for event in events:
+        #    messages.extend(event["messages"])
 
-        answer = messages[-1].content
+        #answer = messages[-1].content
 
         #print(f"Answer: {answer}...")
         
-        #answer = supervisor.invoke(messages)
-        print(f"Agent returning answer: {answer}")
-        return answer
+        answer = supervisor.invoke(messages)
+        final_answer = re.search(r"FINAL ANSWER:\s*(.*)", answer['output'])
+        print(f"Agent returning answer: {final_answer}")
+        return final_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -193,6 +207,7 @@ 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})
+            time.sleep(60)
         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}"})