Update app.py

app.py

@@ -5,17 +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
+import re
+from smolagents.tools import tool
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-@tool(name="knowledge_base")
+@tool
 def search_knowledge_base(issue: str) -> str:
     """
-    Search the knowledge base for solutions related to the provided technical issue.
+    Search the knowledge base for solutions related to the provided issue.
     Args:
         issue (str): The technical issue to search for
     Returns:
@@ -23,12 +23,12 @@ def search_knowledge_base(issue: str) -> str:
     """
     try:
         issue = issue.lower()
-        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"
+        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"
         elif "sound" in issue or "audio" in issue:
-            return "🔊 Audio Problems:\n1. Check volume settings\n2. Verify audio output device selection\n3. Update audio drivers\n4. Test with headphones"
+            return "1. 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)}")
@@ -36,7 +36,7 @@ def search_knowledge_base(issue: str) -> str:
 
 def search_web_duckduckgo(query: str) -> str:
     """
-    Perform web search using DuckDuckGo API.
+    Perform actual web search using DuckDuckGo API.
     Args:
         query (str): The search query
     Returns:
@@ -49,27 +49,24 @@ def search_web_duckduckgo(query: str) -> str:
         response.raise_for_status()
         
         data = response.json()
+        
         results = []
         
         if data.get('AbstractText'):
-            results.append(f"📌 {data['AbstractText']}")
+            results.append(f"Summary: {data['AbstractText']}")
         
         if data.get('RelatedTopics'):
-            results.append("\n🔍 Related Topics:")
-            for i, topic in enumerate(data['RelatedTopics'][:5], 1):
+            results.append("\nRelated Information:")
+            for i, topic in enumerate(data['RelatedTopics'][:3], 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}")
         
-        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 data.get('Definition'):
+            results.append(f"\nDefinition: {data['Definition']}")
         
         return "\n".join(results) if results else search_web_scraper(query)
     except Exception as e:
-        logger.error(f"DuckDuckGo API error: {str(e)}")
+        logger.error(f"Web search API error: {str(e)}")
         return search_web_scraper(query)
 
 def search_web_scraper(query: str) -> str:
@@ -96,18 +93,17 @@ def search_web_scraper(query: str) -> str:
         
         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} - {url}")
+                results.append(f"{i}. {title}")
         
         if results:
-            return f"🔎 Search Results for '{query}':\n" + "\n".join(results)
+            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)}"
 
-@tool(name="web_search")
+@tool
 def search_web(query: str) -> str:
     """
     Main web search function that tries multiple methods.
@@ -124,7 +120,7 @@ def search_web(query: str) -> str:
         logger.error(f"Error in main web search: {str(e)}")
         return f"Error processing your search: {str(e)}"
 
-@tool(name="formatter")
+@tool
 def format_response(raw_steps: str) -> str:
     """
     Format the raw steps into a numbered list.
@@ -137,18 +133,10 @@ def format_response(raw_steps: str) -> str:
         if not raw_steps.strip():
             return "Please enter some text to format."
         
-        steps = re.split(r'[.,;]\s+|\n', raw_steps)
+        steps = re.split(r'[.,;]\s*', raw_steps)
         steps = [step.strip() for step in steps if step.strip()]
-        
-        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)
+        steps = [f"{i+1}. {step}" for i, step in enumerate(steps)]
+        return "\n".join(steps)
     except Exception as e:
         logger.error(f"Error formatting steps: {str(e)}")
         return f"Error formatting your steps: {str(e)}"
@@ -156,31 +144,27 @@ def format_response(raw_steps: str) -> str:
 # Knowledge Base Search Interface
 demo1 = gr.Interface(
     fn=search_knowledge_base,
-    inputs=[gr.Textbox(label="Technical Issue", placeholder="Enter your technical problem (e.g., wifi connection problem)")],
+    inputs=[gr.Textbox(label="Technical Issue", placeholder="Enter your technical problem")],
     outputs=[gr.Textbox(label="Solution")],
-    title="Knowledge Base Search",
-    description="Enter a technical issue to search for solutions in the knowledge base."
+    title="Knowledge Base Search"
 )
 
 # Web Search Interface
 demo2 = gr.Interface(
     fn=search_web,
-    inputs=[gr.Textbox(label="Search Query", placeholder="Enter your search query (e.g., latest tech news)")],
+    inputs=[gr.Textbox(label="Search Query", placeholder="Enter your search query")],
     outputs=[gr.Textbox(label="Search Results")],
-    title="Web Search",
-    description="Enter a search query to get real web search results from DuckDuckGo."
+    title="Web Search"
 )
 
 # Response Formatter Interface
 demo3 = gr.Interface(
     fn=format_response,
-    inputs=[gr.Textbox(label="Raw Steps", placeholder="Enter steps separated by periods or semicolons")],
+    inputs=[gr.Textbox(label="Raw Steps", placeholder="Enter steps separated by periods")],
     outputs=[gr.Textbox(label="Formatted Steps")],
-    title="Response Formatter",
-    description="Enter raw steps separated by periods or semicolons to format them as a numbered list."
+    title="Response Formatter"
 )
 
-# Combine all interfaces using Tabs
 demo = gr.TabbedInterface([demo1, demo2, demo3], ["Knowledge Base", "Web Search", "Formatter"])
 
 if __name__ == "__main__":