Edurag_beta / ui /new_features_tab.py
Nugh75's picture
ristrutturazione file
352ebdd
raw
history blame
1.5 kB
# ui/new_features_tab.py
import gradio as gr
from app.document_handling import search_documents, list_databases
def create_new_features_tab():
"""Crea il tab 'Nuove Funzionalità' dell'interfaccia Gradio."""
def search_documents_callback(query, db_name):
"""Cerca documenti nel database in base alla query."""
results = search_documents(query, db_name)
return "\n".join(results)
# Ottieni la lista dei database
databases = list_databases()
with gr.Tab("Nuove Funzionalità"):
gr.Markdown("## Cerca Documenti e Genera Riassunto")
db_name_new = gr.Dropdown(choices=databases, label="Seleziona Database", value="default_db")
search_input = gr.Textbox(label="Inserisci Termini di Ricerca")
search_button = gr.Button("Cerca Documenti")
search_output = gr.Textbox(label="Documenti Trovati")
summary_button = gr.Button("Genera Riassunto")
summary_output = gr.Textbox(label="Riassunto")
# Evento per il bottone di ricerca
search_button.click(
search_documents_callback,
inputs=[search_input, db_name_new],
outputs=search_output
)
# Evento per il bottone di generazione riassunto (implementare generate_summary se necessario)
# summary_button.click(
# generate_summary,
# inputs=db_name_new,
# outputs=summary_output
# )
return