Update app.py

app.py

@@ -113,7 +113,21 @@ def update_sources_number_display(sources_textbox, figures_cards, current_graphs
 
     return gr.update(label=recommended_content_notif_label), gr.update(label=sources_notif_label), gr.update(label=figures_notif_label), gr.update(label=graphs_notif_label), gr.update(label=papers_notif_label)
 
-
+def create_drias_tab():
+    with gr.Tab("Beta - Talk to DRIAS", elem_id="tab-vanna", id=6) as tab_vanna:
+        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])
+    
 # # UI Layout Components
 def cqa_tab(tab_name):
     # State variables
@@ -153,19 +167,7 @@ def cqa_tab(tab_name):
                                     "<h2>There are no graphs to be displayed at the moment. Try asking another question.</h2>",
                                     elem_id="graphs-container"
                                 )
-                            with gr.Tab("DRIAS", elem_id="tab-vanna", id=6) as tab_vanna:
-                                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,
@@ -189,9 +191,9 @@ def cqa_tab(tab_name):
         "tab_graphs": tab_graphs,
         "tab_papers": tab_papers,
         "graph_container": graphs_container,
-        "vanna_sql_query": vanna_sql_query,
-        "vanna_table" : vanna_table,
-        "vanna_display": vanna_display
+        # "vanna_sql_query": vanna_sql_query,
+        # "vanna_table" : vanna_table,
+        # "vanna_display": vanna_display
     }
                                 
 def config_event_handling(main_tabs_components : list[dict], config_componenets : dict):
@@ -232,9 +234,9 @@ def event_handling(
     tab_graphs = main_tab_components["tab_graphs"]
     tab_papers = main_tab_components["tab_papers"]
     graphs_container = main_tab_components["graph_container"]
-    vanna_sql_query = main_tab_components["vanna_sql_query"]
-    vanna_table = main_tab_components["vanna_table"]
-    vanna_display = main_tab_components["vanna_display"]
+    # vanna_sql_query = main_tab_components["vanna_sql_query"]
+    # vanna_table = main_tab_components["vanna_table"]
+    # vanna_display = main_tab_components["vanna_display"]
     
     
     # config_open = config_components["config_open"]
@@ -317,6 +319,7 @@ def main_ui():
         with gr.Tabs():
             cqa_components = cqa_tab(tab_name = "ClimateQ&A")
             local_cqa_components = cqa_tab(tab_name = "Beta - POC Adapt'Action")
+            create_drias_tab()
         
             create_about_tab()