Create specific_ui.py

specific_ui.py

@@ -0,0 +1,84 @@
+import gradio as gr
+from tools.general_tools import initialize_perplexity
+from tools.perplexity_tools import (
+    get_ai_research_papers,
+    summarize_paper,
+    get_citation,
+    explain_concept,
+)
+
+def create_specific_ui():
+    """Creates the Specific UI components."""
+    with gr.Column() as specific_ui:
+        gr.Markdown("# AI Research Assistant (Specific Mode)")
+        gr.Markdown("Your AI-powered research companion")
+        
+        with gr.Row():
+            gr.Markdown("## Setup Required")
+            gr.Markdown("1. Enter Perplexity API key and select a model")
+            input_api_key = gr.Textbox(label="Enter your Perplexity API Key:")
+            model_dropdown = gr.Dropdown(
+                label="Select Perplexity Model",
+                choices=[
+                    "sonar-reasoning-pro",
+                    "sonar-reasoning",
+                    "sonar-pro",
+                    "sonar"
+                ],
+                value="sonar-pro"  # Default value
+            )
+            initialize_button = gr.Button("Initialize Perplexity API")
+        
+        initialization_status = gr.Textbox(label="Initialization Status")
+        initialize_button.click(
+            fn=initialize_perplexity,
+            inputs=[input_api_key, model_dropdown],
+            outputs=initialization_status
+        )
+        
+        with gr.Row():
+            gr.Markdown("## Research Tools")
+            input_query = gr.Textbox(label="Search query for AI papers:")
+            search_button = gr.Button("Search Papers")
+        
+        search_results = gr.Textbox(label="Search Results")
+        search_button.click(
+            fn=get_ai_research_papers,
+            inputs=[input_query, input_api_key, model_dropdown],
+            outputs=search_results
+        )
+        
+        with gr.Row():
+            input_paper_title = gr.Textbox(label="Paper title to summarize:")
+            summarize_button = gr.Button("Summarize Paper")
+        
+        paper_summary = gr.Textbox(label="Paper Summary")
+        summarize_button.click(
+            fn=summarize_paper,
+            inputs=[input_paper_title, input_api_key, model_dropdown],
+            outputs=paper_summary
+        )
+        
+        with gr.Row():
+            input_citation_title = gr.Textbox(label="Paper title to cite:")
+            citation_button = gr.Button("Generate Citation")
+        
+        citation_output = gr.Textbox(label="Citation")
+        citation_button.click(
+            fn=get_citation,
+            inputs=[input_citation_title, input_api_key, model_dropdown],
+            outputs=citation_output
+        )
+        
+        with gr.Row():
+            input_concept = gr.Textbox(label="Concept to explain:")
+            explain_button = gr.Button("Explain Concept")
+        
+        explanation_output = gr.Textbox(label="Explanation")
+        explain_button.click(
+            fn=explain_concept,
+            inputs=[input_concept, input_api_key, model_dropdown],
+            outputs=explanation_output
+        )
+    
+    return specific_ui