Update app.py

app.py

@@ -1,7 +1,10 @@
+from flask import Flask, request, jsonify
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 import json
 
+app = Flask(__name__)
+
 # Load DialoGPT model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
 model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
@@ -29,12 +32,11 @@ def generate_response(user_input):
         response = tokenizer.decode(response_ids[0], skip_special_tokens=True)
         return response
 
+@app.route("/", methods=["POST"])
+def chat():
+    user_input = request.json["user_input"]
+    response = generate_response(user_input)
+    return jsonify({"response": response})
+
 if __name__ == "__main__":
-    print("Welcome to the UTS Course Chatbot!")
-    print("Type 'help' to see available commands and 'exit' to quit.")
-    while True:
-        user_input = input("You: ")
-        response = generate_response(user_input)
-        print("Chatbot:", response)
-        if user_input.lower() == "exit":
-            break
+    app.run(debug=True)