import openai
from flask import Flask, request, jsonify

app = Flask(__name__)

openai.api_key = 'YOUR_OPENAI_API_KEY'

@app.route('/api/chat', methods=['POST'])
def chat():
    user_input = request.json.get('prompt')
    response = openai.Completion.create(
        model="text-davinci-003",
        prompt=user_input,
        max_tokens=150
    )
    return jsonify({"reply": response.choices[0].text.strip()})

@app.route('/api/generate-image', methods=['POST'])
def generate_image():
    prompt = request.json.get('prompt')
    response = openai.Image.create(
        prompt=prompt,
        n=1,
        size="1024x1024"
    )
    return jsonify({"image_url": response['data'][0]['url']})

if __name__ == '__main__':
    app.run(port=5000)