from flask import Flask, request
from twilio.twiml.messaging_response import MessagingResponse
import openai

app = Flask(__name__)

# Initialize OpenAI API key
openai.api_key = 'sk-proj-3hLD1YvjfJflNqrmvDhdT3BlbkFJuKNTx3CPLEd2j9YCRHWF'

@app.route('/', methods=['GET'])
def index():
    return "WhatsApp ChatGPT API is up and running!", 200

@app.route('/webhook', methods=['POST'])
def webhook():
    incoming_msg = request.values.get('Body', '').strip()
    resp = MessagingResponse()
    msg = resp.message()

    if incoming_msg:
        # Call OpenAI API to get a response
        response = openai.Completion.create(
            engine="davinci",
            prompt=incoming_msg,
            max_tokens=150
        )
        bot_response = response.choices[0].text.strip()
        msg.body(bot_response)
    else:
        msg.body("Sorry, I didn't understand that.")

    return str(resp)

if __name__ == '__main__':
    app.run(debug=True)