app.py

from flask import Flask, request, jsonify
from openai import OpenAI
import os
from dotenv import load_dotenv
from flask_cors import CORS
import traceback

# Load environment variables
load_dotenv()

app = Flask(__name__)
CORS(app)  # Allows frontend to communicate with backend

# Initialize OpenAI client
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))

@app.route("/generate-haiku", methods=["POST"])
def generate_haiku():
    try:
        # Get prompt from request
        data = request.json
        prompt = data.get("prompt", "Write a haiku about nature")

        # Call OpenAI API
        response = client.chat.completions.create(
            model="gpt-3.5-turbo",
            messages=[{"role": "user", "content": prompt}]
        )

        haiku = response.choices[0].message.content
        return jsonify({"haiku": haiku})

    except Exception as e:
        print(f"ERROR: {e}")
        print(traceback.format_exc())
        return jsonify({"error": str(e)}), 500

if __name__ == "__main__":
    app.run(debug=True, host="0.0.0.0", port=8000)