Update app.py

app.py

@@ -4,6 +4,10 @@ import json
 from io import BytesIO
 
 from flask import Flask, jsonify, render_template, request, send_file
+from flask import Flask, redirect, url_for, session, request, jsonify, render_template
+from google.oauth2 import id_token
+from google.auth.transport import requests as grequests
+import os
 
 from modules.inference import infer_t5
 from modules.dataset import query_emotion
@@ -12,6 +16,11 @@ from modules.dataset import query_emotion
 # https://huggingface.co/spaces/{username}/{space}/settings
 API_TOKEN = os.getenv("BIG_GAN_TOKEN")
 
+app.secret_key = os.urandom(24)
+
+CLIENT_ID = "your_google_client_id"
+CLIENT_SECRET = os.environ.get("GOOGLE_CLIENT_SECRET")
+
 app = Flask(__name__)
 
 
@@ -19,6 +28,10 @@ app = Flask(__name__)
 def index():
     return render_template("index.html")
 
+@app.route("/login")
+def login():
+    return redirect("https://accounts.google.com/o/oauth2/auth?response_type=code&client_id={}&redirect_uri=http://localhost:3000/callback&scope=email%20profile&access_type=offline".format(CLIENT_ID))
+
 
 # @app.route("/infer_biggan")
 # def biggan():
@@ -59,24 +72,8 @@ def index():
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)
 
-# from flask import Flask, redirect, url_for, session, request, jsonify, render_template
-# from google.oauth2 import id_token
-# from google.auth.transport import requests as grequests
-# import os
-
-# app = Flask(__name__)
-# app.secret_key = os.urandom(24)
-
-# CLIENT_ID = "your_google_client_id"
-# CLIENT_SECRET = os.environ.get("GOOGLE_CLIENT_SECRET")
 
-# @app.route("/")
-# def index():
-#     return render_template("index.html")
 
-# @app.route("/login")
-# def login():
-#     return redirect("https://accounts.google.com/o/oauth2/auth?response_type=code&client_id={}&redirect_uri=http://localhost:5000/callback&scope=email%20profile&access_type=offline".format(CLIENT_ID))
 
 # @app.route("/callback")
 # def callback():