Upload app.py

app.py

@@ -18,13 +18,20 @@ import warnings
 from typing import Optional, Dict, Any, Tuple, List
 from datetime import datetime
 from transformers import AutoTokenizer, AutoConfig, AutoModelForCausalLM, GPT2Tokenizer
-import requests
-from urllib.parse import quote_plus
-import re
-from bs4 import BeautifulSoup
-import wikipedia
-import threading
-from concurrent.futures import ThreadPoolExecutor, TimeoutError
+# Web search imports - install with: pip install beautifulsoup4 requests
+try:
+    import requests
+    from urllib.parse import quote_plus
+    import re
+    from bs4 import BeautifulSoup
+    import wikipedia
+    import threading
+    from concurrent.futures import ThreadPoolExecutor, TimeoutError
+    WEB_SEARCH_AVAILABLE = True
+except ImportError as e:
+    print(f"⚠️  Web search dependencies not available: {e}")
+    print("📦 Install with: pip install beautifulsoup4 requests")
+    WEB_SEARCH_AVAILABLE = False
 
 # Suppress warnings for cleaner output
 warnings.filterwarnings("ignore")
@@ -736,6 +743,12 @@ class HybridIntelligenceSearchEngine:
         self.search_count = 0
         self.timeout = 10  # seconds
         
+        # Check if web search is available
+        if not WEB_SEARCH_AVAILABLE:
+            print("⚠️  Web search disabled - missing dependencies (beautifulsoup4, requests)")
+            print("📦 Install with: pip install beautifulsoup4 requests")
+            return
+        
         # User-Agent for web requests
         self.headers = {
             'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
@@ -745,6 +758,9 @@ class HybridIntelligenceSearchEngine:
     
     def needs_current_info(self, prompt: str, domain: str) -> bool:
         """Intelligent detection of queries requiring current/real-time information"""
+        if not WEB_SEARCH_AVAILABLE:
+            return False  # No web search available
+            
         prompt_lower = prompt.lower()
         
         # Time-sensitive indicators
@@ -860,6 +876,10 @@ class HybridIntelligenceSearchEngine:
     
     def search_duckduckgo(self, query: str, max_results: int = 5) -> List[Dict[str, str]]:
         """Search using DuckDuckGo Instant Answer API (privacy-focused)"""
+        if not WEB_SEARCH_AVAILABLE:
+            print("🔍 DuckDuckGo search unavailable - missing dependencies")
+            return []
+            
         try:
             # DuckDuckGo Instant Answer API
             url = "https://api.duckduckgo.com/"
@@ -904,6 +924,10 @@ class HybridIntelligenceSearchEngine:
     
     def search_wikipedia(self, query: str, max_results: int = 3) -> List[Dict[str, str]]:
         """Search Wikipedia for factual information"""
+        if not WEB_SEARCH_AVAILABLE:
+            print("📚 Wikipedia search unavailable - missing dependencies")
+            return []
+            
         try:
             # Simple Wikipedia search without the wikipedia library
             search_url = "https://en.wikipedia.org/api/rest_v1/page/summary/"