Update app.py

app.py

@@ -1,10 +1,11 @@
 from flask import Flask, request, jsonify
-from transformers import pipeline
+from transformers import pipeline, set_seed
 
 app = Flask(__name__)
 
-# Initialize the Hugging Face chatbot model (using DialoGPT for simplicity)
-chatbot = pipeline("conversational", model="microsoft/DialoGPT-medium")
+# Initialize the Hugging Face text-generation pipeline (DialoGPT model)
+chatbot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
+set_seed(42)
 
 # Example restaurant data
 menu = {
@@ -23,8 +24,13 @@ def chat():
     # Get user input from the frontend
     user_input = request.json.get('user_input', '')
 
-    # Generate a response from the chatbot model
-    response = chatbot(user_input)
+    # Add user input to the conversation history (we'll use a simple list to store history)
+    conversation_history = user_input
+
+    # Generate a response from the chatbot model using text generation
+    response = chatbot(conversation_history, max_length=100, num_return_sequences=1, pad_token_id=50256)
+
+    # Get the chatbot's response text
     chatbot_reply = response[0]['generated_text']
 
     # Match user preference to the menu (based on keywords in the input)