Update app.py

app.py

@@ -150,7 +150,8 @@ def add_query(to_add, history):
         history.append(to_add)
     return gr.CheckboxGroup(choices=history), history
 
-def qa_summarise(selected_queries, qa_llm_model, text_field, data_df):
+#def qa_summarise(selected_queries, qa_llm_model, text_field, data_df):
+def qa_summarise(selected_queries, qa_llm_model, text_field, response_lang, data_df):
     
     if not selected_queries:
         raise gr.Error(f"Error: You have to select one or more queries to ask.")
@@ -158,8 +159,15 @@ def qa_summarise(selected_queries, qa_llm_model, text_field, data_df):
     qa_input_df = data_df[data_df["event_label"] != "none"].reset_index()
     texts = qa_input_df[text_field].to_list()
     
-    summary = generate_answer(qa_llm_model, texts, selected_queries[0], selected_queries, mode="multi_summarize")
-
+    # summary = generate_answer(qa_llm_model, texts, selected_queries[0], selected_queries, mode="multi_summarize")
+
+    summary = generate_answer(qa_llm_model, 
+                              texts, 
+                              selected_queries[0], 
+                              selected_queries, 
+                              response_lang, 
+                              mode="multi_summarize")
+    
     doc_df = pd.DataFrame()
     doc_df["number"] = [i+1 for i in range(len(texts))]
     doc_df["text"] = texts
@@ -274,6 +282,7 @@ with gr.Blocks(fill_width=True) as demo:
                     with gr.Column():
                         batch_size = gr.Slider(50, 500, value=150, step=1, label="Batch size", info="Choose between 50 and 500", interactive=True)
                         topk = gr.Slider(1, 10, value=5, step=1, label="Number of results to retrieve", info="Choose between 1 and 10", interactive=True)
+            response_lang = gr.Dropdown(["english", "german", "catalan", "spanish"], label="Response language", value="english", interactive=True)
                         
             selected_queries = gr.CheckboxGroup(label="Select at least one query using the checkboxes", interactive=True)
             queries_state = gr.State()
@@ -417,7 +426,7 @@ with gr.Blocks(fill_width=True) as demo:
     QA_addqry_button.click(add_query, inputs=[query_inp, queries_state], outputs=[selected_queries, queries_state])
 
     QA_run_button.click(qa_summarise, 
-                    inputs=[selected_queries, qa_llm_model, T_text_field, T_data_ss_state],
+                    inputs=[selected_queries, qa_llm_model, T_text_field, response_lang, T_data_ss_state], 
                     outputs=[hsummary, qa_df])   
 
     qa_df.select(T_on_select, None, qa_tweetID)