Update app.py

app.py

@@ -3,14 +3,16 @@ import datetime
 import requests
 import pytz
 import yaml
-from transformers import PerplexityForTextGeneration
+import importlib
+from gradio import Blocks
+import json
 
 # Import additional tools
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Initialize Perplexity without API key - we'll get it from the user
-perplexity = None
+# Initialize Perplexity without API key
+perplexity_api_key = None
 
 @tool
 def initialize_perplexity(api_key: str) -> str:
@@ -18,12 +20,55 @@ def initialize_perplexity(api_key: str) -> str:
     Args:
         api_key: Your Perplexity API key
     """
-    global perplexity
+    global perplexity_api_key
+    perplexity_api_key = api_key
+    return "Perplexity API initialized successfully!"
+
+def call_perplexity_api(query: str):
+    """Calls the Perplexity API with the provided query and parameters."""
+    global perplexity_api_key
+    if not perplexity_api_key:
+        return "Error: Perplexity API not initialized. Please initialize it first."
+    
+    url = "https://api.perplexity.ai/chat/completions"
+    
+    payload = {
+        "model": "sonar-medium-online",  # online models are better for research
+        "messages": [
+            {
+                "role": "system",
+                "content": "Be precise and concise."
+            },
+            {
+                "role": "user",
+                "content": query
+            }
+        ],
+        "max_tokens": 1000,
+        "temperature": 0.2,
+        "top_p": 0.9,
+        "search_domain_filter": None,
+        "return_images": False,
+        "return_related_questions": False,
+        "search_recency_filter": "month", #limit recency of search to this month
+        "top_k": 0,
+        "stream": False,
+        "presence_penalty": 0,
+        "frequency_penalty": 1,
+        "response_format": None
+    }
+    headers = {
+        "Authorization": f"Bearer {perplexity_api_key}",
+        "Content-Type": "application/json"
+    }
+    
     try:
-        perplexity = PerplexityForTextGeneration(api_key=api_key)
-        return "Perplexity API initialized successfully!"
-    except Exception as e:
-        return f"Error initializing Perplexity API: {str(e)}"
+        response = requests.post(url, json=payload, headers=headers)
+        response.raise_for_status()  # Raise HTTPError for bad responses (4xx or 5xx)
+        return response.json()  # Return the JSON response
+    except requests.exceptions.RequestException as e:
+        return f"Request failed: {e}"  # Handle request exceptions
+  
 
 @tool
 def get_ai_research_papers(query: str) -> str:
@@ -31,18 +76,18 @@ def get_ai_research_papers(query: str) -> str:
     Args:
         query: The search query for AI research papers.
     """
-    if not perplexity:
-        return "Error: Perplexity API not initialized. Please initialize it first."
-        
     try:
-        response = perplexity.search(
-            query=query,
-            max_results=3,
-            filters={"domain": "AI"}
-        )
-        if response:
-            papers = "\n".join([f" {paper.title}" for paper in response])
-            return f"Found these relevant AI papers about {query}:\n{papers}"
+        response_json = call_perplexity_api(f"search AI research papers about: {query}")
+        if isinstance(response_json, str): #Error message
+            return response_json
+        
+        if response_json and "choices" in response_json:
+            content = response_json["choices"][0]["message"]["content"]
+            citations = response_json.get("citations", [])
+            citation_string = "\n".join([f"{i+1}. {citation}" for i, citation in enumerate(citations)])
+            
+            return f"AI Research Papers:\n{content}\n\nCitations:\n{citation_string if citation_string else 'No citations found.'}"
+            
         return f"No relevant AI papers found for your query: {query}"
     except Exception as e:
         return f"Error fetching research papers: {str(e)}"
@@ -53,21 +98,15 @@ def summarize_paper(paper_title: str) -> str:
     Args:
         paper_title: Title of the paper to summarize.
     """
-    if not perplexity:
-        return "Error: Perplexity API not initialized. Please initialize it first."
-        
     try:
-        try:
-            response = perplexity.search(
-                query=paper_title,
-                max_results=1
-            )
-            if response:
-                paper = response[0]
-                summary = paper.summary
-                return f"Summary of '{paper_title}':\n{summary}"
-        except IndexError:
-            return f"Could not find a paper titled '{paper_title}"
+        response_json = call_perplexity_api(f"Summarize AI research paper: {paper_title}")
+        if isinstance(response_json, str): #Error message
+            return response_json
+        
+        if response_json and "choices" in response_json:
+            content = response_json["choices"][0]["message"]["content"]
+            return f"Summary of '{paper_title}':\n{content}"
+        return f"Could not summarize paper '{paper_title}'"
     except Exception as e:
         return f"Error summarizing paper: {str(e)}"
 
@@ -77,25 +116,15 @@ def get_citation(paper_title: str) -> str:
     Args:
         paper_title: Title of the paper to cite.
     """
-    if not perplexity:
-        return "Error: Perplexity API not initialized. Please initialize it first."
-        
     try:
-        try:
-            response = perplexity.search(
-                query=paper_title,
-                max_results=1
-            )
-            if response:
-                paper = response[0]
-                citation = f"""{paper.title}
-Author(s): {paper.authors}
-Published in: {paper.venue}
-Year: {paper.year}
-DOI: {paper.doi if paper.doi else 'Not available'}"""
-                return citation
-        except IndexError:
-            return f"Could not find a paper titled '{paper_title}"
+         response_json = call_perplexity_api(f"Generate citation for AI research paper: {paper_title}")
+         if isinstance(response_json, str): #Error message
+            return response_json
+         
+         if response_json and "choices" in response_json:
+            content = response_json["choices"][0]["message"]["content"]
+            return f"Citation for '{paper_title}':\n{content}"
+         return f"Could not generate citation for '{paper_title}'"
     except Exception as e:
         return f"Error generating citation: {str(e)}"
 
@@ -105,20 +134,15 @@ def explain_concept(concept: str) -> str:
     Args:
         concept: The concept to explain.
     """
-    if not perplexity:
-        return "Error: Perplexity API not initialized. Please initialize it first."
-        
     try:
-        try:
-            response = perplexity.search(
-                query=f"explain {concept}",
-                max_results=1
-            )
-            if response:
-                explanation = response[0].summary
-                return f"Explanation of {concept}:\n{explanation}"
-        except IndexError:
-            return f"Could not find an explanation for '{concept}"
+         response_json = call_perplexity_api(f"Explain the AI concept: {concept}")
+         if isinstance(response_json, str): #Error message
+            return response_json
+         
+         if response_json and "choices" in response_json:
+            content = response_json["choices"][0]["message"]["content"]
+            return f"Explanation of {concept}:\n{content}"
+         return f"Could not explain the concept '{concept}'"
     except Exception as e:
         return f"Error explaining concept: {str(e)}"
 
@@ -154,70 +178,69 @@ agent = CodeAgent(
     grammar=None,
     planning_interval=None,
     name="AI Research Assistant",
-    description="An AI-powered research assistant for exploring AI-related papers and concepts.",
+    description="An AI-powered research assistant for exploring AI-related papers and concepts using Perplexity API.",
     prompt_templates=prompt_templates
 )
 
-# Modified UI with API key input
+# Enhanced UI with automatic dependency installation
 def create_ui():
-    from gradio import Blocks
+    from gradio import Blocks,Markdown, Row, Textbox, Button, Output
     with Blocks() as demo:
-        with gradio.Row():
-            gradio.Markdown("# AI Research Assistant")
-            gradio.Markdown("Your AI-powered research companion")
+        with Row():
+            Markdown("# AI Research Assistant")
+            Markdown("Your AI-powered research companion")
         
-        with gradio.Row():
-            input_api_key = gradio.Textbox(label="Enter your Perplexity API Key:")
-            initialize_button = gradio.Button("Initialize Perplexity API")
+        with Row():
+             Markdown("## Setup Required")
+             Markdown("1. Enter Perplexity API key")
+             input_api_key = Textbox(label="Enter your Perplexity API Key:")
+             initialize_button = Button("Initialize Perplexity API")
         
         initialize_button.click(
             fn=initialize_perplexity,
             inputs=input_api_key,
-            outputs=gradio.Output(label="Initialization Status")
-        )
-        
-        with gradio.Row():
-            input_query = gradio.Textbox(label="Search query for AI papers:")
-            search_button = gradio.Button("Search Papers")
-        
-        search_button.click(
-            fn=get_ai_research_papers,
-            inputs=input_query,
-            outputs=gradio.Output(label="Search Results")
+            outputs=Output(label="Initialization Status")
         )
-        
-        with gradio.Row():
-            input_paper_title = gradio.Textbox(label="Paper title to summarize:")
-            summarize_button = gradio.Button("Summarize Paper")
-        
+        with Row():
+            Markdown("## Research Tools")
+            input_query = Textbox(label="Search query for AI papers:")
+            search_button = Button("Search Papers")
+            search_button.click(
+                fn=get_ai_research_papers,
+                inputs=input_query,
+                outputs=Output(label="Search Results")
+            )
+                
+        with Row():
+            input_paper_title = Textbox(label="Paper title to summarize:")
+            summarize_button = Button("Summarize Paper")
+                
         summarize_button.click(
             fn=summarize_paper,
             inputs=input_paper_title,
-            outputs=gradio.Output(label="Paper Summary")
+            outputs=Output(label="Paper Summary")
         )
-        
-        with gradio.Row():
-            input_citation_title = gradio.Textbox(label="Paper title to cite:")
-            citation_button = gradio.Button("Generate Citation")
-        
+        with Row():
+            input_citation_title = Textbox(label="Paper title to cite:")
+            citation_button = Button("Generate Citation")
+                
         citation_button.click(
-            fn=get_citation,
-            inputs=input_citation_title,
-            outputs=gradio.Output(label="Citation")
-        )
-        
-        with gradio.Row():
-            input_concept = gradio.Textbox(label="Concept to explain:")
-            explain_button = gradio.Button("Explain Concept")
-        
+             fn=get_citation,
+             inputs=input_citation_title,
+             outputs=Output(label="Citation")
+         )
+        with Row():
+            input_concept = Textbox(label="Concept to explain:")
+            explain_button = Button("Explain Concept")
+                    
         explain_button.click(
-            fn=explain_concept,
-            inputs=input_concept,
-            outputs=gradio.Output(label="Explanation")
-        )
-
+                fn=explain_concept,
+                inputs=input_concept,
+                outputs=Output(label="Explanation")
+                )
     return demo
 
 # Launch the enhanced UI
 if __name__ == "__main__":
-    create_ui()
+    demo = create_ui()
+    demo.launch()