AikidoSec
diff --git a/‎sample-apps/flask-openai/app.py‎
Lines changed: 1 addition & 40 deletions b/‎sample-apps/flask-openai/app.py‎
Lines changed: 1 addition & 40 deletions
diff --git a/‎sample-apps/flask-openai/templates/create_dog.html‎
Lines changed: 0 additions & 17 deletions b/‎sample-apps/flask-openai/templates/create_dog.html‎
Lines changed: 0 additions & 17 deletions
diff --git a/‎sample-apps/flask-openai/templates/dogpage.html‎
Lines changed: 0 additions & 25 deletions b/‎sample-apps/flask-openai/templates/dogpage.html‎
Lines changed: 0 additions & 25 deletions
diff --git a/‎sample-apps/flask-openai/templates/index.html‎
Lines changed: 3 additions & 4 deletions b/‎sample-apps/flask-openai/templates/index.html‎
Lines changed: 3 additions & 4 deletions
@@ -1,55 +1,16 @@
 import aikido_zen # Aikido package import
 aikido_zen.protect()
 
-import os
 from flask import Flask, render_template, request
-import psycopg2
 import openai
 from pydantic import BaseModel
 
 app = Flask(__name__)
 client = openai.OpenAI()
 
-def get_db_connection():
-    return psycopg2.connect(
-        host="localhost",
-        database="db",
-        user="user",
-        password="password")
-
 @app.route("/")
 def homepage():
-    cursor = get_db_connection().cursor()
-    cursor.execute("SELECT * FROM dogs")
-    dogs = cursor.fetchall()
-    return render_template('index.html', title='Homepage', dogs=dogs)
-
-
-@app.route('/dogpage/<int:dog_id>')
-def get_dogpage(dog_id):
-    cursor = get_db_connection().cursor()
-    cursor.execute("SELECT * FROM dogs WHERE id = " + str(dog_id))
-    dog = cursor.fetchmany(1)[0]
-    return render_template('dogpage.html', title=f'Dog', dog=dog, isAdmin=("Yes" if dog[2] else "No"))
-
-@app.route("/create", methods=['GET'])
-def show_create_dog_form():
-    return render_template('create_dog.html')
-
-@app.route("/create_many", methods=['GET'])
-def show_create_dog_form_many():
-    return render_template('create_dog.html')
-
-@app.route("/create", methods=['POST'])
-def create_dog():
-    dog_name = request.form['dog_name']
-    conn = get_db_connection()
-    cursor =  conn.cursor()
-    cursor.execute(f"INSERT INTO dogs (dog_name, isAdmin) VALUES ('%s', FALSE)" % (dog_name))
-    conn.commit()
-    cursor.close()
-    conn.close()
-    return f'Dog {dog_name} created successfully'
+    return render_template('index.html', title='Homepage')
 
 # OpenAI
 
 
@@ -38,11 +38,10 @@
 </head>
 <body>
     <h1>{{ title }}</h1>
-    <h2>List</h2>
     <ul>
-        {% for dog in dogs %}
-            <li><a href="/dogpage/{{ dog[0] }}">{{ dog[1] }}</a></li>
-        {% endfor %}
+        <li><a href="/openai/responses_create">client.responses.create</a></li>
+        <li><a href="/openai/responses_parse">client.responses.parse</a></li>
+        <li><a href="/openai/chat_completions_create">client.chat.completions.create</a></li>
     </ul>
 </body>
 </html>