works?

app.py

@@ -31,9 +31,17 @@ with gr.Blocks() as demo:
     clear = gr.ClearButton([msg, chatbot])
 
     def respond(message, chat_history):
-        user_message = "<s>[INST] Given the following conversation context, generate the upcomming dialogue of Joey in his style. \n CONTEXT: Me: " + message + "[/INST]"
-        bot_message = generator(user_message)[0]["generated_text"]
-        chat_history.append((user_message, bot_message))
+        temp_chat_history = []
+        for c in chat_history:
+            temp_chat_history.append('\n'.join(c))
+        print('\n'.join(temp_chat_history))
+        user_message = "Given the following conversation context, generate the upcomming dialogue of Joey in his style. \n CONTEXT: Me: " + '\n'.join(temp_chat_history) + '\n' + message
+        temp_bot_message = generator(user_message)[0]["generated_text"].split('\n')
+        for i in temp_bot_message:
+          if("Joey:" in i):
+            bot_message = i
+            break;        
+        chat_history.append(("Me: " + message, bot_message))
         time.sleep(2)
         return "", chat_history