Update app.py

app.py

@@ -1,60 +1,53 @@
+from flask import Flask, render_template, request
+import config
 import os
-import requests
-import json
-from io import BytesIO
+import openai
 
-from flask import Flask, jsonify, render_template, request, send_file
 
-from modules.inference import infer_t5
-from modules.dataset import query_emotion
+#creating the 404 page (Optional)
+def page_not_found(e):
+  return render_template('404.html'), 404
 
-# https://huggingface.co/settings/tokens
-# https://huggingface.co/spaces/{username}/{space}/settings
-API_TOKEN = os.getenv("BIG_GAN_TOKEN")
 
-app = Flask(__name__)
 
+##Initialising FLAK
+app = Flask(__name__)
+app.config.from_object(config.config['development'])
+app.register_error_handler(404, page_not_found)
 
-@app.route("/")
-def index():
-    return render_template("index.html")
+### Initialise the OPENAI library with the key saved in the CONFIG file
+openai.api_key = app.config['OPENAI_KEY']
 
 
-@app.route("/infer_biggan")
-def biggan():
-    input = request.args.get("input")
 
-    output = requests.request(
-        "POST",
-        "https://api-inference.huggingface.co/models/osanseviero/BigGAN-deep-128",
-        headers={"Authorization": f"Bearer {API_TOKEN}"},
-        data=json.dumps(input),
-    )
+#####----------START FUNCTIONS--------------------------------------------------------------------
+def createImageFromPrompt(prompt):
+    response = openai.Image.create(prompt=prompt, n=2, size="512x512")
+    return response['data']
 
-    return send_file(BytesIO(output.content), mimetype="image/png")
 
+#####----------END FUNCTIONS--------------------------------------------------------------------
 
-@app.route("/infer_t5")
-def t5():
-    input = request.args.get("input")
 
-    output = infer_t5(input)
 
-    return jsonify({"output": output})
 
+##View Functions
+@app.route('/', methods=["GET", "POST"])
+def index():
 
-@app.route("/query_emotion")
-def emotion():
-    start = request.args.get("start")
-    end = request.args.get("end")
+    if request.method == 'POST':
+        images = []
+        prompt = request.form['prompt']
+        res = createImageFromPrompt(prompt)
 
-    print(start)
-    print(end)
+        if len(res) > 0:
+            for img in res:
+                images.append(img['url'])
 
-    output = query_emotion(int(start), int(end))
+    return render_template('index.html', **locals())
 
-    return jsonify({"output": output})
 
 
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+#Run Flask
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port='8000')