Update app.py

app.py

@@ -3,27 +3,26 @@ import openai
 
 app = Flask(__name__)
 
-# Set your OpenAI API Key
-openai.api_key = "your_openai_api_key"
-
-@app.route("/", methods=["GET"])
-def home():
-    return jsonify({"message": "Flask API is running on Hugging Face!"})
+openai.api_key = "your-openai-api-key"
 
 @app.route("/chat", methods=["POST"])
 def chat():
-    data = request.get_json()
-    prompt = data.get("prompt", "")
+    try:
+        data = request.get_json()
+        if not data or "prompt" not in data:
+            return jsonify({"error": "Missing 'prompt' in request"}), 400
+
+        prompt = data["prompt"]
 
-    if not prompt:
-        return jsonify({"error": "No prompt provided"}), 400
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",  # Use an available model
+            messages=[{"role": "user", "content": prompt}],
+        )
 
-    response = openai.ChatCompletion.create(
-        model="gpt-4",
-        messages=[{"role": "user", "content": prompt}]
-    )
+        return jsonify(response["choices"][0]["message"])  # Extract AI response
 
-    return jsonify({"response": response["choices"][0]["message"]["content"]})
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, debug=True)
+    app.run(host="0.0.0.0", port=7860)