Create app.py

app.py

@@ -0,0 +1,51 @@
+from flask import Flask, request, jsonify
+from transformers import pipeline
+
+app = Flask(__name__)
+
+# Initialize the Hugging Face chatbot model (using DialoGPT for simplicity)
+chatbot = pipeline("conversational", model="microsoft/DialoGPT-medium")
+
+# Example restaurant data
+menu = {
+    'vegan': ['Vegan Burger', 'Tofu Stir Fry', 'Veggie Pizza'],
+    'vegetarian': ['Cheese Pizza', 'Pasta Alfredo', 'Caprese Salad'],
+    'non_vegetarian': ['Chicken Curry', 'Grilled Salmon', 'Steak'],
+    'spicy': ['Spicy Tacos', 'Chili Chicken', 'Spicy Wings'],
+}
+
+@app.route('/')
+def home():
+    return "Welcome to the Restaurant Chatbot API!"
+
+@app.route('/chat', methods=['POST'])
+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)
+    chatbot_reply = response[0]['generated_text']
+
+    # Match user preference to the menu (based on keywords in the input)
+    user_preference = None
+    if 'vegan' in user_input.lower():
+        user_preference = 'vegan'
+    elif 'vegetarian' in user_input.lower():
+        user_preference = 'vegetarian'
+    elif 'non-vegetarian' in user_input.lower():
+        user_preference = 'non_vegetarian'
+    elif 'spicy' in user_input.lower():
+        user_preference = 'spicy'
+
+    # Get menu suggestions based on user preference
+    menu_suggestion = menu.get(user_preference, [])
+
+    # Return the chatbot response and menu suggestion
+    return jsonify({
+        'bot_response': chatbot_reply,
+        'menu_suggestion': menu_suggestion
+    })
+
+if __name__ == '__main__':
+    app.run(debug=True)