split ode in modules

app.py

@@ -1,31 +1,22 @@
-from climateqa.engine.embeddings import get_embeddings_function
-embeddings_function = get_embeddings_function()
-
-from sentence_transformers import CrossEncoder
-
-import gradio as gr
-from gradio_modal import Modal
-import pandas as pd
-import numpy as np
+# Import necessary libraries
 import os
+import json
 import time
 import re
-import json
-
-from gradio import ChatMessage
-
-from io import BytesIO
 import base64
-
 from datetime import datetime
-from azure.storage.fileshare import ShareServiceClient
-
-from utils import create_user_id
+from io import BytesIO
 
+import numpy as np
+import pandas as pd
+import gradio as gr
+from gradio import ChatMessage
 from gradio_modal import Modal
+from sentence_transformers import CrossEncoder
+from azure.storage.fileshare import ShareServiceClient
 
-
-# ClimateQ&A imports
+# Import custom modules
+from climateqa.engine.embeddings import get_embeddings_function
 from climateqa.engine.llm import get_llm
 from climateqa.engine.vectorstore import get_pinecone_vectorstore
 from climateqa.engine.reranker import get_reranker
@@ -33,12 +24,15 @@ from climateqa.sample_questions import QUESTIONS
 from climateqa.constants import POSSIBLE_REPORTS
 from climateqa.utils import get_image_from_azure_blob_storage
 from climateqa.engine.graph import make_graph_agent
-from climateqa.engine.embeddings import get_embeddings_function
 from climateqa.engine.chains.retrieve_papers import find_papers
-
-from front.utils import serialize_docs,process_figures
-
-from climateqa.event_handler import init_audience, handle_retrieved_documents, stream_answer,handle_retrieved_owid_graphs 
+from front.utils import serialize_docs, process_figures
+from climateqa.event_handler import (
+    init_audience,
+    handle_retrieved_documents,
+    stream_answer,
+    handle_retrieved_owid_graphs
+)
+from utils import create_user_id
 
 # Load environment variables in local mode
 try:
@@ -47,7 +41,6 @@ try:
 except Exception as e:
     pass
 
-
 # Set up Gradio Theme
 theme = gr.themes.Base(
     primary_hue="blue",
@@ -55,8 +48,7 @@ theme = gr.themes.Base(
     font=[gr.themes.GoogleFont("Poppins"), "ui-sans-serif", "system-ui", "sans-serif"],
 )
 
-
-
+# Initialize prompt and system template
 init_prompt = """
 Hello, I am ClimateQ&A, a conversational assistant designed to help you understand climate change and biodiversity loss. I will answer your questions by **sifting through the IPCC and IPBES scientific reports**.
 
@@ -72,15 +64,10 @@ Hello, I am ClimateQ&A, a conversational assistant designed to help you understa
 🛈 Information
 Please note that we log your questions for meta-analysis purposes, so avoid sharing any sensitive or personal information.
 
-
 What do you want to learn ?
 """
 
-system_template = {
-    "role": "system",
-    "content": init_prompt,
-}
-
+# Azure Blob Storage credentials
 account_key = os.environ["BLOB_ACCOUNT_KEY"]
 if len(account_key) == 86:
     account_key += "=="
@@ -97,7 +84,7 @@ share_client = service.get_share_client(file_share_name)
 
 user_id = create_user_id()
 
-
+# Citation information
 CITATION_LABEL = "BibTeX citation for ClimateQ&A"
 CITATION_TEXT = r"""@misc{climateqa,
     author={Théo Alves Da Costa, Timothée Bohe},
@@ -112,22 +99,32 @@ CITATION_TEXT = r"""@misc{climateqa,
 }
 """
 
-
-
 # Create vectorstore and retriever
-vectorstore = get_pinecone_vectorstore(embeddings_function, index_name = os.getenv("PINECONE_API_INDEX"))
-vectorstore_graphs = get_pinecone_vectorstore(embeddings_function, index_name = os.getenv("PINECONE_API_INDEX_OWID"), text_key="description")
+embeddings_function = get_embeddings_function()
+vectorstore = get_pinecone_vectorstore(embeddings_function, index_name=os.getenv("PINECONE_API_INDEX"))
+vectorstore_graphs = get_pinecone_vectorstore(embeddings_function, index_name=os.getenv("PINECONE_API_INDEX_OWID"), text_key="description")
 
-llm = get_llm(provider="openai",max_tokens = 1024,temperature = 0.0)
+llm = get_llm(provider="openai", max_tokens=1024, temperature=0.0)
 reranker = get_reranker("nano")
 
 agent = make_graph_agent(llm=llm, vectorstore_ipcc=vectorstore, vectorstore_graphs=vectorstore_graphs, reranker=reranker)
 
+# 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
 
-async def chat(query, history, audience, sources, reports, relevant_content_sources, search_only):
+# Main chat function
+# async def chat(query, history, audience, sources, reports, relevant_content_sources, search_only):
+async def chat(
+    query: str, 
+    history: list[ChatMessage], 
+    audience: str, 
+    sources: list[str], 
+    reports: list[str], 
+    relevant_content_sources: list[str], 
+    search_only: bool
+) -> tuple[list, str, str, str, list, str]:
     """Process a chat query and return response with relevant sources and visualizations.
     
     Args:
@@ -155,7 +152,7 @@ async def chat(query, history, audience, sources, reports, relevant_content_sour
     audience_prompt = init_audience(audience)
     sources = sources or ["IPCC", "IPBES", "IPOS"]
     reports = reports or []
-    
+
     # Prepare inputs for agent
     inputs = {
         "user_input": query,
@@ -164,7 +161,7 @@ async def chat(query, history, audience, sources, reports, relevant_content_sour
         "relevant_content_sources": relevant_content_sources,
         "search_only": search_only
     }
-    
+
     # Get streaming events from agent
     result = agent.astream_events(inputs, version="v1")
 
@@ -267,7 +264,7 @@ async def chat(query, history, audience, sources, reports, relevant_content_sour
 
     yield history, docs_html, output_query, output_language, related_contents, graphs_html 
 
-
+# Function to save feedback
 def save_feedback(feed: str, user_id):
     if len(feed) > 1:
         timestamp = str(datetime.now().timestamp())
@@ -280,9 +277,7 @@ def save_feedback(feed: str, user_id):
         log_on_azure(file, logs, share_client)
         return "Feedback submitted, thank you!"
 
-
-
-
+# Function to log data on Azure
 def log_on_azure(file, logs, share_client):
     logs = json.dumps(logs)
     file_client = share_client.get_file_client(file)
@@ -297,25 +292,6 @@ def log_on_azure(file, logs, share_client):
 # --------------------------------------------------------------------
 
 
-init_prompt = """
-Hello, I am ClimateQ&A, a conversational assistant designed to help you understand climate change and biodiversity loss. I will answer your questions by **sifting through the IPCC and IPBES scientific reports**.
-
-❓ How to use
-- **Language**: You can ask me your questions in any language. 
-- **Audience**: You can specify your audience (children, general public, experts) to get a more adapted answer.
-- **Sources**: You can choose to search in the IPCC or IPBES reports, or both.
-- **Relevant content sources**: You can choose to search for figures, papers, or graphs that can be relevant for your question.
-
-⚠️ Limitations
-*Please note that the AI is not perfect and may sometimes give irrelevant answers. If you are not satisfied with the answer, please ask a more specific question or report your feedback to help us improve the system.*
-
-🛈 Information
-Please note that we log your questions for meta-analysis purposes, so avoid sharing any sensitive or personal information.
-
-
-What do you want to learn ?
-"""
-
 
 def vote(data: gr.LikeData):
     if data.liked:
@@ -331,6 +307,258 @@ def save_graph(saved_graphs_state, embedding, category):
         saved_graphs_state[category].append(embedding)
     return saved_graphs_state, gr.Button("Graph Saved")
 
+
+# Functions to toggle visibility
+def toggle_summary_visibility():
+    global summary_visible
+    summary_visible = not summary_visible
+    return gr.update(visible=summary_visible)
+
+def toggle_relevant_visibility():
+    global relevant_visible
+    relevant_visible = not relevant_visible
+    return gr.update(visible=relevant_visible)
+
+def change_completion_status(current_state):
+    current_state = 1 - current_state
+    return current_state
+
+def update_sources_number_display(sources_textbox, figures_cards, current_graphs, papers_html):
+    sources_number = sources_textbox.count("<h2>")
+    figures_number = figures_cards.count("<h2>")
+    graphs_number = current_graphs.count("<iframe")
+    papers_number = papers_html.count("<h2>")
+    sources_notif_label = f"Sources ({sources_number})"
+    figures_notif_label = f"Figures ({figures_number})"
+    graphs_notif_label = f"Graphs ({graphs_number})"
+    papers_notif_label = f"Papers ({papers_number})"
+    recommended_content_notif_label = f"Recommended content ({figures_number + graphs_number + papers_number})"
+
+    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 change_sample_questions(key):
+    index = list(QUESTIONS.keys()).index(key)
+    visible_bools = [False] * len(samples)
+    visible_bools[index] = True
+    return [gr.update(visible=visible_bools[i]) for i in range(len(samples))]
+
+
+
+# Chat functions
+def start_chat(query, history, search_only):
+    history = history + [ChatMessage(role="user", content=query)]
+    if not search_only:
+        return (gr.update(interactive=False), gr.update(selected=1), history)
+    else:
+        return (gr.update(interactive=False), gr.update(selected=2), history)
+
+def finish_chat():
+    return gr.update(interactive=True, value="")
+
+# Initialize visibility states
+summary_visible = False
+relevant_visible = False
+
+# UI Layout Components
+def create_chat_interface():
+    chatbot = gr.Chatbot(
+        value=[ChatMessage(role="assistant", content=init_prompt)],
+        type="messages", 
+        show_copy_button=True,
+        show_label=False,
+        elem_id="chatbot",
+        layout="panel",
+        avatar_images=(None, "https://i.ibb.co/YNyd5W2/logo4.png"),
+        max_height="80vh",
+        height="100vh"
+    )
+    
+    with gr.Row(elem_id="input-message"):
+        
+        textbox = gr.Textbox(
+            placeholder="Ask me anything here!",
+            show_label=False,
+            scale=7,
+            lines=1,
+            interactive=True,
+            elem_id="input-textbox"
+        )
+        
+        config_button = gr.Button("", elem_id="config-button")
+    
+    return chatbot, textbox, config_button
+
+def create_examples_tab():
+    examples_hidden = gr.Textbox(visible=False)
+    first_key = list(QUESTIONS.keys())[0]
+    dropdown_samples = gr.Dropdown(
+        choices=QUESTIONS.keys(),
+        value=first_key,
+        interactive=True,
+        label="Select a category of sample questions",
+        elem_id="dropdown-samples"
+    )
+
+    samples = []
+    for i, key in enumerate(QUESTIONS.keys()):
+        examples_visible = (i == 0)
+        with gr.Row(visible=examples_visible) as group_examples:
+            examples_questions = gr.Examples(
+                examples=QUESTIONS[key],
+                inputs=[examples_hidden],
+                examples_per_page=8,
+                run_on_click=False,
+                elem_id=f"examples{i}",
+                api_name=f"examples{i}"
+            )
+        samples.append(group_examples)
+        
+    return examples_hidden, dropdown_samples, samples
+
+def create_figures_tab():
+    sources_raw = gr.State()
+    
+    with Modal(visible=False, elem_id="modal_figure_galery") as figure_modal:
+        gallery_component = gr.Gallery(
+            object_fit='scale-down',
+            elem_id="gallery-component",
+            height="80vh"
+        )
+        
+    show_full_size_figures = gr.Button(
+        "Show figures in full size",
+        elem_id="show-figures",
+        interactive=True
+    )
+    show_full_size_figures.click(
+        lambda: Modal(visible=True),
+        None,
+        figure_modal
+    )
+
+    figures_cards = gr.HTML(show_label=False, elem_id="sources-figures")
+    
+    return sources_raw, gallery_component, figures_cards, figure_modal
+
+def create_papers_tab():
+    with gr.Accordion(
+        visible=True,
+        elem_id="papers-summary-popup",
+        label="See summary of relevant papers",
+        open=False
+    ) as summary_popup:
+        papers_summary = gr.Markdown("", visible=True, elem_id="papers-summary")
+
+    with gr.Accordion(
+        visible=True,
+        elem_id="papers-relevant-popup",
+        label="See relevant papers",
+        open=False
+    ) as relevant_popup:
+        papers_html = gr.HTML(show_label=False, elem_id="papers-textbox")
+
+    btn_citations_network = gr.Button("Explore papers citations network")
+    with Modal(visible=False) as papers_modal:
+        citations_network = gr.HTML(
+            "<h3>Citations Network Graph</h3>",
+            visible=True,
+            elem_id="papers-citations-network"
+        )
+    btn_citations_network.click(
+        lambda: Modal(visible=True),
+        None,
+        papers_modal
+    )
+    
+    return papers_summary, papers_html, citations_network, papers_modal
+
+def create_config_modal(config_open):
+    with Modal(visible=False, elem_id="modal-config") as config_modal:
+        gr.Markdown("Reminders: You can talk in any language, ClimateQ&A is multi-lingual!")
+
+        dropdown_sources = gr.CheckboxGroup(
+            choices=["IPCC", "IPBES", "IPOS"],
+            label="Select source (by default search in all sources)",
+            value=["IPCC"],
+            interactive=True
+        )
+
+        dropdown_reports = gr.Dropdown(
+            choices=POSSIBLE_REPORTS,
+            label="Or select specific reports",
+            multiselect=True,
+            value=None,
+            interactive=True
+        )
+
+        dropdown_external_sources = gr.CheckboxGroup(
+            choices=["IPCC figures", "OpenAlex", "OurWorldInData"],
+            label="Select database to search for relevant content",
+            value=["IPCC figures"],
+            interactive=True
+        )
+
+        search_only = gr.Checkbox(
+            label="Search only for recommended content without chating",
+            value=False,
+            interactive=True,
+            elem_id="checkbox-chat"
+        )
+
+        dropdown_audience = gr.Dropdown(
+            choices=["Children", "General public", "Experts"],
+            label="Select audience", 
+            value="Experts",
+            interactive=True
+        )
+
+        after = gr.Slider(
+            minimum=1950,
+            maximum=2023,
+            step=1,
+            value=1960,
+            label="Publication date",
+            show_label=True,
+            interactive=True,
+            elem_id="date-papers",
+            visible=False
+        )
+
+        output_query = gr.Textbox(
+            label="Query used for retrieval",
+            show_label=True,
+            elem_id="reformulated-query",
+            lines=2,
+            interactive=False,
+            visible=False
+        )
+
+        output_language = gr.Textbox(
+            label="Language",
+            show_label=True,
+            elem_id="language",
+            lines=1,
+            interactive=False,
+            visible=False
+        )
+
+        dropdown_external_sources.change(
+            lambda x: gr.update(visible="OpenAlex" in x),
+            inputs=[dropdown_external_sources],
+            outputs=[after]
+        )
+
+        close_config_modal = gr.Button("Validate and Close", elem_id="close-config-modal")
+        close_config_modal.click(
+            fn=update_config_modal_visibility,
+            inputs=[config_open],
+            outputs=[config_modal, config_open]
+        )
+        
+        return (config_modal, dropdown_sources, dropdown_reports, dropdown_external_sources, 
+                search_only, dropdown_audience, after, output_query, output_language)
+
+# Main UI Assembly
 with gr.Blocks(title="Climate Q&A", css_paths=os.getcwd()+ "/style.css", theme=theme, elem_id="main-component") as demo:
     # State variables
     chat_completed_state = gr.State(0)
@@ -342,57 +570,14 @@ with gr.Blocks(title="Climate Q&A", css_paths=os.getcwd()+ "/style.css", theme=t
         with gr.Row(elem_id="chatbot-row"):
             # Left column - Chat interface
             with gr.Column(scale=2):
-                chatbot = gr.Chatbot(
-                    value=[ChatMessage(role="assistant", content=init_prompt)],
-                    type="messages",
-                    show_copy_button=True,
-                    show_label=False,
-                    elem_id="chatbot",
-                    layout="panel", 
-                    avatar_images=(None, "https://i.ibb.co/YNyd5W2/logo4.png"),
-                    max_height="80vh",
-                    height="100vh"
-                )
-
-                with gr.Row(elem_id="input-message"):
-                    textbox = gr.Textbox(
-                        placeholder="Ask me anything here!",
-                        show_label=False,
-                        scale=7,
-                        lines=1,
-                        interactive=True,
-                        elem_id="input-textbox"
-                    )
-                    config_button = gr.Button("", elem_id="config-button")
+                chatbot, textbox, config_button = create_chat_interface()
 
             # Right column - Content panels
             with gr.Column(scale=2, variant="panel", elem_id="right-panel"):
                 with gr.Tabs(elem_id="right_panel_tab") as tabs:
                     # Examples tab
                     with gr.TabItem("Examples", elem_id="tab-examples", id=0):
-                        examples_hidden = gr.Textbox(visible=False)
-                        first_key = list(QUESTIONS.keys())[0]
-                        dropdown_samples = gr.Dropdown(
-                            choices=QUESTIONS.keys(),
-                            value=first_key,
-                            interactive=True,
-                            label="Select a category of sample questions",
-                            elem_id="dropdown-samples"
-                        )
-
-                        samples = []
-                        for i, key in enumerate(QUESTIONS.keys()):
-                            examples_visible = (i == 0)
-                            with gr.Row(visible=examples_visible) as group_examples:
-                                examples_questions = gr.Examples(
-                                    examples=QUESTIONS[key],
-                                    inputs=[examples_hidden],
-                                    examples_per_page=8,
-                                    run_on_click=False,
-                                    elem_id=f"examples{i}",
-                                    api_name=f"examples{i}"
-                                )
-                            samples.append(group_examples)
+                        examples_hidden, dropdown_samples, samples = create_examples_tab()
 
                     # Sources tab
                     with gr.Tab("Sources", elem_id="tab-sources", id=1) as tab_sources:
@@ -403,58 +588,11 @@ with gr.Blocks(title="Climate Q&A", css_paths=os.getcwd()+ "/style.css", theme=t
                         with gr.Tabs(elem_id="group-subtabs") as tabs_recommended_content:
                             # Figures subtab
                             with gr.Tab("Figures", elem_id="tab-figures", id=3) as tab_figures:
-                                sources_raw = gr.State()
-                                
-                                with Modal(visible=False, elem_id="modal_figure_galery") as figure_modal:
-                                    gallery_component = gr.Gallery(
-                                        object_fit='scale-down',
-                                        elem_id="gallery-component",
-                                        height="80vh"
-                                    )
-                                    
-                                show_full_size_figures = gr.Button(
-                                    "Show figures in full size",
-                                    elem_id="show-figures",
-                                    interactive=True
-                                )
-                                show_full_size_figures.click(
-                                    lambda: Modal(visible=True),
-                                    None,
-                                    figure_modal
-                                )
-
-                                figures_cards = gr.HTML(show_label=False, elem_id="sources-figures")
+                                sources_raw, gallery_component, figures_cards, figure_modal = create_figures_tab()
 
                             # Papers subtab
                             with gr.Tab("Papers", elem_id="tab-citations", id=4) as tab_papers:
-                                with gr.Accordion(
-                                    visible=True,
-                                    elem_id="papers-summary-popup",
-                                    label="See summary of relevant papers",
-                                    open=False
-                                ) as summary_popup:
-                                    papers_summary = gr.Markdown("", visible=True, elem_id="papers-summary")
-
-                                with gr.Accordion(
-                                    visible=True,
-                                    elem_id="papers-relevant-popup",
-                                    label="See relevant papers",
-                                    open=False
-                                ) as relevant_popup:
-                                    papers_html = gr.HTML(show_label=False, elem_id="papers-textbox")
-
-                                btn_citations_network = gr.Button("Explore papers citations network")
-                                with Modal(visible=False) as papers_modal:
-                                    citations_network = gr.HTML(
-                                        "<h3>Citations Network Graph</h3>",
-                                        visible=True,
-                                        elem_id="papers-citations-network"
-                                    )
-                                btn_citations_network.click(
-                                    lambda: Modal(visible=True),
-                                    None,
-                                    papers_modal
-                                )
+                                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:
@@ -468,107 +606,12 @@ with gr.Blocks(title="Climate Q&A", css_paths=os.getcwd()+ "/style.css", theme=t
                                     outputs=[graphs_container]
                                 )
 
-            # Configuration modal
-            with Modal(visible=False, elem_id="modal-config") as config_modal:
-                gr.Markdown("Reminders: You can talk in any language, ClimateQ&A is multi-lingual!")
+           
 
-                dropdown_sources = gr.CheckboxGroup(
-                    choices=["IPCC", "IPBES", "IPOS"],
-                    label="Select source (by default search in all sources)",
-                    value=["IPCC"],
-                    interactive=True
-                )
-
-                dropdown_reports = gr.Dropdown(
-                    choices=POSSIBLE_REPORTS,
-                    label="Or select specific reports",
-                    multiselect=True,
-                    value=None,
-                    interactive=True
-                )
-
-                dropdown_external_sources = gr.CheckboxGroup(
-                    choices=["IPCC figures", "OpenAlex", "OurWorldInData"],
-                    label="Select database to search for relevant content", 
-                    value=["IPCC figures"],
-                    interactive=True
-                )
-
-                search_only = gr.Checkbox(
-                    label="Search only for recommended content without chating",
-                    value=False,
-                    interactive=True,
-                    elem_id="checkbox-chat"
-                )
-
-                dropdown_audience = gr.Dropdown(
-                    choices=["Children", "General public", "Experts"],
-                    label="Select audience",
-                    value="Experts",
-                    interactive=True
-                )
-
-                after = gr.Slider(
-                    minimum=1950,
-                    maximum=2023,
-                    step=1,
-                    value=1960,
-                    label="Publication date",
-                    show_label=True,
-                    interactive=True,
-                    elem_id="date-papers",
-                    visible=False
-                )
-
-                output_query = gr.Textbox(
-                    label="Query used for retrieval",
-                    show_label=True,
-                    elem_id="reformulated-query",
-                    lines=2,
-                    interactive=False,
-                    visible=False
-                )
-
-                output_language = gr.Textbox(
-                    label="Language",
-                    show_label=True,
-                    elem_id="language",
-                    lines=1,
-                    interactive=False,
-                    visible=False
-                )
-
-                dropdown_external_sources.change(
-                    lambda x: gr.update(visible="OpenAlex" in x),
-                    inputs=[dropdown_external_sources],
-                    outputs=[after]
-                )
-
-                close_config_modal = gr.Button("Validate and Close", elem_id="close-config-modal")
-                close_config_modal.click(
-                    fn=update_config_modal_visibility,
-                    inputs=[config_open],
-                    outputs=[config_modal, config_open]
-                )
-
-            config_button.click(
-                fn=update_config_modal_visibility,
-                inputs=[config_open],
-                outputs=[config_modal, config_open]
-            )
-
-
-#---------------------------------------------------------------------------------------
-# OTHER TABS
-#---------------------------------------------------------------------------------------
-
-    with gr.Tab("About",elem_classes = "max-height other-tabs"):
+    # Other tabs
+    with gr.Tab("About", elem_classes="max-height other-tabs"):
         with gr.Row():
             with gr.Column(scale=1):
-
-
-
-
                 gr.Markdown(
                     """
                     ### More info
@@ -578,8 +621,7 @@ with gr.Blocks(title="Climate Q&A", css_paths=os.getcwd()+ "/style.css", theme=t
                     ### Citation
                     """
                 )
-                with gr.Accordion(CITATION_LABEL,elem_id="citation", open = False,):
-                    # # Display citation label and text)
+                with gr.Accordion(CITATION_LABEL, elem_id="citation", open=False):
                     gr.Textbox(
                         value=CITATION_TEXT,
                         label="",
@@ -588,91 +630,43 @@ with gr.Blocks(title="Climate Q&A", css_paths=os.getcwd()+ "/style.css", theme=t
                         lines=len(CITATION_TEXT.split('\n')),
                     )
 
-
-
-    def start_chat(query,history,search_only):
-        history = history + [ChatMessage(role="user", content=query)]
-        if not search_only:
-            return (gr.update(interactive = False),gr.update(selected=1),history)
-        else:
-            return (gr.update(interactive = False),gr.update(selected=2),history)
-    
-    def finish_chat():
-        return gr.update(interactive = True,value = "")
+    # Event handlers
+    config_modal, dropdown_sources, dropdown_reports, dropdown_external_sources, search_only, dropdown_audience, after, output_query, output_language = create_config_modal(config_open)
     
-    # Initialize visibility states
-    summary_visible = False
-    relevant_visible = False
-
-    # Functions to toggle visibility
-    def toggle_summary_visibility():
-        global summary_visible
-        summary_visible = not summary_visible
-        return gr.update(visible=summary_visible)
-
-    def toggle_relevant_visibility():
-        global relevant_visible
-        relevant_visible = not relevant_visible
-        return gr.update(visible=relevant_visible)
-
-
-    def change_completion_status(current_state):
-        current_state = 1 - current_state
-        return current_state
+    config_button.click(
+        fn=update_config_modal_visibility,
+        inputs=[config_open],
+        outputs=[config_modal, config_open]
+    )
     
-    def update_sources_number_display(sources_textbox, figures_cards, current_graphs, papers_html):
-        sources_number = sources_textbox.count("<h2>")
-        figures_number = figures_cards.count("<h2>")
-        graphs_number = current_graphs.count("<iframe")
-        papers_number = papers_html.count("<h2>")
-        sources_notif_label = f"Sources ({sources_number})"
-        figures_notif_label = f"Figures ({figures_number})"
-        graphs_notif_label = f"Graphs ({graphs_number})"
-        papers_notif_label = f"Papers ({papers_number})"
-        recommended_content_notif_label = f"Recommended content ({figures_number + graphs_number + papers_number})"
-
-        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)
     
     (textbox
-        .submit(start_chat, [textbox,chatbot, search_only], [textbox,tabs,chatbot],queue = False,api_name = "start_chat_textbox")
-        .then(chat, [textbox,chatbot,dropdown_audience, dropdown_sources,dropdown_reports, dropdown_external_sources, search_only] ,[chatbot,sources_textbox,output_query,output_language, sources_raw, current_graphs],concurrency_limit = 8,api_name = "chat_textbox")
-        .then(finish_chat, None, [textbox],api_name = "finish_chat_textbox")
-        # .then(update_sources_number_display, [sources_textbox, figures_cards, current_graphs,papers_html],[tab_sources, tab_figures, tab_graphs, tab_papers] )
+        .submit(start_chat, [textbox, chatbot, search_only], [textbox, tabs, chatbot], queue=False, api_name="start_chat_textbox")
+        .then(chat, [textbox, chatbot, dropdown_audience, dropdown_sources, dropdown_reports, dropdown_external_sources, search_only], [chatbot, sources_textbox, output_query, output_language, sources_raw, current_graphs], concurrency_limit=8, api_name="chat_textbox")
+        .then(finish_chat, None, [textbox], api_name="finish_chat_textbox")
     )
 
     (examples_hidden
-        .change(start_chat, [examples_hidden,chatbot, search_only], [textbox,tabs,chatbot],queue = False,api_name = "start_chat_examples")
-        .then(chat, [examples_hidden,chatbot,dropdown_audience, dropdown_sources,dropdown_reports, dropdown_external_sources, search_only] ,[chatbot,sources_textbox,output_query,output_language, sources_raw, current_graphs],concurrency_limit = 8,api_name = "chat_textbox")
-        .then(finish_chat, None, [textbox],api_name = "finish_chat_examples")
-        # .then(update_sources_number_display, [sources_textbox, figures_cards, current_graphs,papers_html],[tab_sources, tab_figures, tab_graphs, tab_papers] )
+        .change(start_chat, [examples_hidden, chatbot, search_only], [textbox, tabs, chatbot], queue=False, api_name="start_chat_examples")
+        .then(chat, [examples_hidden, chatbot, dropdown_audience, dropdown_sources, dropdown_reports, dropdown_external_sources, search_only], [chatbot, sources_textbox, output_query, output_language, sources_raw, current_graphs], concurrency_limit=8, api_name="chat_textbox")
+        .then(finish_chat, None, [textbox], api_name="finish_chat_examples")
     )
 
-
-    def change_sample_questions(key):
-        index = list(QUESTIONS.keys()).index(key)
-        visible_bools = [False] * len(samples)
-        visible_bools[index] = True
-        return [gr.update(visible=visible_bools[i]) for i in range(len(samples))]
-
-
     sources_raw.change(process_figures, inputs=[sources_raw], outputs=[figures_cards, gallery_component])
-    
-    # update sources numbers
-    sources_textbox.change(update_sources_number_display, [sources_textbox, figures_cards, current_graphs,papers_html],[tab_recommended_content, tab_sources, tab_figures, tab_graphs, tab_papers])
-    figures_cards.change(update_sources_number_display, [sources_textbox, figures_cards, current_graphs,papers_html],[tab_recommended_content, tab_sources, tab_figures, tab_graphs, tab_papers])
-    current_graphs.change(update_sources_number_display, [sources_textbox, figures_cards, current_graphs,papers_html],[tab_recommended_content, tab_sources, tab_figures, tab_graphs, tab_papers])
-    papers_html.change(update_sources_number_display, [sources_textbox, figures_cards, current_graphs,papers_html],[tab_recommended_content, tab_sources, tab_figures, tab_graphs, tab_papers])
 
-    # other questions examples
-    dropdown_samples.change(change_sample_questions,dropdown_samples,samples)
+    # Update sources numbers
+    sources_textbox.change(update_sources_number_display, [sources_textbox, figures_cards, current_graphs, papers_html], [tab_recommended_content, tab_sources, tab_figures, tab_graphs, tab_papers])
+    figures_cards.change(update_sources_number_display, [sources_textbox, figures_cards, current_graphs, papers_html], [tab_recommended_content, tab_sources, tab_figures, tab_graphs, tab_papers])
+    current_graphs.change(update_sources_number_display, [sources_textbox, figures_cards, current_graphs, papers_html], [tab_recommended_content, tab_sources, tab_figures, tab_graphs, tab_papers])
+    papers_html.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
-    textbox.submit(find_papers,[textbox,after, dropdown_external_sources], [papers_html,citations_network,papers_summary])
-    examples_hidden.change(find_papers,[examples_hidden,after,dropdown_external_sources], [papers_html,citations_network,papers_summary])
+    # Other questions examples
+    dropdown_samples.change(change_sample_questions, dropdown_samples, samples)
 
-    # btn_summary.click(toggle_summary_visibility, outputs=summary_popup)
-    # btn_relevant_papers.click(toggle_relevant_visibility, outputs=relevant_popup)
+    # Search for papers
+    textbox.submit(find_papers, [textbox, after, dropdown_external_sources], [papers_html, citations_network, papers_summary])
+    examples_hidden.change(find_papers, [examples_hidden, after, dropdown_external_sources], [papers_html, citations_network, papers_summary])
 
     demo.queue()
-
+    
 demo.launch(ssr_mode=False)