pushing for test

app.py

@@ -1,77 +1,97 @@
-import os
-import gradio as gr
-from dotenv import load_dotenv
-import openai
-from utils import compress
-from google_manager.fassade import Fassade
-
-from description import DESCRIPTION
-
-load_dotenv()
-
-# configuring openai package
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-openai.api_key = OPENAI_API_KEY
+from fastapi import FastAPI, Request, Depends
+from fastapi.responses import RedirectResponse
+from fastapi.responses import HTMLResponse
+from fastapi.staticfiles import StaticFiles
+from google_manager.constants import SCOPES
+from google_manager.auth import get_oauth_url
+from pages.auth_page import html_content
+import google_auth_oauthlib.flow
+
+# from fastapi_sessions import SessionMiddleware, get_session
+from starlette.middleware.sessions import SessionMiddleware
 
+import gradio as gr
+from gpt_summary import Ui
 
-def load_prompt(path):
-    with open(path) as f:
-        lines = f.readlines()
-        return "".join(lines)
 
+app = FastAPI()
+app.add_middleware(SessionMiddleware, secret_key="mysecret")
+app.mount("/assets", StaticFiles(directory="assets"), name="assets")
 
-def chat(passage, max_tokens=256, temprature=0, debug=False):
 
-    if debug:
-        passage = """
-        A car or automobile is a motor vehicle with wheels. Most definitions of cars say that they run primarily on roads, seat one to eight people, have four wheels, and mainly transport people (rather than goods).
-        """
+@app.get("/", response_class=HTMLResponse)
+async def home():
+    return HTMLResponse(content=html_content, status_code=200)
 
-    prompt = load_prompt("summary_prompt.txt").replace("<<SUMMARY>>", passage)
 
-    summary = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        messages=[{"role": "user", "content": prompt}],
+@app.get("/auth")
+async def integreate_google(request: Request):
+    # url = get_oauth_url(SCOPES, "http://127.0.0.1:8000/auth_callback")
+    flow = google_auth_oauthlib.flow.Flow.from_client_secrets_file(
+        "credentials.json", scopes=SCOPES
     )
 
-    return summary["choices"][0]["message"]["content"].strip()
-
-
-def transcribe(audio_file):
-    audio_file = open(audio_file, "rb")
-    transcription = openai.Audio.transcribe("whisper-1", audio_file, language="en")
-    transcription = transcription["text"]
-    return transcription
-
-
-def predict(input, history=[]):
-    compress(input)
-    transcription = transcribe(input)
-    answer = chat(transcription)
+    # flow.redirect_uri = "http://127.0.0.1:8000/gradio/?__theme=dark"
+    # flow.redirect_uri = app.url_path_for(redirect_url)
+    flow.redirect_uri = "http://127.0.0.1:8000/auth_callback"
+
+    # Generate URL for request to Google's OAuth 2.0 server.
+    # Use kwargs to set optional request parameters.
+    authorization_url, state = flow.authorization_url(
+        # Enable offline access so that you can refresh an access token without
+        # re-prompting the user for permission. Recommended for web server apps.
+        access_type="offline",
+        # Enable incremental authorization. Recommended as a best practice.
+        include_granted_scopes="true",
+    )
+    request.session["state"] = state
+    print("authorization_url")
+    print(authorization_url)
+    return RedirectResponse(url=authorization_url)
+
+
+@app.get("/auth_callback")
+async def auth_callback(request: Request):
+    # url = get_oauth_url(SCOPES)
+    print("request")
+    # print(request["code"])
+    # return {"callback": "here"}
+
+    # Specify the state when creating the flow in the callback so that it can
+    # verified in the authorization server response.
+    # if False:
+    state = request.session["state"]
+
+    flow = google_auth_oauthlib.flow.Flow.from_client_secrets_file(
+        "credentials.json", scopes=SCOPES, state=state
+    )
+    flow.redirect_uri = "http://127.0.0.1:8000/auth_callback"
 
-    # upload the input/answer to google drive
-    doc_content = f"""
-user:
-{transcription}
+    # Use the authorization server's response to fetch the OAuth 2.0 tokens.
+    authorization_response = str(request.url)
 
-summary:
-{answer}
-"""
-    Fassade.upload_to_drive(doc_content)
+    flow.fetch_token(authorization_response=authorization_response)
+    if False:
+        # flow.fetch_token(authorization_response=authorization_response)
 
-    history.append((transcription, answer))
-    response = history
-    return response, history
+        # Store credentials in the session.
+        # ACTION ITEM: In a production app, you likely want to save these
+        #              credentials in a persistent database instead.
+        credentials = flow.credentials
+        request.session["credentials"] = credentials_to_dict(credentials)
 
+    return RedirectResponse("/gradio")
 
-with gr.Blocks() as demo:
-    gr.Markdown(DESCRIPTION)
-    chatbot = gr.Chatbot()
-    state = gr.State([])
 
-    with gr.Row():
-        audio_file = gr.Audio(label="Audio", source="microphone", type="filepath")
+app = gr.mount_gradio_app(app, Ui, path="/gradio")
 
-    audio_file.change(predict, [audio_file, state], [chatbot, state])
 
-demo.launch()
+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,
+    }

credentials.json

@@ -1 +1 @@
-{"web":{"client_id":"802510675416-0929ppk1vqugs5nnuiakamt1ugec6qa7.apps.googleusercontent.com","project_id":"gptsummary","auth_uri":"https://accounts.google.com/o/oauth2/auth","token_uri":"https://oauth2.googleapis.com/token","auth_provider_x509_cert_url":"https://www.googleapis.com/oauth2/v1/certs","client_secret":"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","redirect_uris":["http://localhost:7860/"],"javascript_origins":["http://localhost:7860"]}}
+{"web":{"client_id":"802510675416-0929ppk1vqugs5nnuiakamt1ugec6qa7.apps.googleusercontent.com","project_id":"gptsummary","auth_uri":"https://accounts.google.com/o/oauth2/auth","token_uri":"https://oauth2.googleapis.com/token","auth_provider_x509_cert_url":"https://www.googleapis.com/oauth2/v1/certs","client_secret":"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","redirect_uris":["https://zinoubm-clienttest.hf.space/?__theme=dark","http://127.0.0.1:8000/gradio/?__theme=dark","http://127.0.0.1:8000/auth_callback"],"javascript_origins":["https://zinoubm-clienttest.hf.space","http://127.0.0.1:8000"]}}

google_manager/auth.py

@@ -2,6 +2,13 @@ import os.path
 from google.auth.transport.requests import Request
 from google.oauth2.credentials import Credentials
 from google_auth_oauthlib.flow import InstalledAppFlow
+from google_manager.constants import SCOPES
+
+from starlette.responses import RedirectResponse
+from gradio.routes import App
+
+import google.oauth2.credentials
+import google_auth_oauthlib.flow
 
 
 def authenticate(SCOPES):
@@ -26,3 +33,48 @@ def authenticate(SCOPES):
             token.write(creds.to_json())
 
     return creds
+
+
+def get_oauth_url(SCOPES, redirect_url):
+    flow = google_auth_oauthlib.flow.Flow.from_client_secrets_file(
+        "credentials.json", scopes=SCOPES
+    )
+
+    # flow.redirect_uri = "http://127.0.0.1:8000/gradio/?__theme=dark"
+    # flow.redirect_uri = app.url_path_for(redirect_url)
+    flow.redirect_uri = redirect_url
+
+    # Generate URL for request to Google's OAuth 2.0 server.
+    # Use kwargs to set optional request parameters.
+    authorization_url, state = flow.authorization_url(
+        # Enable offline access so that you can refresh an access token without
+        # re-prompting the user for permission. Recommended for web server apps.
+        access_type="offline",
+        # Enable incremental authorization. Recommended as a best practice.
+        include_granted_scopes="true",
+    )
+    print("state")
+    print(state)
+
+    return authorization_url
+
+
+def authenticate_production(SCOPES):
+    creds = None
+
+    if os.path.exists("token.json"):
+        creds = Credentials.from_authorized_user_file("token.json", SCOPES)
+
+    # If there are no (valid) credentials available, let the user log in.
+    if not creds or not creds.valid:
+        if creds and creds.expired and creds.refresh_token:
+            creds.refresh(Request())
+        else:
+            flow = InstalledAppFlow.from_client_secrets_file("credentials.json", SCOPES)
+            creds = flow.run_local_server(port=7200)
+
+        # Save the credentials for the next run
+        with open("token.json", "w") as token:
+            token.write(creds.to_json())
+
+    return creds

google_manager/fassade.py

@@ -1,13 +1,17 @@
-from google_manager.auth import authenticate
+from google_manager.auth import authenticate, authenticate_production
 from google_manager.drive import create_folder, search_folder
 from google_manager.docs import save_doc, move_doc, name_doc
 from google_manager.constants import SCOPES, FOLDER_NAME
 
 
 class Fassade:
+    def __init__(self):
+        self.creds = None
+
     def upload_to_drive(content, FOLDER_NAME=FOLDER_NAME):
         FOLDER_NAME = "GptSummary"
-        creds = authenticate(SCOPES)
+
+        creds = authenticate_production(SCOPES)
         files = search_folder(creds, FOLDER_NAME)
 
         if not files:

gpt_summary.py

@@ -0,0 +1,82 @@
+import os
+import gradio as gr
+from dotenv import load_dotenv
+import openai
+from utils import compress
+from google_manager.fassade import Fassade
+
+
+from description import DESCRIPTION
+
+# fastApi
+
+import fastapi
+import gradio as gr
+
+
+load_dotenv()
+
+# configuring openai package
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+openai.api_key = OPENAI_API_KEY
+
+
+def load_prompt(path):
+    with open(path) as f:
+        lines = f.readlines()
+        return "".join(lines)
+
+
+def chat(passage, max_tokens=256, temprature=0, debug=False):
+
+    if debug:
+        passage = """
+        A car or automobile is a motor vehicle with wheels. Most definitions of cars say that they run primarily on roads, seat one to eight people, have four wheels, and mainly transport people (rather than goods).
+        """
+
+    prompt = load_prompt("summary_prompt.txt").replace("<<SUMMARY>>", passage)
+
+    summary = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[{"role": "user", "content": prompt}],
+    )
+
+    return summary["choices"][0]["message"]["content"].strip()
+
+
+def transcribe(audio_file):
+    audio_file = open(audio_file, "rb")
+    transcription = openai.Audio.transcribe("whisper-1", audio_file, language="en")
+    transcription = transcription["text"]
+    return transcription
+
+
+def predict(input, history=[]):
+    compress(input)
+    transcription = transcribe(input)
+    answer = chat(transcription)
+
+    # upload the input/answer to google drive
+    doc_content = f"""
+user:
+{transcription}
+
+summary:
+{answer}
+"""
+    Fassade.upload_to_drive(doc_content)
+
+    history.append((transcription, answer))
+    response = history
+    return response, history
+
+
+with gr.Blocks() as Ui:
+    gr.Markdown(DESCRIPTION)
+    chatbot = gr.Chatbot()
+    state = gr.State([])
+
+    with gr.Row():
+        audio_file = gr.Audio(label="Audio", source="microphone", type="filepath")
+
+    audio_file.change(predict, [audio_file, state], [chatbot, state])

main.py

@@ -0,0 +1,6 @@
+import uvicorn
+from pathlib import Path
+
+
+if __name__ == "__main__":
+    uvicorn.run(f"app:app", port=8000, reload=True)

pages/auth_page.py

@@ -0,0 +1,54 @@
+html_content = """
+    <html>
+        <head>
+            <title>Button Page</title>
+            <link rel="preconnect" href="https://fonts.googleapis.com">
+            <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
+            <link href="https://fonts.googleapis.com/css2?family=Poppins:wght@600&display=swap" rel="stylesheet">
+            <style>
+            body {
+                background-color: #090F1E;
+                font-family: 'Poppins', sans-serif;
+  
+            }
+            .button-wrapper {
+                height: 100vh;
+                display: flex;
+                justify-content: center;
+                align-items: center;
+            }
+            button {
+                border: none;
+                margin-right: 10px;
+                border-radius: 5px;
+                color: #090F1E;
+                font-weight: 400;
+                font-size: 1rem;
+            }
+            .integrate {
+                display: flex;
+
+                align-items: center;
+            }
+            .skip {
+                color: #95BDFF;
+                text-decoration: none;
+            }
+
+            img {
+                width: 2rem;
+            }
+            </style>
+        </head>
+        <body>
+            <div class="button-wrapper">
+                <div>
+                    <button onclick="location.href='/auth'" class="integrate">
+                        <img src="assets/google_logo.svg"> Integrate Google Cloud
+                    </button>
+                </div>
+                    <a class="skip" href="/gradio?__theme=dark">skip</a>
+            </div>
+        </body>
+    </html>
+    """

requirements.txt

@@ -6,3 +6,4 @@ openai==0.27.0
 protobuf==4.22.0
 python-dotenv==1.0.0
 soundfile==0.12.1
+fastapi-sessions