Update app.py

app.py

@@ -29,9 +29,11 @@ class BasicAgent:
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         fixed_answer = self.process_request(question)
-        print(f"Agent returning answer: {fixed_answer}")
+        print(f"Agent returning answer: {fixed_answer[:100]}...")  # Truncate long output
         return fixed_answer
     
+    def process_request(self, question: str) -> str:
+        return "This is a default answer."
 
 class SearchAgent(BasicAgent):
     def __init__(self):
@@ -39,18 +41,23 @@ class SearchAgent(BasicAgent):
         print("SearchAgent specialized initialization.")
     
     def process_request(self, query: str) -> str:
-            """Returns mock search results in JSON format"""
-            # Generate 3 mock results with increasing relevance
-            mock_results = [
-                {
-                    "url": f"https://www.goole.com/search?q={query.replace(' ', '+')}&result={i}",
-                    "title": f"Example Result {i} for '{query[:15]}...'",
-                    "snippet": f"This is a mock search result snippet about {query[:15]}... showing result number {i}.",
-                    "relevance": i/3  # Score from 0.33 to 1.0
-                }
-                for i in range(1, 4)
-            ]
-        
+        """Returns mock search results in JSON format"""
+        mock_results = [
+            {
+                "url": f"https://www.google.com/search?q={query.replace(' ', '+')}&result={i}",
+                "title": f"Example Result {i} for '{query[:15]}...'",
+                "snippet": f"This is a mock search result snippet about {query[:15]}... showing result number {i}.",
+                "relevance": i/3  # Score from 0.33 to 1.0
+            }
+            for i in range(1, 4)
+        ]
+        return json.dumps({
+            "query": query,
+            "results": mock_results,
+            "count": len(mock_results),
+            "status": "success"
+        })
+
 class BrowserAgent(BasicAgent):
     def __init__(self):
         super().__init__()
@@ -70,9 +77,17 @@ class BrowserAgent(BasicAgent):
                 'timestamp': datetime.now()
             }
             self.history.append(self.current_page)
-            return f"Successfully retrieved page: {url}"
+            return json.dumps({
+                "status": "success",
+                "url": url,
+                "message": f"Successfully retrieved page: {url}"
+            })
         except Exception as e:
-            return f"Error visiting {url}: {str(e)}"
+            return json.dumps({
+                "status": "error",
+                "url": url,
+                "message": f"Error visiting {url}: {str(e)}"
+            })
 
 class ContentExtractorAgent(BasicAgent):
     def __init__(self):
@@ -94,7 +109,7 @@ class ContentExtractorAgent(BasicAgent):
             'text': main_content.get_text(separator='\n', strip=True),
             'links': [a['href'] for a in main_content.find_all('a', href=True)]
         }
-        return str(extracted)
+        return json.dumps(extracted)
 
 class WebNavigator(BasicAgent):
     def __init__(self):
@@ -114,40 +129,58 @@ class WebNavigator(BasicAgent):
         return self.search_and_extract(question)
     
     def search_and_extract(self, query: str) -> str:
-        search_results = eval(self.search_agent(query))  # Convert string output back to list
-        extracted_data = []
-        
-        for result in search_results:
-            visit_result = self.browser_agent(result['url'])
-            if "Successfully" in visit_result:
-                html = eval(self.browser_agent.current_page['content'])  # Get stored HTML
-                content = self.extractor_agent(html)
-                extracted_data.append({
-                    'query': query,
-                    'url': result['url'],
-                    'content': eval(content)  # Convert string output back to dict
-                })
-        
-        self.search_history.append({
-            'query': query,
-            'timestamp': datetime.now(),
-            'results': extracted_data
-        })
-        
-        return str(extracted_data)
+        try:
+            search_results = json.loads(self.search_agent(query))
+            if search_results['status'] != 'success':
+                return json.dumps({"error": "Search failed", "details": search_results})
+            
+            extracted_data = []
+            for result in search_results['results']:
+                visit_result = json.loads(self.browser_agent(result['url']))
+                
+                if visit_result['status'] == 'success':
+                    content = json.loads(self.extractor_agent(self.browser_agent.current_page['content']))
+                    extracted_data.append({
+                        'query': query,
+                        'url': result['url'],
+                        'content': content
+                    })
+            
+            self.search_history.append({
+                'query': query,
+                'timestamp': datetime.now().isoformat(),
+                'results': extracted_data
+            })
+            
+            return json.dumps({
+                "status": "success",
+                "query": query,
+                "results": extracted_data
+            })
+        except Exception as e:
+            return json.dumps({
+                "status": "error",
+                "message": str(e)
+            })
     
     def get_page_summary(self, url: str) -> str:
-        visit_result = self.browser_agent(url)
-        if "Successfully" in visit_result:
-            html = eval(self.browser_agent.current_page['content'])
-            content = eval(self.extractor_agent(html))
-            return str({
-                'url': url,
-                'title': content['title'],
-                'summary': ' '.join(content['text'].split()[:100]) + '...'
+        try:
+            visit_result = json.loads(self.browser_agent(url))
+            if visit_result['status'] != 'success':
+                return json.dumps(visit_result)
+            
+            content = json.loads(self.extractor_agent(self.browser_agent.current_page['content']))
+            return json.dumps({
+                "status": "success",
+                "url": url,
+                "title": content['title'],
+                "summary": ' '.join(content['text'].split()[:100]) + '...'
+            })
+        except Exception as e:
+            return json.dumps({
+                "status": "error",
+                "message": str(e)
             })
-        return visit_result  # Return the error message
-