Update app.py

app.py

@@ -5,14 +5,17 @@ import json
 from urllib.parse import quote_plus
 import time
 import logging
+import re  # Added missing import
+from smolagents.tools import tool  # Added for tool decorators
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-def search_knowledge_base(issue):
+@tool(name="knowledge_base")
+def search_knowledge_base(issue: str) -> str:
     """
-    Search the knowledge base for solutions related to the provided issue.
+    Search the knowledge base for solutions related to the provided technical issue.
     Args:
         issue (str): The technical issue to search for
     Returns:
@@ -20,64 +23,56 @@ def search_knowledge_base(issue):
     """
     try:
         issue = issue.lower()
-        if "wifi" in issue:
-            return "1. Check if your router is powered on\n2. Restart your router\n3. Try connecting again\n4. If problem persists, contact your ISP"
-        elif "screen" in issue:
-            return "1. Adjust display cable connections\n2. Update graphics drivers\n3. Restart the system\n4. Try a different monitor if available"
+        if "wifi" in issue or "internet" in issue or "network" in issue:
+            return "🔧 WiFi Troubleshooting:\n1. Check if your router is powered on\n2. Restart your router\n3. Try connecting again\n4. If problem persists, contact your ISP"
+        elif "screen" in issue or "display" in issue:
+            return "🖥️ Display Issues:\n1. Adjust display cable connections\n2. Update graphics drivers\n3. Restart the system\n4. Try a different monitor if available"
         elif "sound" in issue or "audio" in issue:
-            return "1. Check volume settings\n2. Verify audio output device selection\n3. Update audio drivers\n4. Test with headphones"
+            return "🔊 Audio Problems:\n1. Check volume settings\n2. Verify audio output device selection\n3. Update audio drivers\n4. Test with headphones"
         return "No predefined solution found in the knowledge base. Please try our web search tool for more information."
     except Exception as e:
         logger.error(f"Error in knowledge base search: {str(e)}")
         return f"Error searching knowledge base: {str(e)}"
 
-def search_web_duckduckgo(query):
+def search_web_duckduckgo(query: str) -> str:
     """
-    Perform actual web search using DuckDuckGo API.
+    Perform web search using DuckDuckGo API.
     Args:
         query (str): The search query
     Returns:
         str: Formatted search results
     """
     try:
-        # DuckDuckGo Instant Answer API
         url = f"https://api.duckduckgo.com/?q={quote_plus(query)}&format=json&no_html=1&skip_disambig=1"
         
         response = requests.get(url, timeout=10)
         response.raise_for_status()
         
         data = response.json()
-        
         results = []
         
-        # Get instant answer if available
         if data.get('AbstractText'):
-            results.append(f"**Summary:** {data['AbstractText']}")
+            results.append(f"📌 {data['AbstractText']}")
         
-        # Get related topics
         if data.get('RelatedTopics'):
-            results.append("\n**Related Information:**")
-            for i, topic in enumerate(data['RelatedTopics'][:3], 1):
+            results.append("\n🔍 Related Topics:")
+            for i, topic in enumerate(data['RelatedTopics'][:5], 1):
                 if isinstance(topic, dict) and topic.get('Text'):
                     results.append(f"{i}. {topic['Text']}")
+                elif hasattr(topic, 'FirstURL'):
+                    results.append(f"{i}. {getattr(topic, 'Text', 'No description')} - {topic.FirstURL}")
         
-        # Get definition if available
-        if data.get('Definition'):
-            results.append(f"\n**Definition:** {data['Definition']}")
+        if data.get('Results'):
+            results.append("\n🌐 Web Results:")
+            for i, result in enumerate(data['Results'][:3], 1):
+                results.append(f"{i}. {result.get('Text', 'No title')} - {result.get('FirstURL', 'No URL')}")
         
-        if results:
-            return "\n".join(results)
-        else:
-            return f"Search completed for '{query}' but no detailed results available. Try a more specific query."
-            
-    except requests.RequestException as e:
-        logger.error(f"Web search API error: {str(e)}")
-        return f"Error performing web search: {str(e)}"
+        return "\n".join(results) if results else search_web_scraper(query)
     except Exception as e:
-        logger.error(f"Unexpected web search error: {str(e)}")
-        return f"Unexpected error: {str(e)}"
+        logger.error(f"DuckDuckGo API error: {str(e)}")
+        return search_web_scraper(query)
 
-def search_web_scraper(query):
+def search_web_scraper(query: str) -> str:
     """
     Alternative web search using web scraping (backup method).
     Args:
@@ -86,7 +81,6 @@ def search_web_scraper(query):
         str: Formatted search results
     """
     try:
-        # Use DuckDuckGo HTML search as backup
         search_url = f"https://duckduckgo.com/html/?q={quote_plus(query)}"
         
         headers = {
@@ -97,26 +91,24 @@ def search_web_scraper(query):
         response.raise_for_status()
         
         soup = BeautifulSoup(response.content, 'html.parser')
-        
-        # Extract search results
         results = []
-        result_links = soup.find_all('a', class_='result__a')[:5]  # Get top 5 results
+        result_links = soup.find_all('a', class_='result__a')[:5]
         
         for i, link in enumerate(result_links, 1):
             title = link.get_text(strip=True)
+            url = link['href'] if 'href' in link.attrs else '#'
             if title:
-                results.append(f"{i}. {title}")
+                results.append(f"{i}. {title} - {url}")
         
         if results:
-            return f"**Search Results for '{query}':**\n" + "\n".join(results)
-        else:
-            return f"No results found for '{query}'. Try different keywords."
-            
+            return f"🔎 Search Results for '{query}':\n" + "\n".join(results)
+        return f"No results found for '{query}'. Try different keywords."
     except Exception as e:
         logger.error(f"Web scraping error: {str(e)}")
         return f"Error in backup search: {str(e)}"
 
-def search_web(query):
+@tool(name="web_search")
+def search_web(query: str) -> str:
     """
     Main web search function that tries multiple methods.
     Args:
@@ -127,21 +119,13 @@ def search_web(query):
     try:
         if not query.strip():
             return "Please enter a search query."
-        
-        # Try DuckDuckGo API first
-        result = search_web_duckduckgo(query)
-        
-        # If API fails, try scraping method
-        if "Error" in result:
-            logger.info("API method failed, trying scraping method...")
-            result = search_web_scraper(query)
-        
-        return result
+        return search_web_duckduckgo(query)
     except Exception as e:
         logger.error(f"Error in main web search: {str(e)}")
         return f"Error processing your search: {str(e)}"
 
-def format_response(raw_steps):
+@tool(name="formatter")
+def format_response(raw_steps: str) -> str:
     """
     Format the raw steps into a numbered list.
     Args:
@@ -153,11 +137,18 @@ def format_response(raw_steps):
         if not raw_steps.strip():
             return "Please enter some text to format."
         
-        # Handle multiple delimiters
-        steps = re.split(r'[.,;]\s*', raw_steps)
+        steps = re.split(r'[.,;]\s+|\n', raw_steps)
         steps = [step.strip() for step in steps if step.strip()]
-        steps = [f"{i+1}. {step}" for i, step in enumerate(steps)]
-        return "\n".join(steps)
+        
+        formatted_steps = []
+        for i, step in enumerate(steps, 1):
+            if step:
+                formatted_step = f"{i}. {step[0].upper()}{step[1:]}" 
+                if not step.endswith('.'):
+                    formatted_step += '.'
+                formatted_steps.append(formatted_step)
+        
+        return "\n".join(formatted_steps)
     except Exception as e:
         logger.error(f"Error formatting steps: {str(e)}")
         return f"Error formatting your steps: {str(e)}"