Update app.py

app.py

@@ -1,15 +1,40 @@
-from transformers import pipeline
-from transformers import Conversation
+from transformers import Conversation, pipeline
 import gradio as gr
+
+# Load the chatbot pipeline
 chatbot = pipeline(model="facebook/blenderbot-400M-distill")
+
+# Initialize message history lists
 message_list = []
 response_list = []
 
+# Function to interact with the chatbot
 def vanilla_chatbot(message, history):
-    conversation = Conversation(text=message, past_user_inputs=message_list,
-    generated_responses=response_list)
-    bot = chatbot(conversation.messages[0]['content']) # working code
-    return bot[-1]['generated_text']
+    # Create a Conversation object
+    conversation = Conversation(
+        text=message,
+        past_user_inputs=message_list,
+        generated_responses=response_list
+    )
+    
+    # Generate bot response
+    bot_response = chatbot(conversation.messages[0]['content'])
+    
+    # Append message and response to history lists
+    message_list.append(message)
+    response_list.append(bot_response[0]['generated_text'])
+    
+    # Return the generated response
+    return bot_response[0]['generated_text']
+
+# Create a Gradio chat interface
+demo_chatbot = gr.Interface(
+    fn=vanilla_chatbot,
+    inputs=gr.Textbox(lines=2, placeholder="Enter your message here..."),
+    outputs=gr.Textbox(placeholder="Bot response will appear here...", readonly=True),
+    title="Mashdemy Chatbot",
+    description="Enter text to start chatting."
+)
 
-demo_chatbot = gr.ChatInterface(vanilla_chatbot, title="Mashdemy Chatbot", description="Enter text to start chatting.")
-demo_chatbot.launch(share = True)
+# Launch the Gradio interface
+demo_chatbot.launch(share=True)