Update app.py

app.py

@@ -1,9 +1,33 @@
 import streamlit as st
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool
+from smolagents import CodeAgent
 from typing import Union, List, Dict
 from groq import Groq
 import os
+from duckduckgo_search import DDGS
+
+class DuckDuckGoSearch:
+    """Custom DuckDuckGo search implementation"""
+    def __init__(self):
+        self.ddgs = DDGS()
+    
+    def __call__(self, query: str, max_results: int = 5) -> str:
+        try:
+            # Perform the search and get results
+            search_results = list(self.ddgs.text(query, max_results=max_results))
+            
+            # Format the results into a readable string
+            formatted_results = []
+            for idx, result in enumerate(search_results, 1):
+                formatted_results.append(
+                    f"{idx}. Title: {result['title']}\n"
+                    f"   Summary: {result['body']}\n"
+                    f"   Source: {result['link']}\n"
+                )
+            
+            return "\n".join(formatted_results)
+        except Exception as e:
+            return f"Search error: {str(e)}"
 
 class GroqLLM:
     """Compatible LLM interface for smolagents CodeAgent"""
@@ -13,13 +37,8 @@ class GroqLLM:
     
     def __call__(self, prompt: Union[str, dict, List[Dict]]) -> str:
         try:
-            # Handle different prompt formats
-            if isinstance(prompt, (dict, list)):
-                prompt_str = str(prompt)
-            else:
-                prompt_str = str(prompt)
+            prompt_str = str(prompt) if isinstance(prompt, (dict, list)) else prompt
             
-            # Create a properly formatted message
             completion = self.client.chat.completions.create(
                 model=self.model_name,
                 messages=[{
@@ -41,14 +60,30 @@ def create_analysis_prompt(topic: str, search_results: str) -> str:
     return f"""Analyze the following news information about {topic}. 
     Search Results: {search_results}
     
-    Please provide:
-    1. Summary of key points
-    2. Main stakeholders involved
-    3. Potential implications
-    4. Analysis of different perspectives
-    5. Fact-check of major claims (if applicable)
+    Please provide a comprehensive analysis including:
+    1. Key Points Summary:
+       - Main events and developments
+       - Critical updates and changes
+    
+    2. Stakeholder Analysis:
+       - Primary parties involved
+       - Their roles and positions
+    
+    3. Impact Assessment:
+       - Immediate implications
+       - Potential long-term effects
+       - Broader context and significance
+    
+    4. Multiple Perspectives:
+       - Different viewpoints on the issue
+       - Areas of agreement and contention
     
-    Format the analysis in a clear, journalistic style."""
+    5. Fact Check & Reliability:
+       - Verification of major claims
+       - Consistency across sources
+       - Source credibility assessment
+    
+    Please format the analysis in a clear, journalistic style with section headers."""
 
 def log_agent_activity(prompt: str, result: str, agent_name: str):
     """Logs agent activities in the Streamlit interface"""
@@ -65,22 +100,16 @@ st.set_page_config(page_title="News Analysis Tool", layout="wide")
 # Title and description
 st.title("🔍 AI News Analysis Tool")
 st.write("""
-This tool uses advanced AI to analyze news topics, providing comprehensive insights 
-and analysis using real-time data from the web. Powered by Groq's LLama 3.1 8B 
-Instant model and DuckDuckGo search.
+This tool combines the power of Groq's LLama 3.1 8B Instant model with DuckDuckGo 
+search to provide in-depth news analysis. Get comprehensive insights and multiple 
+perspectives on any news topic.
 """)
 
-# Initialize the agents
+# Initialize the components
 try:
-    # Initialize LLM and tools
+    # Initialize LLM and search tool
     llm = GroqLLM()
-    search_tool = DuckDuckGoSearchTool()
-    
-    # Create the analysis agent
-    news_agent = CodeAgent(
-        tools=[search_tool],
-        model=llm
-    )
+    search_tool = DuckDuckGoSearch()
     
     # Input section
     news_topic = st.text_input(
@@ -108,28 +137,32 @@ try:
         if news_topic:
             with st.spinner("Gathering information and analyzing..."):
                 try:
-                    # First, get search results
-                    search_results = search_tool.run(
-                        f"Latest news about {news_topic} last 7 days"
-                    )
-                    
-                    # Create analysis prompt
-                    analysis_prompt = create_analysis_prompt(news_topic, search_results)
-                    
-                    # Get analysis from the agent
-                    analysis_result = news_agent.run(analysis_prompt)
-                    
-                    # Display results
-                    st.subheader("📊 Analysis Results")
-                    st.markdown(analysis_result)
-                    
-                    # Log the activity
-                    log_agent_activity(
-                        analysis_prompt,
-                        analysis_result,
-                        "News Analysis Agent"
+                    # Perform search
+                    search_results = search_tool(
+                        f"Latest news about {news_topic} last 7 days",
+                        max_results=search_depth
                     )
                     
+                    if not search_results.startswith("Search error"):
+                        # Create analysis prompt
+                        analysis_prompt = create_analysis_prompt(news_topic, search_results)
+                        
+                        # Get analysis from LLM
+                        analysis_result = llm(analysis_prompt)
+                        
+                        # Display results
+                        st.subheader("📊 Analysis Results")
+                        st.markdown(analysis_result)
+                        
+                        # Log the activity
+                        log_agent_activity(
+                            analysis_prompt,
+                            analysis_result,
+                            "News Analysis Agent"
+                        )
+                    else:
+                        st.error(f"Search failed: {search_results}")
+                        
                 except Exception as e:
                     st.error(f"An error occurred during analysis: {str(e)}")
         else:
@@ -142,6 +175,7 @@ try:
         - Use keywords related to recent events for timely information
         - Consider including timeframes in your query
         - Try different analysis types for various perspectives
+        - For complex topics, start with a broader search and then narrow down
         """)
 
 except Exception as e:
@@ -150,7 +184,8 @@ except Exception as e:
     
     Please ensure:
     1. Your GROQ_API_KEY is properly set in environment variables
-    2. All required packages are installed
+    2. All required packages are installed:
+       - pip install streamlit groq duckduckgo-search
     3. You have internet connectivity for DuckDuckGo searches
     """)