Create server.py

server.py

@@ -0,0 +1,64 @@
+from flask import Flask, redirect, request, session, jsonify, send_from_directory
+import requests
+import os
+
+app = Flask(__name__)
+app.secret_key = os.getenv("SECRET_KEY", "supersecret")  # for sessions
+
+# Google OAuth config
+GOOGLE_CLIENT_ID = os.getenv("GOOGLE_CLIENT_ID")
+GOOGLE_CLIENT_SECRET = os.getenv("GOOGLE_CLIENT_SECRET")
+REDIRECT_URI = f"https://{os.getenv('SPACE_ID')}.hf.space/google/auth/callback"
+
+@app.route("/")
+def index():
+    return send_from_directory(".", "frontend.html")
+
+@app.route("/google/login")
+def google_login():
+    google_auth_url = (
+        "https://accounts.google.com/o/oauth2/v2/auth"
+        "?response_type=code"
+        f"&client_id={GOOGLE_CLIENT_ID}"
+        f"&redirect_uri={REDIRECT_URI}"
+        "&scope=openid%20email%20profile"
+    )
+    return redirect(google_auth_url)
+
+@app.route("/google/auth/callback")
+def google_callback():
+    code = request.args.get("code")
+    if not code:
+        return "Error: No code provided", 400
+
+    # Exchange code for tokens
+    token_url = "https://oauth2.googleapis.com/token"
+    data = {
+        "code": code,
+        "client_id": GOOGLE_CLIENT_ID,
+        "client_secret": GOOGLE_CLIENT_SECRET,
+        "redirect_uri": REDIRECT_URI,
+        "grant_type": "authorization_code",
+    }
+
+    r = requests.post(token_url, data=data)
+    tokens = r.json()
+
+    # Get user info
+    user_info = requests.get(
+        "https://www.googleapis.com/oauth2/v2/userinfo",
+        headers={"Authorization": f"Bearer {tokens['access_token']}"}
+    ).json()
+
+    session["user"] = user_info
+    return redirect("/")
+
+@app.route("/api/user")
+def get_user():
+    user = session.get("user")
+    if user:
+        return jsonify(user)
+    return jsonify({"error": "Not logged in"}), 401
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)