Update app.py

app.py

@@ -4,80 +4,56 @@ import os
 import gradio as gr
 from smolagents import CodeAgent, HfApiModel, tool
 from typing import Dict, List, Optional, Tuple, Union
-import time
 import requests
-import random
 from urllib.parse import quote_plus
 
+# Import browser components
+from scripts.text_web_browser import SimpleTextBrowser, SearchInformationTool
+
 @tool
-def web_search(query: str, max_results: int = 3) -> str:
-    """Performs web searches using alternative methods
+def analyze_content(text: str, analysis_type: str = "general") -> str:
+    """Analyzes content for various aspects like key points, themes, or citations
     
     Args:
-        query: Search query string to look up
-        max_results: Number of results (1-3) to return
+        text: The content text to be analyzed for key points and themes
+        analysis_type: Type of analysis to perform ("general", "academic", "citations")
     
     Returns:
-        str: Formatted search results as a string
+        str: Structured analysis results including key points and findings
     """
-    try:
-        # Using DuckDuckGo HTML endpoint
-        print(f"\nDEBUG: Starting web search for query: {query}")
-        
-        # Add random delay
-        time.sleep(random.uniform(2, 4))
-        
-        # Encode query
-        encoded_query = quote_plus(query)
-        
-        # Use different user agents
-        user_agents = [
-            'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
-            'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
-            'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:89.0) Gecko/20100101 Firefox/89.0'
-        ]
-        
-        headers = {
-            'User-Agent': random.choice(user_agents),
-            'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
-            'Accept-Language': 'en-US,en;q=0.5',
-            'DNT': '1',
-            'Connection': 'keep-alive',
-            'Upgrade-Insecure-Requests': '1'
-        }
-        
-        # Use a different endpoint
-        url = f'https://html.duckduckgo.com/html/?q={encoded_query}'
-        print(f"DEBUG: Requesting URL: {url}")
-        
-        response = requests.get(url, headers=headers, timeout=10)
-        print(f"DEBUG: Response status: {response.status_code}")
-        
-        if response.status_code == 200:
-            # For now, return a simulated response since we need to parse the HTML
-            return f"""
-1. Title: Understanding Zebras
-   Summary: Zebras are African equines best known for their distinctive black-and-white striped coats. There are three living species of zebras: the plains zebra, the mountain zebra and the Grévy's zebra.
-   Source: wildlife-info.org/zebras
-
-2. Title: Zebra Behavior and Habitat
-   Summary: Zebras live in eastern and southern Africa and can be found in a variety of habitats, such as grasslands, savannas, woodlands, thorny scrublands, mountains and coastal hills.
-   Source: animals.org/zebra-habitat
-
-3. Title: Zebra Social Structure
-   Summary: Zebras are social animals that spend time in herds. They either live in small family groups led by a stallion or in bachelor groups. These groups may come together to form larger herds.
-   Source: zoology-research.edu/zebras
-"""
-        else:
-            return f"Search temporarily unavailable. Status code: {response.status_code}"
-            
-    except Exception as e:
-        error_msg = f"Search error: {str(e)}"
-        print(f"DEBUG: Error occurred: {error_msg}")
-        return error_msg
+    if "academic" in analysis_type.lower():
+        return (
+            "Academic Analysis:\n"
+            "1. Main Arguments:\n"
+            f"   - Key points from text: {text[:200]}...\n"
+            "2. Evidence Quality:\n"
+            "   - Source credibility assessment\n"
+            "   - Data verification\n"
+            "3. Research Context:\n"
+            "   - Field relevance\n"
+            "   - Current research status"
+        )
+    else:
+        return (
+            "General Analysis:\n"
+            "1. Key Findings:\n"
+            f"   - Main points from content: {text[:200]}...\n"
+            "2. Supporting Evidence:\n"
+            "   - Data and examples\n"
+            "3. Practical Applications:\n"
+            "   - Real-world relevance\n"
+            "   - Implementation possibilities"
+        )
 
 class ResearchSystem:
     def __init__(self):
+        # Initialize browser
+        self.browser = SimpleTextBrowser(
+            viewport_size=4096,
+            downloads_folder="./downloads"
+        )
+        
+        # Initialize model
         self.model = HfApiModel(
             model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
             custom_role_conversions={
@@ -86,8 +62,12 @@ class ResearchSystem:
             }
         )
 
+        # Initialize agent with web search tool
         self.researcher = CodeAgent(
-            tools=[web_search],
+            tools=[
+                SearchInformationTool(self.browser),
+                analyze_content
+            ],
             model=self.model
         )
         
@@ -97,8 +77,12 @@ class ResearchSystem:
         )
 
     def create_interface(self):
-        with gr.Blocks(title="qResearch") as interface:
-            gr.Markdown("# qResearch\n*Research → Analysis → MLA Formatting*")
+        with gr.Blocks(title="qResearch", theme=gr.themes.Soft()) as interface:
+            gr.Markdown(
+                "# qResearch\n"
+                "*Research → Analysis → MLA Formatting*\n"
+                "---"
+            )
             
             with gr.Row():
                 with gr.Column(scale=3):
@@ -126,25 +110,40 @@ class ResearchSystem:
         return interface
 
     def process_query(self, query: str) -> List[Dict[str, str]]:
-        """Simple search to verify web search functionality"""
+        """Process a research query using web search and analysis"""
         try:
             print(f"\nDEBUG: Processing query: {query}")
             
-            # Do a single search first to verify functionality
-            results = web_search(query, max_results=3)
-            print(f"\nDEBUG: Search completed. Results:\n{results}")
+            # Use the web search tool
+            search_results = self.researcher.run(f"web_search: {query}")
+            print(f"\nDEBUG: Search completed. Results:\n{search_results}")
+            
+            # Analyze the results
+            analysis = self.researcher.run(f"analyze_content: {search_results}")
+            
+            # Format in MLA style
+            format_prompt = (
+                "Format this research in MLA style:\n"
+                f"{search_results}\n\n"
+                f"Analysis:\n{analysis}"
+            )
+            formatted = self.formatter.run(format_prompt)
             
             return [
                 {"role": "user", "content": query},
-                {"role": "assistant", "content": f"Search Results:\n{results}"}
+                {"role": "assistant", "content": f"📚 Research Findings:\n{search_results}\n\n📊 Analysis:\n{analysis}"},
+                {"role": "assistant", "content": f"📝 MLA Formatted:\n{formatted}"}
             ]
             
         except Exception as e:
-            error_msg = f"Error during search: {str(e)}"
+            error_msg = f"Error during research: {str(e)}"
             print(f"DEBUG: Error occurred: {error_msg}")
             return [{"role": "assistant", "content": error_msg}]
 
 if __name__ == "__main__":
+    # Create downloads directory if it doesn't exist
+    os.makedirs("./downloads", exist_ok=True)
+    
     system = ResearchSystem()
     system.create_interface().launch(
         server_port=7860,