Update app.py

app.py

@@ -980,7 +980,7 @@ with gr.Blocks(theme=theme) as block:
         chat, query, response = chat_gen.get_chat_engine(dna_documents, api_type)
         chat_engine.append(chat)
         query_engine.append(query)
-        return chat_engine, query_engine, response, 
+        return chat_engine, query_engine, response, f"Chatbot is using {api_type}"
                 
     with gr.Column(visible=True, elem_id='login') as login:
         options = get_collection_names()
@@ -1036,6 +1036,7 @@ with gr.Blocks(theme=theme) as block:
             This DNA chatbot uses the uploaded dna documents to answer questions
             ''')
             chat_llm = gr.Radio(llms, value=api_type.value, label='Choose LLM', info='Choose which LLM you want to use for content')
+            chatbot_status = gr.Label(label='Chabot_Status')
             chatbot = gr.Chatbot()
             msg = gr.Textbox()
             clear = gr.ClearButton([msg, chatbot])
@@ -1056,7 +1057,7 @@ with gr.Blocks(theme=theme) as block:
             def generate_new_chat_engine(dna_documents, chat_engine, chat_llm):
                 chat = chat_gen.get_new_chat_engine(dna_documents, chat_llm)
                 chat_engine[0] = chat
-                return chat_engine
+                return chat_engine, f"Chatbot is using {chat_llm}"
     
     with gr.Tab("E"):
         with gr.Column(visible=True):
@@ -1094,7 +1095,7 @@ with gr.Blocks(theme=theme) as block:
                 bot, [chatbot, chat_engine], chatbot)
     api_type.change(generate_chat_engine)
     clear.click(lambda: None, None, chatbot, queue=False)
-    chat_llm.change(fn=generate_new_chat_engine, inputs=[dna_documents, chat_engine, chat_llm], outputs=chat_engine)
+    chat_llm.change(fn=generate_new_chat_engine, inputs=[dna_documents, chat_engine, chat_llm], outputs=[chat_engine, chatbot_status])
     con_gen_btn.click(generate_content, inputs=[website_layout_file, query_engine], outputs=[status, data_preview, download_btn])
     algo_event = algorithm.click(fn=import_client_data, inputs=[selection, zipfile, topical_map, password, project_bool, ai_bool, query_engine], outputs=[upload, err_imgs])
     bf_event = bf_button.click(fn=bf_trigger.run_preprocess_ai, inputs=[bf_topical_map, bf_selection, section, query_engine], outputs=[bf_upload])
@@ -1102,7 +1103,7 @@ with gr.Blocks(theme=theme) as block:
     stop.click(fn=None, inputs=None, outputs=None, cancels=[algo_event])
     stop_bf.click(fn=None, inputs=None, outputs=None, cancels=[bf_event])
     # upload_btn.upload(upload_file, upload_btn, zipfile)
-    chat_gen_btn.click(generate_chat_engine, inputs=[dna_documents, chat_engine, query_engine, api_type], outputs=[chat_engine, query_engine, chat_gen_progress])
+    chat_gen_btn.click(generate_chat_engine, inputs=[dna_documents, chat_engine, query_engine, api_type], outputs=[chat_engine, query_engine, chat_gen_progress, chatbot_status])
     # stop_offline.click(fn=None, inputs=None, outputs=None, cancels=[offline_event])
    
 block.queue(default_concurrency_limit=5)