Spaces:
Sleeping
Sleeping
File size: 3,081 Bytes
fc04fdd b3bdcce |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 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, input_api_key, model_dropdown |