cookbook-ai / example.py
markeff's picture
Update example.py
e7c0cb6 verified
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)