changes

app.py

@@ -9,8 +9,21 @@ from transformers import PerplexityForTextGeneration
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Initialize Perplexity for research capabilities
-perplexity = PerplexityForTextGeneration()
+# Initialize Perplexity without API key - we'll get it from the user
+perplexity = None
+
+@tool
+def initialize_perplexity(api_key: str) -> str:
+    """Initialize Perplexity API with your API key.
+    Args:
+        api_key: Your Perplexity API key
+    """
+    global perplexity
+    try:
+        perplexity = PerplexityForTextGeneration(api_key=api_key)
+        return "Perplexity API initialized successfully!"
+    except Exception as e:
+        return f"Error initializing Perplexity API: {str(e)}"
 
 @tool
 def get_ai_research_papers(query: str) -> str:
@@ -18,14 +31,17 @@ 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=8,
+            max_results=3,
             filters={"domain": "AI"}
         )
         if response:
-            papers = "\n".join([f"📄 {paper.title}" for paper in response])
+            papers = "\n".join([f" {paper.title}" for paper in response])
             return f"Found these relevant AI papers about {query}:\n{papers}"
         return f"No relevant AI papers found for your query: {query}"
     except Exception as e:
@@ -37,6 +53,9 @@ 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(
@@ -58,6 +77,9 @@ 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(
@@ -70,7 +92,7 @@ def get_citation(paper_title: str) -> str:
 Author(s): {paper.authors}
 Published in: {paper.venue}
 Year: {paper.year}
-DOI: {paper.doi if paper.doi else 'Not available}"""
+DOI: {paper.doi if paper.doi else 'Not available'}"""
                 return citation
         except IndexError:
             return f"Could not find a paper titled '{paper_title}"
@@ -83,6 +105,9 @@ 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(
@@ -117,6 +142,7 @@ agent = CodeAgent(
     model=model,
     tools=[
         final_answer,
+        initialize_perplexity,
         get_ai_research_papers,
         summarize_paper,
         get_citation,
@@ -132,11 +158,66 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
+# Modified UI with API key input
+def create_ui():
+    from gradio import Blocks
+    with Blocks() as demo:
+        with gradio.Row():
+            gradio.Markdown("# AI Research Assistant")
+            gradio.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")
+        
+        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")
+        )
+        
+        with gradio.Row():
+            input_paper_title = gradio.Textbox(label="Paper title to summarize:")
+            summarize_button = gradio.Button("Summarize Paper")
+        
+        summarize_button.click(
+            fn=summarize_paper,
+            inputs=input_paper_title,
+            outputs=gradio.Output(label="Paper Summary")
+        )
+        
+        with gradio.Row():
+            input_citation_title = gradio.Textbox(label="Paper title to cite:")
+            citation_button = gradio.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")
+        
+        explain_button.click(
+            fn=explain_concept,
+            inputs=input_concept,
+            outputs=gradio.Output(label="Explanation")
+        )
+
+    return demo
+
 # Launch the enhanced UI
-GradioUI(
-    agent,
-    theme="peach",
-    title="AI Research Assistant",
-    description="Your go-to assistant for finding and understanding AI research papers.",
-    allow_flagging="never"
-).launch()
+if __name__ == "__main__":
+    create_ui()