Update app.py

app.py

@@ -1,10 +1,24 @@
-import gradio as gr
+from flask import Flask, request, jsonify
 from transformers import pipeline
 
+# 初始化 GPT-2 模型
 generator = pipeline("text-generation", model="gpt2")
 
-def generate_text(prompt):
-    result = generator(prompt, max_length=100, num_return_sequences=1)
-    return result[0]['generated_text']
+# 建立 Flask 應用
+app = Flask(__name__)
 
-gr.Interface(fn=generate_text, inputs="text", outputs="text", title="GPT-2 Text Generator").launch()
+@app.route("/chat", methods=["POST"])
+def chat():
+    try:
+        data = request.get_json()
+        prompt = data.get("text", "")
+        result = generator(prompt, max_length=100, num_return_sequences=1)
+        reply = result[0]["generated_text"]
+        return jsonify({"reply": reply})
+    except Exception as e:
+        print("❌ 模型回應錯誤：", e)
+        return jsonify({"reply": "目前無法取得模型回覆，請稍後再試。"})
+
+# 啟動 Flask（Hugging Face 會自動監聽）
+if __name__ == "__main__":
+    app.run()