Create app.py

app.py

@@ -0,0 +1,289 @@
+import os
+
+from flask import Flask, request, jsonify
+
+import openai
+
+
+
+app = Flask(__name__)
+
+
+
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+
+
+global requests_list
+
+requests_list = []
+
+
+
+@app.route("/")
+
+def home():
+
+    return jsonify(requests_list)
+
+
+
+@app.route("/api/completions", methods=["POST"])
+
+def completions():
+
+    payload = request.json
+
+    requests_list.append(payload)
+
+    response = openai.Completion.create(
+
+        model=payload["model"],
+
+        prompt=payload["prompt"],
+
+        max_tokens=payload.get("max_tokens", 100),
+
+        temperature=payload.get("temperature", 0.8),
+
+    )
+
+    return jsonify(response.choices[0].text.strip())
+
+
+
+@app.route("/api/chat/completions", methods=["POST"])
+
+def chat():
+
+    payload = request.json
+
+    requests_list.append(payload)
+
+    response = openai.ChatCompletion.create(
+
+        model=payload["model"],
+
+        messages=payload["messages"]
+
+    )
+
+    return jsonify(response)
+
+
+
+@app.route("/api/edits", methods=["POST"])
+
+def edits():
+
+    payload = request.json
+
+    requests_list.append(payload)
+
+    response = openai.Edit.create(
+
+        model=payload["model"],
+
+        input=payload["input"],
+
+        instruction=payload["instruction"]
+
+    )
+
+    return jsonify(response.choices[0].text)
+
+
+
+@app.route('/api/audio/transcriptions', methods=['POST'])
+
+def transcriptions():
+
+    audio_file = request.files['file']
+
+    request_info = {
+
+        'filename': audio_file.filename,
+
+        'content_type': audio_file.content_type,
+
+        'headers': dict(request.headers)
+
+    }
+
+    requests_list.append(request_info)
+
+    
+
+    # save to a temporary file
+
+    tmp_file_name = 'temp_audio.wav'
+
+    audio_file.save(tmp_file_name)
+
+
+
+    # use the saved file for transcription
+
+    with open(tmp_file_name, 'rb') as f:
+
+        response = openai.Audio.transcribe("whisper-1", f)
+
+
+
+    # delete the temporary file
+
+    os.remove(tmp_file_name)
+
+    
+
+    return jsonify(response)
+
+
+
+@app.route("/api/images", methods=["POST"])
+
+def images():
+
+    payload = request.json
+
+    requests_list.append(payload)
+
+    response = openai.Image.create(
+
+        prompt=payload["prompt"],
+
+        n=payload.get("n", 1),
+
+        size=payload.get("size", "512x512")
+
+    )
+
+    return jsonify(response)
+
+
+
+@app.route("/api/images/edits", methods=["POST"])
+
+def image_edits():
+
+    image_file = request.files["image"]
+
+    prompt = request.form["prompt"]
+
+    n = int(request.form.get("n", 1))
+
+    size = request.form.get("size", "1024x1024")
+
+    response_format = request.form.get("response_format", "url")
+
+
+
+    request_info = {
+
+        'filename': image_file.filename,
+
+        'content_type': image_file.content_type,
+
+        'headers': dict(request.headers),
+
+        'prompt': prompt,
+
+        'n': n,
+
+        'size': size,
+
+        'response_format': response_format
+
+    }
+
+    requests_list.append(request_info)
+
+
+
+    response = openai.Image.create_edit(
+
+        image=image_file,
+
+        prompt=prompt,
+
+        n=n,
+
+        size=size,
+
+        response_format=response_format
+
+    )
+
+
+
+    return jsonify(response)
+
+
+
+@app.route("/api/images/variations", methods=["POST"])
+
+def image_variations():
+
+    image_file = request.files["image"]
+
+    n = request.form.get("n", 1)
+
+    size = request.form.get("size", "1024x1024")
+
+
+
+    request_info = {
+
+        'filename': image_file.filename,
+
+        'content_type': image_file.content_type,
+
+        'headers': dict(request.headers),
+
+        'n': n,
+
+        'size': size
+
+    }
+
+    requests_list.append(request_info)
+
+
+
+    response = openai.Image.create_variation(
+
+        image=image_file,
+
+        n=int(n),
+
+        size=size
+
+    )
+
+    return jsonify(response)
+
+
+
+@app.route("/api/moderations", methods=["POST"])
+
+def moderations():
+
+    payload = request.json
+
+    requests_list.append(payload)
+
+    response = openai.Moderation.create(
+
+        input=payload["input"]
+
+    )
+
+    return jsonify(response.results)
+
+
+
+if __name__ == "__main__":
+
+    port = int(os.environ.get("PORT", 5000))
+
+    app.run(host="0.0.0.0", port=port)