Update app.py

app.py

@@ -158,9 +158,7 @@ def conversation(qa_chain, message, history):
 
 def initialize_llm_no_doc(llm_model, temperature, max_tokens, top_k, initial_prompt, progress=gr.Progress()):
     progress(0.1, desc="Initializing HF tokenizer...")
-    
     progress(0.5, desc="Initializing HF Hub...")
-
     llm = HuggingFaceEndpoint(
         repo_id=llm_model,
         huggingfacehub_api_token=api_token,
@@ -168,7 +166,6 @@ def initialize_llm_no_doc(llm_model, temperature, max_tokens, top_k, initial_pro
         max_new_tokens=max_tokens,
         top_k=top_k,
     )
-
     progress(0.75, desc="Defining buffer memory...")
     memory = ConversationBufferMemory(
         memory_key="chat_history",
@@ -302,16 +299,6 @@ def demo():
             outputs=[chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page], 
             queue=False)
 
-        # Chatbot events without document
-        submit_btn_no_doc.click(conversation_no_doc, 
-            inputs=[llm_no_doc, msg_no_doc, chatbot_no_doc], 
-            outputs=[llm_no_doc, msg_no_doc, chatbot_no_doc], 
-            queue=False)
-        clear_btn_no_doc.click(lambda:[None,""], 
-            inputs=None, 
-            outputs=[chatbot_no_doc, msg_no_doc], 
-            queue=False)
-
         # Initialize LLM without document for conversation
         with gr.Tab("Initialize LLM for Chatbot without document"):
             with gr.Row():
@@ -328,11 +315,25 @@ def demo():
                 llm_no_doc_progress = gr.Textbox(value="None", label="LLM initialization for chatbot without document")
             with gr.Row():
                 llm_no_doc_init_btn = gr.Button("Initialize LLM for Chatbot without document")
+            with gr.Row():
+                msg_no_doc = gr.Textbox(placeholder="Type message to chat with lucIAna", container=True)
+            with gr.Row():
+                submit_btn_no_doc = gr.Button("Submit message")
+                clear_btn_no_doc = gr.ClearButton([msg_no_doc, chatbot_no_doc], value="Clear conversation")
         
         llm_no_doc_init_btn.click(initialize_llm_no_doc, 
             inputs=[llm_no_doc_btn, slider_temperature_no_doc, slider_maxtokens_no_doc, slider_topk_no_doc, initial_prompt], 
             outputs=[llm_no_doc, llm_no_doc_progress])
 
+        submit_btn_no_doc.click(conversation_no_doc, 
+            inputs=[llm_no_doc, msg_no_doc, chatbot_no_doc], 
+            outputs=[llm_no_doc, msg_no_doc, chatbot_no_doc], 
+            queue=False)
+        clear_btn_no_doc.click(lambda:[None,""], 
+            inputs=None, 
+            outputs=[chatbot_no_doc, msg_no_doc], 
+            queue=False)
+
     demo.queue().launch(debug=True, share=True)
 
 if __name__ == "__main__":