|
1 | | -import os |
2 | | -import json |
3 | | - |
4 | 1 | import aikido_zen # Aikido package import |
5 | 2 | aikido_zen.protect() |
6 | 3 |
|
| 4 | +import os |
7 | 5 | from flask import Flask, render_template, request |
8 | 6 | import psycopg2 |
| 7 | +import openai |
9 | 8 |
|
10 | 9 | app = Flask(__name__) |
| 10 | +client = openai.OpenAI() |
11 | 11 |
|
12 | 12 | def get_db_connection(): |
13 | 13 | return psycopg2.connect( |
@@ -50,26 +50,18 @@ def create_dog(): |
50 | 50 | conn.close() |
51 | 51 | return f'Dog {dog_name} created successfully' |
52 | 52 |
|
53 | | -@app.route("/create/:id", methods=["GET"]) |
54 | | -@app.route("/create_many", methods=['POST']) |
55 | | -def create_dog_many(): |
56 | | - dog_name = request.form['dog_name'] |
57 | | - conn = get_db_connection() |
58 | | - cursor = conn.cursor() |
59 | | - cursor.executemany([f"INSERT INTO dogs (dog_name, isAdmin) VALUES ('%s', FALSE)" % (dog_name)], []) |
60 | | - conn.commit() |
61 | | - cursor.close() |
62 | | - conn.close() |
63 | | - return f'Dog {dog_name} created successfully' |
| 53 | +@app.route("/ask_openai", methods=['GET']) |
| 54 | +def show_ask_openai_form(): |
| 55 | + return render_template('ask_openai.html') |
64 | 56 |
|
65 | | -@app.route("/create_with_cookie", methods=['GET']) |
66 | | -def create_dog_with_cookie(): |
67 | | - dog_name = request.cookies.get('dog_name') |
| 57 | +@app.route("/ask_openai", methods=['POST']) |
| 58 | +def ask_openai(): |
| 59 | + question = request.form['question'] |
68 | 60 |
|
69 | | - conn = get_db_connection() |
70 | | - cursor = conn.cursor() |
71 | | - cursor.execute(f"INSERT INTO dogs (dog_name, isAdmin) VALUES ('%s', FALSE)" % (dog_name)) |
72 | | - conn.commit() |
73 | | - cursor.close() |
74 | | - conn.close() |
75 | | - return f'Dog {dog_name} created successfully' |
| 61 | + response = client.responses.create( |
| 62 | + model="gpt-4.1", |
| 63 | + input=question |
| 64 | + ) |
| 65 | + answer = response.output_text |
| 66 | + |
| 67 | + return render_template('ask_openai.html', question=question, answer=answer) |
0 commit comments