Display image in a parallel task

app.py

@@ -177,10 +177,7 @@ async def chat(query,history,audience,sources,reports,current_graphs):
                         print(f"Error getting documents: {e}")
                         print(event)
                         
-                    
-                        
-                    used_figures, figures, gallery = process_figures(docs, figures, gallery, used_figures)
-
+                
  
                 elif event["name"] in steps_display.keys() and event["event"] == "on_chain_start": #display steps
                     event_description,display_output = steps_display[node]
@@ -245,7 +242,7 @@ async def chat(query,history,audience,sources,reports,current_graphs):
                 if event["name"] == "categorize_intent" and event["event"] == "on_chain_start":
                     print("X")
             
-            yield history, docs_html, output_query, output_language, gallery, figures, graphs_html #,output_query,output_keywords
+            yield history, docs_html, output_query, output_language, docs , graphs_html#gallery, figures,  #,output_query,output_keywords
  
     except Exception as e:
         print(event, "has failed")
@@ -274,17 +271,19 @@ async def chat(query,history,audience,sources,reports,current_graphs):
         raise gr.Error(f"ClimateQ&A Error: {str(e)[:100]} - The error has been noted, try another question and if the error remains, you can contact us :)")
 
         
-        
-    # figures, gallery = process_figures(docs, figures, gallery)
+
 
    
         
     
 
-    yield history, docs_html, output_query, output_language, gallery, figures, graphs_html#,output_query,output_keywords
+    yield history, docs_html, output_query, output_language, docs, graphs_html # gallery, figures, graphs_html#,output_query,output_keywords
 
-def process_figures(docs, figures, gallery, used_figures):
-    
+# def process_figures(docs, figures, gallery, used_figures =[]):
+def process_figures(docs):
+    gallery=[]
+    used_figures =[]
+    figures = '<div class="figures-container"><p></p> </div>'
     docs_figures = [d for d in docs if d.metadata["chunk_type"] == "image"]
     for i, doc in enumerate(docs_figures):
         if doc.metadata["chunk_type"] == "image":
@@ -315,7 +314,7 @@ def process_figures(docs, figures, gallery, used_figures):
                 except Exception as e:
                     print(f"Skipped adding image {i} because of {e}")
 
-    return used_figures, figures, gallery
+    return figures, gallery
 
 def save_feedback(feed: str, user_id):
     if len(feed) > 1:
@@ -469,6 +468,8 @@ with gr.Blocks(title="Climate Q&A", css_paths=os.getcwd()+ "/style.css", theme=t
                             
                     
                     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 modal:
                             gallery_component = gr.Gallery(object_fit='scale-down',elem_id="gallery-component", height="80vh")
                             
@@ -661,25 +662,21 @@ with gr.Blocks(title="Climate Q&A", css_paths=os.getcwd()+ "/style.css", theme=t
         sources_notif_label = f"Sources ({sources_number})"
         figures_notif_label = f"Figures ({figures_number})"
         graphs_notif_label = f"Recommended content ({graphs_number})"
-        # sources_notif_label = f"Sources (🆕)"
-        # figures_notif_label = f"Figures (🆕)"
-        # graphs_notif_label = f"Recommended content (🆕)"
-         
+
         return gr.update(label = sources_notif_label), gr.update(label = figures_notif_label), gr.update(label = graphs_notif_label)
     
     (textbox
         .submit(start_chat, [textbox,chatbot], [textbox,tabs,chatbot],queue = False,api_name = "start_chat_textbox")
-        .then(chat, [textbox,chatbot,dropdown_audience, dropdown_sources,dropdown_reports, current_graphs], [chatbot,sources_textbox,output_query,output_language,gallery_component, figures_cards, current_graphs],concurrency_limit = 8,api_name = "chat_textbox")
+        .then(chat, [textbox,chatbot,dropdown_audience, dropdown_sources,dropdown_reports, current_graphs] ,[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],[tab_sources, tab_figures, tab_recommended_content] )
     )
 
     (examples_hidden
         .change(start_chat, [examples_hidden,chatbot], [textbox,tabs,chatbot],queue = False,api_name = "start_chat_examples")
-        .then(chat, [examples_hidden,chatbot,dropdown_audience, dropdown_sources,dropdown_reports, current_graphs], [chatbot,sources_textbox,output_query,output_language,gallery_component, figures_cards, current_graphs],concurrency_limit = 8,api_name = "chat_examples")
+        .then(chat, [examples_hidden,chatbot,dropdown_audience, dropdown_sources,dropdown_reports, current_graphs] ,[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],[tab_sources, tab_figures, tab_recommended_content] )
-
     )
 
 
@@ -690,6 +687,10 @@ with gr.Blocks(title="Climate Q&A", css_paths=os.getcwd()+ "/style.css", theme=t
         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])
+    sources_textbox.change(update_sources_number_display, [sources_textbox, figures_cards, current_graphs],[tab_sources, tab_figures, tab_recommended_content])
+    figures_cards.change(update_sources_number_display, [sources_textbox, figures_cards, current_graphs],[tab_sources, tab_figures, tab_recommended_content])
+    current_graphs.change(update_sources_number_display, [sources_textbox, figures_cards, current_graphs],[tab_sources, tab_figures, tab_recommended_content])
 
     dropdown_samples.change(change_sample_questions,dropdown_samples,samples)