Update app.py

app.py

@@ -1,67 +1,64 @@
-import requests
-from flask import Flask, request, redirect, url_for, session
+import uvicorn
+from fastapi import FastAPI, HTTPException, Request, Depends, BackgroundTasks
+from fastapi.responses import HTMLResponse, RedirectResponse
+from fastapi.templating import Jinja2Templates
+from fastapi.security import OAuth2AuthorizationCodeBearer, OAuth2AuthorizeRequest
 from google.oauth2 import id_token
 from google.auth.transport import requests as grequests
 
-app = Flask(__name__)
-app.secret_key = "your_secret_key"
+app = FastAPI(debug=True)
+templates = Jinja2Templates(directory="templates")
 
 CLIENT_ID = "your_google_client_id"
+CLIENT_SECRET = "your_google_client_secret"
+REDIRECT_URI = "http://localhost:7860/auth/google/callback"
 
-# Google API endpoints
-DISCOVERY_URL = "https://accounts.google.com/.well-known/openid-configuration"
-TOKEN_ENDPOINT = "https://oauth2.googleapis.com/token"
-DOCS_API_ENDPOINT = "https://docs.googleapis.com/v1/documents/{documentId}"
+oauth2_scheme = OAuth2AuthorizationCodeBearer(
+    authorizationUrl="https://accounts.google.com/o/oauth2/auth",
+    tokenUrl="https://oauth2.googleapis.com/token",
+)
 
-@app.route("/")
-def index():
-    return "Welcome to the Google Docs Shiny app!"
+@app.get("/")
+async def index(request: Request):
+    return templates.TemplateResponse("index.html", {"request": request})
 
-@app.route("/login")
-def login():
-    # Generate Google login URL
-    authorization_endpoint = requests.get(DISCOVERY_URL).json()["authorization_endpoint"]
-    redirect_uri = request.base_url + "/callback"
-    login_url = f"{authorization_endpoint}?response_type=code&client_id={CLIENT_ID}&redirect_uri={redirect_uri}&scope=email%20profile%20openid&access_type=offline"
-    return redirect(login_url)
+@app.get("/auth/google")
+async def auth_google(request: Request):
+    google_auth_url = (
+        f"https://accounts.google.com/o/oauth2/auth"
+        f"?client_id={CLIENT_ID}"
+        f"&redirect_uri={REDIRECT_URI}"
+        f"&response_type=code"
+        f"&scope=email%20profile%20openid"
+    )
+    return RedirectResponse(url=google_auth_url)
 
-@app.route("/login/callback")
-def login_callback():
-    # Exchange authorization code for access token
-    code = request.args.get("code")
+@app.get("/auth/google/callback")
+async def auth_google_callback(code: str):
     token_request_data = {
         "code": code,
         "client_id": CLIENT_ID,
-        "client_secret": "your_client_secret",
-        "redirect_uri": request.base_url,
+        "client_secret": CLIENT_SECRET,
+        "redirect_uri": REDIRECT_URI,
         "grant_type": "authorization_code"
     }
-    token_response = requests.post(TOKEN_ENDPOINT, data=token_request_data)
+    token_response = grequests.post("https://oauth2.googleapis.com/token", data=token_request_data)
     access_token = token_response.json().get("access_token")
-    
-    if access_token:
-        session["access_token"] = access_token
-        return redirect(url_for("fetch_comments"))
-    else:
-        return "Failed to obtain access token."
+    # Store access_token in session or database for future use
+    return {"access_token": access_token}
 
-@app.route("/fetch_comments")
-def fetch_comments():
-    if "access_token" in session:
-        # Fetch comments from Google Docs API
-        document_id = "your_document_id"
-        headers = {"Authorization": f"Bearer {session['access_token']}"}
-        response = requests.get(DOCS_API_ENDPOINT.format(documentId=document_id), headers=headers)
-        
-        if response.status_code == 200:
-            # Parse comments from response
-            comments = response.json().get("comments", [])
-            return str(comments)
-        else:
-            return f"Failed to fetch comments. Status code: {response.status_code}"
+@app.get("/fetch_comments")
+async def fetch_comments(access_token: str = Depends(oauth2_scheme)):
+    # Fetch comments from Google Docs API using access_token
+    document_id = "your_document_id"
+    headers = {"Authorization": f"Bearer {access_token}"}
+    response = grequests.get(f"https://docs.googleapis.com/v1/documents/{document_id}", headers=headers)
+    
+    if response.status_code == 200:
+        comments = response.json().get("comments", [])
+        return comments
     else:
-        return "User not authenticated."
+        raise HTTPException(status_code=response.status_code, detail="Failed to fetch comments")
 
 if __name__ == "__main__":
-    app.run(debug=True)
-
+    uvicorn.run(app, host="0.0.0.0", port=7860)