UI improvment

front/tabs/tab_drias.py

@@ -37,8 +37,14 @@ def show_results(sql_queries_state, dataframes_state, plots_state):
 
 def filter_by_model(dataframes, figures, index_state, model_selection):
     df = dataframes[index_state]
+    if df.empty:
+        return df, None
+    if "model" not in df.columns:
+        return df, figures[index_state](df)
     if model_selection != "ALL":
         df = df[df["model"] == model_selection]
+        if df.empty:
+            return df, None
     figure = figures[index_state](df)
     return df, figure
 
@@ -93,7 +99,7 @@ def create_drias_tab():
     table_names_list = gr.State([])
 
     with gr.Tab("Beta - Talk to DRIAS", elem_id="tab-vanna", id=6):
-        with gr.Accordion(label="Details"):
+        with gr.Accordion(label="Details") as details_accordion:
             gr.Markdown(DRIAS_UI_TEXT)
             
                 # Add examples for common questions
@@ -154,6 +160,10 @@ def create_drias_tab():
 
         # Handle example selection
         examples_hidden.change(
+            lambda x: (gr.Accordion(open=False), gr.Textbox(value=x)),
+            inputs=[examples_hidden],
+            outputs=[details_accordion, drias_direct_question]
+        ).then(
             ask_drias_query,
             inputs=[examples_hidden, index_state],
             outputs=[
@@ -191,6 +201,10 @@ def create_drias_tab():
         )
 
         drias_direct_question.submit(
+            lambda: gr.Accordion(open=False),
+            inputs=None,
+            outputs=[details_accordion]
+        ).then(
             ask_drias_query,
             inputs=[drias_direct_question, index_state],
             outputs=[