Update app.py

app.py

@@ -49,7 +49,6 @@ class BasicAgent:
         """Creates the master prompt, dynamically injecting a file-handling example if a URL is provided."""
         tool_docs = "\n".join([f"- {name}: {inspect.getdoc(func)}" for name, func in self.tools.items()])
 
-        # Default web search example
         web_search_example = """
 **Example: Web Search**
 Question: Who was the prime minister of the UK in 1999?
@@ -64,39 +63,27 @@ Observation: [Page content confirming Tony Blair was Prime Minister from 1997 to
 Thought: I have confirmed the answer from a reliable source.
 Final Answer: Tony Blair"""
 
-        # Dynamic file analysis example
         file_analysis_example = ""
         if file_url:
-            code_snippet = "# This is a placeholder, will be replaced by a specific file handler\n"
+            code_snippet = ""
             if file_url.endswith(('.xlsx', '.csv')):
                 code_snippet = f"""
 import pandas as pd
 import requests
 import io
-# The user's file is at this URL, which MUST be used.
 url = '{file_url}'
 response = requests.get(url)
 df = pd.read_excel(io.BytesIO(response.content))
-# Now, I must analyze the dataframe `df` to answer the question.
-# For example, to see the first few rows, I can print(df.head()).
-# To calculate total sales, I would use print(df['Sales'].sum()).
-print(df.head())
+print(df.to_string())
 """
             elif file_url.endswith('.py'):
                  code_snippet = f"""
 import requests
-# The user's Python code file is at this URL, which MUST be used.
 url = '{file_url}'
 response = requests.get(url)
-python_code = response.text
-# Now, I must execute this code to find the output.
-# I will use another python action to run the code.
-print("Code downloaded. Ready for execution in the next step.")
+python_code_to_run = response.text
+print(python_code_to_run)
 """
-            elif file_url.endswith(('.mp3', '.wav')):
-                # Handling audio is complex and often requires external APIs or heavy libraries.
-                # For this exam, we will state a clear limitation.
-                file_analysis_example = "\n**Limitation:** I cannot process audio files. Please inform the user of this limitation."
             
             if code_snippet:
                 file_analysis_example = f"""
@@ -112,10 +99,6 @@ Final Answer: [Answer based on the script's output]"""
 
         return f"""
 You are a helpful assistant that answers questions by thinking step-by-step and using the tools provided.
-
-You have access to the following tools:
-{tool_docs}
-
 **Process:**
 1.  **Thought:** Analyze the user's question and create a plan. If you see an example below that matches your plan, follow it exactly.
 2.  **Action:** Choose ONE tool from the list: {", ".join(self.tools.keys())}.
@@ -125,6 +108,9 @@ You have access to the following tools:
 6.  **Thought:** Conclude that you have the final answer.
 7.  **Final Answer:** Provide the final, direct answer to the user's question.
 
+You have access to the following tools:
+{tool_docs}
+
 {web_search_example}
 {file_analysis_example}
 
@@ -134,16 +120,13 @@ Begin!
     # --- Tool Definitions ---
     def search(self, query: str) -> str:
         """Searches the web with DuckDuckGo to find relevant URLs and information."""
-        print(f"Tool: search, Query: {query}")
         try:
             with DDGS() as ddgs:
-                results = [r for r in ddgs.text(query, max_results=4)]
-            return str(results) if results else "No results found."
+                return str([r for r in ddgs.text(query, max_results=4)])
         except Exception as e: return f"Error during search: {e}"
 
     def browse(self, url: str) -> str:
         """Gets the full, clean text content of a single webpage URL."""
-        print(f"Tool: browse, URL: {url}")
         try:
             response = requests.get(url, timeout=10, headers={'User-Agent': 'Mozilla/5.0'})
             soup = BeautifulSoup(response.content, 'html.parser')
@@ -152,7 +135,6 @@ Begin!
 
     def python(self, code: str) -> str:
         """Executes Python code to analyze data or files. Use `requests` to download files from URLs."""
-        print(f"Tool: python, Code:\n{code}")
         code = code.strip().strip("`").replace("python\n", "").strip()
         buffer = io.StringIO()
         try:
@@ -164,17 +146,14 @@ Begin!
 
     def youtube_transcript(self, url: str) -> str:
         """Fetches the full transcript of a YouTube video from its URL."""
-        print(f"Tool: youtube_transcript, URL: {url}")
         try:
             video_id = re.search(r"(?<=v=)[\w-]+", url).group(0)
-            transcript_list = YouTubeTranscriptApi.get_transcript(video_id)
-            return " ".join([item['text'] for item in transcript_list])
+            return " ".join([item['text'] for item in YouTubeTranscriptApi.get_transcript(video_id)])
         except Exception as e: return f"Error fetching transcript: {e}"
 
     # --- Main ReAct Loop ---
     def __call__(self, task: Dict[str, Any]) -> str:
         file_url = task.get("files", [None])[0]
-        # Dynamically create the prompt for each task
         system_prompt = self._create_system_prompt(file_url=file_url)
         
         question = task.get("question", "")
@@ -185,18 +164,18 @@ Begin!
         history = ""
 
         for i in range(8):
-            print(f"--- Step {i+1} ---")
             full_prompt = prompt + history
             llm_response = self.llm.invoke(full_prompt).content.strip()
             history += f"\n{llm_response}"
 
             final_answer_match = re.search(r"Final Answer:\s*(.*)", llm_response, re.DOTALL)
             if final_answer_match:
-                answer = final_answer_match.group(1).strip()
-                print(f"Final Answer Found: {answer}")
-                return answer
+                return final_answer_match.group(1).strip()
+            
+            # *** THIS IS THE ONLY LINE THAT HAS BEEN CHANGED ***
+            # Corrected the typo from `ll.response` to `llm_response`
+            action_match = re.search(r"Action:\s*(\w+)\s*Action Input:((.|\n)*)", llm_response)
             
-            action_match = re.search(r"Action:\s*(\w+)\s*Action Input:((.|\n)*)", ll.response)
             if action_match:
                 tool_name = action_match.group(1).strip()
                 tool_input = action_match.group(2).strip(' \n"`')