Update app.py

app.py

@@ -1,60 +1,116 @@
-import os
-import requests
-import json
-from io import BytesIO
+# import os
+# import requests
+# import json
+# from io import BytesIO
 
-from flask import Flask, jsonify, render_template, request, send_file
+# from flask import Flask, jsonify, render_template, request, send_file
 
-from modules.inference import infer_t5
-from modules.dataset import query_emotion
+# from modules.inference import infer_t5
+# from modules.dataset import query_emotion
 
-# https://huggingface.co/settings/tokens
-# https://huggingface.co/spaces/{username}/{space}/settings
-API_TOKEN = os.getenv("BIG_GAN_TOKEN")
+# # https://huggingface.co/settings/tokens
+# # https://huggingface.co/spaces/{username}/{space}/settings
+# API_TOKEN = os.getenv("BIG_GAN_TOKEN")
 
-app = Flask(__name__)
+# app = Flask(__name__)
 
 
-@app.route("/")
-def index():
-    return render_template("index.html")
+# @app.route("/")
+# def index():
+#     return render_template("index.html")
 
 
-@app.route("/infer_biggan")
-def biggan():
-    input = request.args.get("input")
+# @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),
-    )
+#     output = requests.request(
+#         "POST",
+#         "https://api-inference.huggingface.co/models/osanseviero/BigGAN-deep-128",
+#         headers={"Authorization": f"Bearer {API_TOKEN}"},
+#         data=json.dumps(input),
+#     )
 
-    return send_file(BytesIO(output.content), mimetype="image/png")
+#     return send_file(BytesIO(output.content), mimetype="image/png")
 
 
-@app.route("/infer_t5")
-def t5():
-    input = request.args.get("input")
+# @app.route("/infer_t5")
+# def t5():
+#     input = request.args.get("input")
 
-    output = infer_t5(input)
+#     output = infer_t5(input)
 
-    return jsonify({"output": output})
+#     return jsonify({"output": output})
 
 
-@app.route("/query_emotion")
-def emotion():
-    start = request.args.get("start")
-    end = request.args.get("end")
+# @app.route("/query_emotion")
+# def emotion():
+#     start = request.args.get("start")
+#     end = request.args.get("end")
 
-    print(start)
-    print(end)
+#     print(start)
+#     print(end)
 
-    output = query_emotion(int(start), int(end))
+#     output = query_emotion(int(start), int(end))
 
-    return jsonify({"output": output})
+#     return jsonify({"output": output})
 
 
+# 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():
+    code = request.args.get("code")
+    token_response = grequests.post("https://oauth2.googleapis.com/token", 
+                                    data={
+                                        "code": code,
+                                        "client_id": CLIENT_ID,
+                                        "client_secret": CLIENT_SECRET,
+                                        "redirect_uri": "http://localhost:3000/callback",
+                                        "grant_type": "authorization_code"
+                                    })
+    token_data = token_response.json()
+    access_token = token_data.get("access_token")
+    id_info = id_token.verify_oauth2_token(token_data.get("id_token"), grequests.Request(), CLIENT_ID)
+
+    # Store user information in session
+    session['user'] = id_info
+
+    return redirect(url_for("profile"))
+
+@app.route("/profile")
+def profile():
+    user = session.get('user')
+    if user:
+        return render_template("profile.html", user=user)
+    else:
+        return "User not authenticated"
+
+@app.route("/logout")
+def logout():
+    session.pop('user', None)
+    return redirect(url_for("index"))
+
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+    app.run(debug=True)
+