Update app.py

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 import random
-import time
+from simpletransformers.conv_ai import ConvAIModel
+
+# Load the pre-trained conversational AI model
+model = ConvAIModel("gpt", "gpt_personachat_cache")
 
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
@@ -8,9 +11,12 @@ with gr.Blocks() as demo:
     clear = gr.ClearButton([msg, chatbot])
 
     def respond(message, chat_history):
-        bot_message = select.choice(["Hi","How are you?", "Hello","I'm Fine","Had your lunch?" ,"yup","Nope","what are you doing?","Nothing much","Playing","I'm very hungry"])
-        chat_history.append((message, bot_message))
-        time.sleep(2)
+        # Append the user message to the chat history
+        chat_history.append(("User", message))
+        # Generate response using the conversational AI model
+        bot_message = model.generate_responses(message)[0]
+        # Append the bot's response to the chat history
+        chat_history.append(("Chatbot", bot_message))
         return "", chat_history
 
     msg.submit(respond, [msg, chatbot], [msg, chatbot])
@@ -18,4 +24,5 @@ with gr.Blocks() as demo:
 if __name__ == "__main__":
     demo.launch()
 
+