Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import google.generativeai as genai
-from duckduckgo_search import DDGS
+from ddgs import DDGS
 import requests
 from bs4 import BeautifulSoup
 import time
@@ -9,16 +9,57 @@ import re
 import json
 from typing import List, Dict, Any
 
+# Validate Gemini API key
+def validate_api_key(api_key: str) -> tuple[bool, str]:
+    """Validate if the Gemini API key is working"""
+    if not api_key or not api_key.strip():
+        return False, "API key is empty. Please enter a valid Gemini API key."
+    
+    if not api_key.startswith('AI'):
+        return False, "Invalid API key format. Gemini API keys should start with 'AI'."
+    
+    try:
+        # Test the API key with a simple request
+        genai.configure(api_key=api_key.strip())
+        model = genai.GenerativeModel('gemini-2.0-flash')
+        
+        # Try a minimal test generation
+        response = model.generate_content("Hello")
+        return True, "API key is valid."
+    
+    except Exception as e:
+        error_msg = str(e).lower()
+        if "api key not valid" in error_msg or "api_key_invalid" in error_msg:
+            return False, "Invalid API key. Please check your Gemini API key and try again."
+        elif "quota" in error_msg:
+            return False, "API quota exceeded. Please check your Gemini API usage limits."
+        elif "permission" in error_msg:
+            return False, "API key doesn't have required permissions. Please check your API key settings."
+        else:
+            return False, f"API key validation failed: {str(e)}"
+
 # Search the web for relevant information using DuckDuckGo
 def web_search(query: str, max_results: int = 10) -> List[Dict[str, str]]:
     """Search the web for relevant information using DuckDuckGo"""
     try:
         with DDGS() as ddgs:
-            results = list(ddgs.text(query, max_results=max_results))
+            # Add timeout and retry logic
+            results = []
+            for result in ddgs.text(query, max_results=max_results):
+                results.append(result)
+                if len(results) >= max_results:
+                    break
             return results
     except Exception as e:
         print(f"Search error: {e}")
-        return []
+        # Try with a simpler approach if the first fails
+        try:
+            with DDGS() as ddgs:
+                results = list(ddgs.text(query, max_results=min(max_results, 5)))
+                return results
+        except Exception as e2:
+            print(f"Retry search error: {e2}")
+            return []
 
 # Fetch and extract content from a URL
 def fetch_url_content(url: str) -> str:
@@ -87,11 +128,15 @@ def perform_research(query: str, max_sources: int = 5) -> Dict[str, Any]:
 def generate_research_report(research_data: Dict[str, Any], gemini_api_key: str) -> str:
     """Generate a comprehensive research report using Gemini"""
     if not gemini_api_key:
-        return "Gemini API key is required to generate the report."
+        return "❌ Gemini API key is required to generate the report."
+    
+    # Validate API key first
+    is_valid, validation_message = validate_api_key(gemini_api_key)
+    if not is_valid:
+        return f"❌ {validation_message}"
     
     try:
-        # Initialize Gemini
-        genai.configure(api_key=gemini_api_key)
+        # Initialize Gemini (already configured in validation)
         model = genai.GenerativeModel('gemini-2.0-flash')
         
         prompt = f"""
@@ -114,23 +159,36 @@ def generate_research_report(research_data: Dict[str, Any], gemini_api_key: str)
         response = model.generate_content(prompt)
         return response.text
     except Exception as e:
-        return f"Error generating report: {str(e)}"
+        error_msg = str(e).lower()
+        if "api key not valid" in error_msg or "api_key_invalid" in error_msg:
+            return "❌ Invalid API key. Please check your Gemini API key and try again."
+        elif "quota" in error_msg:
+            return "❌ API quota exceeded. Please check your Gemini API usage limits."
+        elif "permission" in error_msg:
+            return "❌ API key doesn't have required permissions. Please check your API key settings."
+        else:
+            return f"❌ Error generating report: {str(e)}"
 
 # Main research function
 def run_research(topic: str, gemini_api_key: str):
     """Run the complete research process"""
-    if not gemini_api_key:
-        return "Please enter your Gemini API key.", None, gr.update(visible=False)
+    if not gemini_api_key.strip():
+        return "❌ Please enter your Gemini API key.", None, gr.update(visible=False)
+    
+    if not topic.strip():
+        return "❌ Please enter a research topic.", None, gr.update(visible=False)
     
-    if not topic:
-        return "Please enter a research topic.", None, gr.update(visible=False)
+    # First validate the API key
+    is_valid, validation_message = validate_api_key(gemini_api_key)
+    if not is_valid:
+        return f"❌ {validation_message}", None, gr.update(visible=False)
     
     try:
         # Perform research
         research_data = perform_research(topic)
         
         if not research_data['sources']:
-            return "No relevant sources found. Please try a different search term.", None, gr.update(visible=False)
+            return "❌ No relevant sources found. Please try a different search term.", None, gr.update(visible=False)
         
         # Generate report
         report = generate_research_report(research_data, gemini_api_key)
@@ -141,7 +199,7 @@ def run_research(topic: str, gemini_api_key: str):
         return report, filename, gr.update(visible=True)
         
     except Exception as e:
-        error_msg = f"An error occurred: {str(e)}"
+        error_msg = f"❌ An error occurred: {str(e)}"
         return error_msg, None, gr.update(visible=False)
 
 # Gradio interface
@@ -150,23 +208,44 @@ def create_interface():
         gr.Markdown("# 📘 Gemini Deep Research Agent")
         gr.Markdown("This agent performs deep research on any topic using Google's Gemini and DuckDuckGo search")
         
+        # Add API key help section
+        with gr.Accordion("🔑 How to get your Gemini API Key", open=False):
+            gr.Markdown("""
+            1. Visit [Google AI Studio](https://aistudio.google.com/)
+            2. Sign in with your Google account
+            3. Click "Get API Key" 
+            4. Create a new API key
+            5. Copy and paste it below
+            
+            **Note:** Your API key should start with "AI" and be kept secure.
+            """)
+        
         with gr.Row():
             with gr.Column(scale=1):
                 gr.Markdown("## API Configuration")
                 gemini_key = gr.Textbox(
                     label="Gemini API Key",
                     type="password",
-                    placeholder="Enter your Gemini API key (get it from https://aistudio.google.com/)"
+                    placeholder="Enter your Gemini API key (starts with 'AI')",
+                    info="Get your free API key from https://aistudio.google.com/"
+                )
+                
+                # Add API key validation button
+                validate_btn = gr.Button("🔍 Validate API Key", size="sm")
+                validation_output = gr.Textbox(
+                    label="Validation Status", 
+                    interactive=False,
+                    visible=False
                 )
                 
             with gr.Column(scale=2):
                 research_topic = gr.Textbox(
                     label="Research Topic",
-                    placeholder="e.g., Latest developments in AI",
+                    placeholder="e.g., Latest developments in AI, Climate change solutions, Cryptocurrency trends",
                     lines=2
                 )
                 
-                research_btn = gr.Button("Start Research", variant="primary")
+                research_btn = gr.Button("🚀 Start Research", variant="primary")
                 
                 output = gr.Markdown(
                     label="Research Report",
@@ -174,11 +253,29 @@ def create_interface():
                 )
                 
                 download_btn = gr.DownloadButton(
-                    "Download Report",
+                    "📥 Download Report",
                     visible=False
                 )
         
-        # Set up the button action
+        # API key validation function
+        def validate_key(api_key):
+            if not api_key:
+                return gr.update(visible=True, value="❌ Please enter an API key"), gr.update()
+            
+            is_valid, message = validate_api_key(api_key)
+            if is_valid:
+                return gr.update(visible=True, value=f"✅ {message}"), gr.update()
+            else:
+                return gr.update(visible=True, value=f"❌ {message}"), gr.update()
+        
+        # Set up the validation button
+        validate_btn.click(
+            fn=validate_key,
+            inputs=[gemini_key],
+            outputs=[validation_output, validation_output]
+        )
+        
+        # Set up the research button
         research_btn.click(
             fn=run_research,
             inputs=[research_topic, gemini_key],
@@ -200,4 +297,4 @@ def create_interface():
 # Main execution
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch()
+    demo.launch()