from flask import Flask, request, jsonify, render_template
import sqlite3
import openai

app = Flask(__name__)

# Configure OpenAI API Key (replace with your own)
openai.api_key = os.getenv("OPENAI_API_KEY")  # Uses Hugging Face secret

def init_db():
    with sqlite3.connect("recipes.db") as conn:
        cursor = conn.cursor()
        cursor.execute('''CREATE TABLE IF NOT EXISTS recipes 
                          (id INTEGER PRIMARY KEY, user_input TEXT, ai_response TEXT)''')
        conn.commit()

@app.route('/')
def home():
    return render_template('index.html')

@app.route('/get-recipe', methods=['POST'])
def get_recipe():
    user_input = request.json.get("user_input")
    
    if not user_input:
        return jsonify({"error": "No input provided."}), 400
    
    # OpenAI API Call
    response = openai.ChatCompletion.create(
        model="gpt-4-turbo",
        messages=[{"role": "system", "content": "You are a cooking assistant."},
                  {"role": "user", "content": user_input}]
    )
    
    ai_response = response["choices"][0]["message"]["content"]
    
    # Store in DB
    with sqlite3.connect("recipes.db") as conn:
        cursor = conn.cursor()
        cursor.execute("INSERT INTO recipes (user_input, ai_response) VALUES (?, ?)", (user_input, ai_response))
        conn.commit()
    
    return jsonify({"response": ai_response})

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