fix drais display

app.py

@@ -16,6 +16,7 @@ from climateqa.engine.talk_to_data.main import ask_vanna
 
 from front.tabs import (create_config_modal, create_examples_tab, create_papers_tab, create_figures_tab, create_chat_interface, create_about_tab)
 from front.utils import process_figures
+from gradio_modal import Modal
 
 
 from utils import create_user_id
@@ -95,7 +96,7 @@ async def chat_poc(query, history, audience, sources, reports, relevant_content_
 # Function to update modal visibility
 def update_config_modal_visibility(config_open):
     new_config_visibility_status = not config_open
-    return gr.update(visible=new_config_visibility_status), new_config_visibility_status
+    return Modal(visible=new_config_visibility_status), new_config_visibility_status
     
 
 def update_sources_number_display(sources_textbox, figures_cards, current_graphs, papers_html):
@@ -143,7 +144,7 @@ def cqa_tab(tab_name):
 
                             # Papers subtab
                             with gr.Tab("Papers", elem_id="tab-citations", id=4) as tab_papers:
-                                papers_summary, papers_html, citations_network, papers_modal = create_papers_tab()
+                                papers_direct_search, papers_summary, papers_html, citations_network, papers_modal = create_papers_tab()
 
                             # Graphs subtab
                             with gr.Tab("Graphs", elem_id="tab-graphs", id=5) as tab_graphs:
@@ -152,9 +153,18 @@ def cqa_tab(tab_name):
                                     elem_id="graphs-container"
                                 )
                             with gr.Tab("DRIAS", elem_id="tab-vanna", id=6) as tab_vanna:
-                                vanna_sql_query = gr.Textbox(label="SQL Query", elem_id="sql-query", lines=1, interactive=False)
-                                vanna_table = gr.DataFrame([], elem_id="vanna-display")
+                                vanna_direct_question = gr.Textbox(label="Direct Question", placeholder="You can write direct question here",elem_id="direct-question", interactive=True)
+                                with gr.Accordion("Details",elem_id = 'vanna-details', open=False) as vanna_details :
+                                    vanna_sql_query = gr.Textbox(label="SQL Query Used", elem_id="sql-query", interactive=False)
+                                    show_vanna_table = gr.Button("Show Table", elem_id="show-table")
+                                    with Modal(visible=False) as vanna_table_modal:
+                                        vanna_table = gr.DataFrame([], elem_id="vanna-table")
+                                        close_vanna_modal = gr.Button("Close", elem_id="close-vanna-modal")
+                                        close_vanna_modal.click(lambda: Modal(visible=False),None, [vanna_table_modal])
+                                    show_vanna_table.click(lambda: Modal(visible=True),None ,[vanna_table_modal])
+
                                 vanna_display = gr.Plot()
+                                vanna_direct_question.submit(ask_vanna, [vanna_direct_question], [vanna_sql_query ,vanna_table, vanna_display])
                                 
     return {
         "chatbot": chatbot,
@@ -168,6 +178,7 @@ def cqa_tab(tab_name):
         "figures_cards": figures_cards,
         "gallery_component": gallery_component,
         "config_button": config_button,
+        "papers_direct_search" : papers_direct_search,
         "papers_html": papers_html,
         "citations_network": citations_network,
         "papers_summary": papers_summary,
@@ -200,6 +211,7 @@ def event_handling(
     figures_cards = main_tab_components["figures_cards"]
     gallery_component = main_tab_components["gallery_component"]
     config_button = main_tab_components["config_button"]
+    papers_direct_search = main_tab_components["papers_direct_search"]
     papers_html = main_tab_components["papers_html"]
     citations_network = main_tab_components["citations_network"]
     papers_summary = main_tab_components["papers_summary"]
@@ -278,9 +290,9 @@ def event_handling(
         component.change(update_sources_number_display, [sources_textbox, figures_cards, current_graphs, papers_html], [tab_recommended_content, tab_sources, tab_figures, tab_graphs, tab_papers])
     
     # Search for papers
-    for component in [textbox, examples_hidden]:
+    for component in [textbox, examples_hidden, papers_direct_search]:
         component.submit(find_papers, [component, after, dropdown_external_sources], [papers_html, citations_network, papers_summary])
-        
+    
         
 
     if tab_name == "Beta - POC Adapt'Action":

front/tabs/tab_papers.py

@@ -3,6 +3,8 @@ from gradio_modal import Modal
 
 
 def create_papers_tab():
+    direct_search_textbox = gr.Textbox(label="Direct search for papers", placeholder= "What is climate change ?", elem_id="papers-search")
+
     with gr.Accordion(
         visible=True,
         elem_id="papers-summary-popup",
@@ -32,5 +34,5 @@ def create_papers_tab():
         papers_modal
     )
     
-    return papers_summary, papers_html, citations_network, papers_modal
+    return direct_search_textbox, papers_summary, papers_html, citations_network, papers_modal
 

style.css

@@ -608,9 +608,13 @@ a {
 
 #vanna-display {
     max-height: 300px;
-    overflow-y: auto;
+    /* overflow-y: scroll; */
 }
 #sql-query{
     max-height: 100px;
     overflow-y:scroll;
 }
+#vanna-details{
+    max-height: 500px;
+    overflow-y:scroll;
+}