Fix : Dynamic follow up examples

app.py

@@ -208,8 +208,8 @@ def event_handling(
             .then(finish_chat, None, [textbox], api_name=f"finish_chat_{examples_hidden.elem_id}")
         )
         (follow_up_examples_hidden
-            .change(start_chat, [examples_hidden, chatbot, search_only], [follow_up_examples_hidden, tabs, chatbot, sources_raw], queue=False, api_name=f"start_chat_{examples_hidden.elem_id}")
-            .then(chat, [examples_hidden, chatbot, dropdown_audience, dropdown_sources, dropdown_reports, dropdown_external_sources, search_only], [chatbot, new_sources_hmtl, output_query, output_language, new_figures, current_graphs,follow_up_examples.dataset], concurrency_limit=8, api_name=f"chat_{examples_hidden.elem_id}")
+            .change(start_chat, [follow_up_examples_hidden, chatbot, search_only], [follow_up_examples_hidden, tabs, chatbot, sources_raw], queue=False, api_name=f"start_chat_{examples_hidden.elem_id}")
+            .then(chat, [follow_up_examples_hidden, chatbot, dropdown_audience, dropdown_sources, dropdown_reports, dropdown_external_sources, search_only], [chatbot, new_sources_hmtl, output_query, output_language, new_figures, current_graphs,follow_up_examples.dataset], concurrency_limit=8, api_name=f"chat_{examples_hidden.elem_id}")
             .then(finish_chat, None, [textbox], api_name=f"finish_chat_{follow_up_examples_hidden.elem_id}")
         )
     
@@ -227,6 +227,11 @@ def event_handling(
             .then(chat_poc, [examples_hidden, chatbot, dropdown_audience, dropdown_sources, dropdown_reports, dropdown_external_sources, search_only], [chatbot, new_sources_hmtl, output_query, output_language, new_figures, current_graphs], concurrency_limit=8, api_name=f"chat_{examples_hidden.elem_id}")
             .then(finish_chat, None, [textbox], api_name=f"finish_chat_{examples_hidden.elem_id}")
         )
+        (follow_up_examples_hidden
+            .change(start_chat, [follow_up_examples_hidden, chatbot, search_only], [follow_up_examples_hidden, tabs, chatbot, sources_raw], queue=False, api_name=f"start_chat_{examples_hidden.elem_id}")
+            .then(chat, [follow_up_examples_hidden, chatbot, dropdown_audience, dropdown_sources, dropdown_reports, dropdown_external_sources, search_only], [chatbot, new_sources_hmtl, output_query, output_language, new_figures, current_graphs,follow_up_examples.dataset], concurrency_limit=8, api_name=f"chat_{examples_hidden.elem_id}")
+            .then(finish_chat, None, [textbox], api_name=f"finish_chat_{follow_up_examples_hidden.elem_id}")
+        )
     
     
     new_sources_hmtl.change(lambda x : x, inputs = [new_sources_hmtl], outputs = [sources_textbox])

climateqa/engine/graph.py

@@ -218,8 +218,6 @@ def make_graph_agent(llm, vectorstore_ipcc, vectorstore_graphs, vectorstore_regi
     workflow.add_edge("retrieve_graphs", END)
     workflow.add_edge("answer_rag", "generate_follow_up")
     workflow.add_edge("answer_rag_no_docs", "generate_follow_up")
-    # workflow.add_edge("answer_rag", END)
-    # workflow.add_edge("answer_rag_no_docs", END)
     workflow.add_edge("answer_chitchat", "chitchat_categorize_intent")
     workflow.add_edge("retrieve_graphs_chitchat", END)
 
@@ -287,7 +285,6 @@ def make_graph_agent_poc(llm, vectorstore_ipcc, vectorstore_graphs, vectorstore_
     workflow.add_node("answer_rag", answer_rag)
     workflow.add_node("answer_rag_no_docs", answer_rag_no_docs)
     workflow.add_node("generate_follow_up", generate_follow_up)
-    workflow.add_node("process_follow_up", standalone_question_node)
 
     # Entry point
     workflow.set_entry_point("standalone_question")
@@ -322,12 +319,6 @@ def make_graph_agent_poc(llm, vectorstore_ipcc, vectorstore_graphs, vectorstore_
         make_id_dict(["retrieve_graphs", END])
     )
 
-    workflow.add_conditional_edges(
-        "generate_follow_up",
-        route_follow_up,
-        make_id_dict(["process_follow_up", END])
-    )
-
     # Define the edges
     workflow.add_edge("standalone_question", "categorize_intent")
     workflow.add_edge("translate_query", "transform_query")
@@ -338,14 +329,13 @@ def make_graph_agent_poc(llm, vectorstore_ipcc, vectorstore_graphs, vectorstore_
     workflow.add_edge("retrieve_graphs", END)
     workflow.add_edge("answer_rag", "generate_follow_up")
     workflow.add_edge("answer_rag_no_docs", "generate_follow_up")
-    workflow.add_edge("answer_rag", END)
-    workflow.add_edge("answer_rag_no_docs", END)
     workflow.add_edge("answer_chitchat", "chitchat_categorize_intent")
     workflow.add_edge("retrieve_graphs_chitchat", END)
 
     workflow.add_edge("retrieve_local_data", "answer_search")
     workflow.add_edge("retrieve_documents", "answer_search")
-    workflow.add_edge("process_follow_up", "categorize_intent")
+    workflow.add_edge("generate_follow_up",END)
+
 
     # Compile
     app = workflow.compile()

front/tabs/chat_interface.py

@@ -56,7 +56,7 @@ def create_chat_interface(tab):
     )
     with gr.Row(elem_id="follow-up-examples"):
         follow_up_examples_hidden = gr.Textbox(visible=False, elem_id="follow-up-hidden")
-        follow_up_examples = gr.Examples(examples=["sample_1","sample_2"], label="Follow up questions", inputs= [follow_up_examples_hidden], elem_id="follow-up-button", run_on_click=False)
+        follow_up_examples = gr.Examples(examples=[ ], label="Follow up questions", inputs= [follow_up_examples_hidden], elem_id="follow-up-button", run_on_click=False)
 
     with gr.Row(elem_id="input-message"):
         

front/tabs/main_tab.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from gradio.helpers import Examples
 from typing import TypedDict
 from .chat_interface import create_chat_interface
 from .tab_examples import create_examples_tab
@@ -29,7 +30,7 @@ class MainTabPanel:
     tab_graphs: gr.Tab
     tab_papers: gr.Tab
     graph_container: gr.HTML
-    follow_up_examples : gr.Examples
+    follow_up_examples : Examples
     follow_up_examples_hidden : gr.Textbox
 
 def cqa_tab(tab_name):