Update app.py

app.py

@@ -8,6 +8,7 @@ import tempfile
 from threading import Thread
 import base64
 import shutil
+import re
 
 import gradio as gr
 import spaces
@@ -112,6 +113,80 @@ class Model:
         export_to_ply(images[0], ply_path.name)
         return self.to_glb(ply_path.name)
 
+# -----------------------------------------------------------------------------
+# New Tools for Web Functionality using DuckDuckGo and smolagents
+# -----------------------------------------------------------------------------
+
+from typing import Any, Optional
+from smolagents.tools import Tool
+import duckduckgo_search
+
+class DuckDuckGoSearchTool(Tool):
+    name = "web_search"
+    description = "Performs a duckduckgo web search based on your query (think a Google search) then returns the top search results."
+    inputs = {'query': {'type': 'string', 'description': 'The search query to perform.'}}
+    output_type = "string"
+
+    def __init__(self, max_results=10, **kwargs):
+        super().__init__()
+        self.max_results = max_results
+        try:
+            from duckduckgo_search import DDGS
+        except ImportError as e:
+            raise ImportError(
+                "You must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`."
+            ) from e
+        self.ddgs = DDGS(**kwargs)
+
+    def forward(self, query: str) -> str:
+        results = self.ddgs.text(query, max_results=self.max_results)
+        if len(results) == 0:
+            raise Exception("No results found! Try a less restrictive/shorter query.")
+        postprocessed_results = [
+            f"[{result['title']}]({result['href']})\n{result['body']}" for result in results
+        ]
+        return "## Search Results\n\n" + "\n\n".join(postprocessed_results)
+
+class VisitWebpageTool(Tool):
+    name = "visit_webpage"
+    description = "Visits a webpage at the given url and reads its content as a markdown string. Use this to browse webpages."
+    inputs = {'url': {'type': 'string', 'description': 'The url of the webpage to visit.'}}
+    output_type = "string"
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False
+
+    def forward(self, url: str) -> str:
+        try:
+            import requests
+            from markdownify import markdownify
+            from requests.exceptions import RequestException
+
+            from smolagents.utils import truncate_content
+        except ImportError as e:
+            raise ImportError(
+                "You must install packages `markdownify` and `requests` to run this tool: for instance run `pip install markdownify requests`."
+            ) from e
+        try:
+            # Send a GET request to the URL with a 20-second timeout
+            response = requests.get(url, timeout=20)
+            response.raise_for_status()  # Raise an exception for bad status codes
+
+            # Convert the HTML content to Markdown
+            markdown_content = markdownify(response.text).strip()
+
+            # Remove multiple line breaks
+            markdown_content = re.sub(r"\n{3,}", "\n\n", markdown_content)
+
+            return truncate_content(markdown_content, 10000)
+
+        except requests.exceptions.Timeout:
+            return "The request timed out. Please try again later or check the URL."
+        except RequestException as e:
+            return f"Error fetching the webpage: {str(e)}"
+        except Exception as e:
+            return f"An unexpected error occurred: {str(e)}"
+
 # -----------------------------------------------------------------------------
 # Gradio UI configuration
 # -----------------------------------------------------------------------------
@@ -297,7 +372,7 @@ def generate_3d_fn(
     return glb_path, seed
 
 # -----------------------------------------------------------------------------
-# Chat Generation Function with support for @tts, @image, and @3d commands
+# Chat Generation Function with support for @tts, @image, @3d, and now @web commands
 # -----------------------------------------------------------------------------
 
 @spaces.GPU
@@ -312,12 +387,13 @@ def generate(
 ):
     """
     Generates chatbot responses with support for multimodal input, TTS, image generation,
-    and 3D model generation.
-    
+    3D model generation, and web search/visit.
+
     Special commands:
       - "@tts1" or "@tts2": triggers text-to-speech.
       - "@image": triggers image generation using the SDXL pipeline.
       - "@3d": triggers 3D model generation using the ShapE pipeline.
+      - "@web": triggers a web search or webpage visit. Use "visit" after @web to fetch a page.
     """
     text = input_dict["text"]
     files = input_dict.get("files", [])
@@ -364,6 +440,25 @@ def generate(
         yield gr.Image(image_paths[0])
         return
 
+    # --- Web Search/Visit branch ---
+    if text.strip().lower().startswith("@web"):
+        web_command = text[len("@web"):].strip()
+        # If the command starts with "visit", then treat the rest as a URL
+        if web_command.lower().startswith("visit"):
+            url = web_command[len("visit"):].strip()
+            yield "Visiting webpage..."
+            visitor = VisitWebpageTool()
+            content = visitor.forward(url)
+            yield content
+        else:
+            # Otherwise, treat the rest as a search query.
+            query = web_command
+            yield "Searching the web..."
+            searcher = DuckDuckGoSearchTool()
+            results = searcher.forward(query)
+            yield results
+        return
+
     # --- Text and TTS branch ---
     tts_prefix = "@tts"
     is_tts = any(text.strip().lower().startswith(f"{tts_prefix}{i}") for i in range(1, 3))
@@ -454,12 +549,12 @@ demo = gr.ChatInterface(
         gr.Slider(label="Repetition penalty", minimum=1.0, maximum=2.0, step=0.05, value=1.2),
     ],
     examples=[
-        ["@tts1 Who is Nikola Tesla, and why did he die?"],
+        ["@tts2 What causes rainbows to form?"],
         ["@3d A birthday cupcake with cherry"],
         [{"text": "summarize the letter", "files": ["examples/1.png"]}],
         ["@image Chocolate dripping from a donut against a yellow background, in the style of brocore, hyper-realistic"],
-        ["Write a Python function to check if a number is prime."],
-        ["@tts2 What causes rainbows to form?"],
+        ["Write a Python Code String Reverse With Example!"],
+        ["@web latest breakthroughs in renewable energy"],
     ],
     cache_examples=False,
     type="messages",