Fixing get website tool

main.py

@@ -180,7 +180,7 @@ no_auth = args.no_auth
 
 with gr.Blocks(title="HASHIRU AI", css=css, fill_width=True, fill_height=True) as demo:
     model_manager = GeminiManager(
-        gemini_model="gemini-2.0-flash", modes=[mode for mode in Mode])
+        gemini_model="gemini-2.5-flash-preview-04-17", modes=[mode for mode in Mode])
 
     def update_model(modeIndexes: List[int]):
         modes = [Mode(i+1) for i in modeIndexes]
@@ -230,6 +230,6 @@ if __name__ == "__main__":
     import uvicorn
 
     if no_auth:
-        demo.launch(favicon_path="favicon.ico", server_name="localhost")
+        demo.launch(favicon_path="favicon.ico", server_name="localhost", share=True)
     else:
         uvicorn.run(app, host="0.0.0.0", port=7860)

src/manager/manager.py

@@ -20,6 +20,7 @@ from google.genai.errors import APIError
 import backoff
 import mimetypes
 import json
+import traceback
 
 logger = logging.getLogger(__name__)
 handler = logging.StreamHandler(sys.stdout)
@@ -326,6 +327,7 @@ class GeminiManager:
                 messages = messages + function_call_requests
             yield messages
         except Exception as e:
+            traceback.print_exc(file=sys.stdout)
             print(messages)
             print(chat_history)
             messages.append({

src/tools/default_tools/get_website_tool.py

@@ -21,9 +21,9 @@ class GetWebsite():
                 },
                 "output_type": {
                     "type": "string",
-                    "enum": ["summary", "full_text"],
-                    "description": "The type of output to return. 'summary' returns a summary of the text, 'full_text' returns the full text content.",
-                    "default": "full_text"
+                    "enum": ["summary", "full_text", "html"],
+                    "description": "The type of output to return. 'summary' returns a summary of the text, 'full_text' returns the full text content, and 'html' returns the raw HTML content.",
+                    "default": "summary"
                 },
                 "css_selector": {
                     "type": "string",
@@ -121,17 +121,26 @@ class GetWebsite():
             response = requests.get(url, headers=headers, timeout=10)
             response.raise_for_status()  # Raise HTTPError for bad responses (4xx or 5xx)
             response.encoding = response.apparent_encoding  # Handle encoding
+            if output_type == "html":
+                # Return the raw HTML content
+                return {
+                    "status": "success",
+                    "message": "Search completed successfully",
+                    "output": response.text,
+                }
 
             # Parse the content using BeautifulSoup
-            soup = BeautifulSoup(response.content, 'html.parser')
+            soup = BeautifulSoup(response.text, 'html.parser')
 
             if css_selector:
                 # Extract text from the selected elements
                 elements = soup.select(css_selector)
-                text = ('\n'.join([element.get_text() for element in elements])).encode('utf-8', 'ignore').decode('utf-8')
+                text = ('\n'.join([element.get_text() for element in elements]))
+                text = text.encode('utf-8', 'ignore').decode('utf-8')
             else:
                 # Extract text from the parsed HTML
                 text = soup.get_text()
+                text = text.encode('utf-8', 'ignore').decode('utf-8')
 
             if output_type == "summary":
                 # Summarize the text