Update app.py

app.py

@@ -1,7 +1,10 @@
-from smolagents import CodeAgent, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, HfApiModel, tool
 import requests
+from requests.exceptions import RequestException  # Add this import
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
+import markdownify
+import re
 
 @tool
 def get_btc_price() -> str:
@@ -16,16 +19,11 @@ def get_btc_price() -> str:
         return '{"error": "Bitcoin price not found"}'
     return f"The current Bitcoin price is ${price:.2f} USD."
 
-
-
-
 @tool
 def visit_webpage(url: str) -> str:
     """Visits a webpage at the given URL and returns its content as a markdown string.
-
     Args:
         url: The URL of the webpage to visit.
-
     Returns:
         The content of the webpage converted to Markdown, or an error message if the request fails.
     """
@@ -35,35 +33,29 @@ def visit_webpage(url: str) -> str:
         response.raise_for_status()  # Raise an exception for bad status codes
 
         # Convert the HTML content to Markdown
-        markdown_content = markdownify(response.text).strip()
+        markdown_content = markdownify.markdownify(response.text).strip()
 
         # Remove multiple line breaks
         markdown_content = re.sub(r"\n{3,}", "\n\n", markdown_content)
 
         return markdown_content
 
-    except RequestException as e:
+    except RequestException as e:  # Updated to use the imported RequestException
         return f"Error fetching the webpage: {str(e)}"
     except Exception as e:
         return f"An unexpected error occurred: {str(e)}"
 
-
-        
 # Initialize the FinalAnswerTool
 final_answer = FinalAnswerTool()
 
 # Initialize the model
 model = HfApiModel(
-    max_tokens=1000,
+    max_tokens=1000,  # Reduced from 2096 to avoid token limit errors
     temperature=0.5,
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
     custom_role_conversions=None,
 )
 
-# Load additional tools if needed
-# For example, to load an image generation tool:
-# image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
 # Create the agent
 agent = CodeAgent(
     model=model,
@@ -79,4 +71,3 @@ agent = CodeAgent(
 
 # Launch the Gradio interface
 GradioUI(agent).launch()
-