Update app.py

app.py

@@ -71,13 +71,12 @@ def generate_text(input_text, selected_model):
 # Define the chatbot function
 def generate_response(model_name, message, chat_history):
     model = load_model(model_name)
-    # chat_history = []
-    # if len(chat_history) > 0:
-    #     input_text = chat_history + " " + message
-    # else:
-    #     input_text = message
-    chat_history = ["Display 'How may I help you today?'"] # list of strings
-    input_text = chat_history.append(message)
+    chat_history = []
+    if len(chat_history) > 0:
+        past_chat = ", ".join(chat_history)
+        input_text = past_chat + " " + message
+    else:
+        input_text = message
     response = model.generate(input_text, max_tokens=100)
     chat_history.append((input_text, response))
     return chat_history, response
@@ -95,7 +94,7 @@ with gr.Blocks() as demo:
                 label="LLMs to choose from"
             )
         with gr.Column(scale=4):
-            chatbot = gr.Chatbot(label="Conversation")
+            chatbot = gr.Chatbot(label="Conversation", value=[(None, "How may I help you today?")])
             
     message = gr.Textbox(label="Message")
     state = gr.State()