app.py

from fastapi import FastAPI, Request
from fastapi.responses import RedirectResponse, HTMLResponse
from google.oauth2.credentials import Credentials
from fastapi.staticfiles import StaticFiles
from google_manager.constants import SCOPES
from pages.load_page import load_page
import google_auth_oauthlib.flow
from starlette.middleware.sessions import SessionMiddleware
import gradio as gr
from gpt_summary import Ui

# Create an instance of the FastAPI app
app = FastAPI()

# Add Session middleware to the app
app.add_middleware(SessionMiddleware, secret_key="mysecret")

# Mount the static files directory
app.mount("/assets", StaticFiles(directory="assets"), name="assets")


@app.get("/", response_class=HTMLResponse)
async def home(request: Request):
    # Load the html content of the auth page
    html_content = load_page("./pages/auth_page.html")
    # print("request session")
    # print(request.session)
    # creds = request.session.get("credentials", None)
    creds = request.session.get("credentials", {})

    if creds:
        try:
            creds = Credentials.from_authorized_user_info(info=creds)

        except Exception as e:
            raise Exception(
                f"Invalid credentials in session with the following error{e}"
            )

    if not creds or not creds.valid:
        return HTMLResponse(content=html_content, status_code=200)
    else:
        return RedirectResponse("/gradio/?__theme=dark")


# Define the endpoint for Google authentication
@app.get("/auth")
async def integrate_google(request: Request):
    # print(SCOPES)
    flow = google_auth_oauthlib.flow.Flow.from_client_secrets_file(
        "credentials.json", scopes=SCOPES
    )
    # flow.redirect_uri = "https://gpt-summary-u8pr.onrender.com/auth_callback"
    # flow.redirect_uri = "http://127.0.0.1:8000/auth_callback"
    flow.redirect_uri = (
        request.url.scheme
        + "://"
        + request.url.hostname
        + (":" + str(request.url.port) if request.url.port else "")
        + app.url_path_for("auth_callback")
    )
    authorization_url, state = flow.authorization_url(
        access_type="offline",
        include_granted_scopes="true",
    )
    request.session["state"] = state
    return RedirectResponse(url=authorization_url)


# Define the callback endpoint after successful authentication
@app.get("/auth_callback")
async def auth_callback(request: Request):
    state = request.session["state"]
    flow = google_auth_oauthlib.flow.Flow.from_client_secrets_file(
        "credentials.json", scopes=SCOPES, state=state
    )
    print(request.url.hostname)
    # if request.url.hostname == "0.0.0.0":
    #     flow.redirect_uri = "https://gpt-summary-u8pr.onrender.com/auth_callback"
    # else:
    #     flow.redirect_uri = (
    #         request.url.scheme
    #         + "://"
    #         + request.url.hostname
    #         + (":" + str(request.url.port) if request.url.port else "")
    #         + app.url_path_for("auth_callback")
    #     )
    # flow.redirect_uri = "https://gpt-summary-u8pr.onrender.com/auth_callback"
    # flow.redirect_uri = "http://127.0.0.1:8000/auth_callback"
    flow.redirect_uri = (
        request.url.scheme
        + "://"
        + request.url.hostname
        + (":" + str(request.url.port) if request.url.port else "")
        + app.url_path_for("auth_callback")
    )

    # Use the authorization server's response to fetch the OAuth 2.0 tokens.
    authorization_response = str(request.url)
    flow.fetch_token(authorization_response=authorization_response)
    credentials = flow.credentials
    request.session["credentials"] = credentials_to_dict(credentials)
    return RedirectResponse("/gradio?__theme=dark")


# Mount the Gradio UI
app = gr.mount_gradio_app(app, Ui, path="/gradio")

# Define a helper function to convert credentials to dictionary
def credentials_to_dict(credentials):
    return {
        "token": credentials.token,
        "refresh_token": credentials.refresh_token,
        "token_uri": credentials.token_uri,
        "client_id": credentials.client_id,
        "client_secret": credentials.client_secret,
        "scopes": credentials.scopes,
    }