adding google docs functionality

app.py

@@ -2,9 +2,8 @@ import os
 import gradio as gr
 from dotenv import load_dotenv
 import openai
-
-# from utils import serialize
 from utils import compress
+from google_manager.fassade import Fassade
 
 from description import DESCRIPTION
 
@@ -38,21 +37,6 @@ def chat(passage, max_tokens=256, temprature=0, debug=False):
     return summary["choices"][0]["message"]["content"].strip()
 
 
-# def chat(message, history):
-#     """
-#     Sends a request to the OpenAi api based on the user input and the history
-#     """
-#     messages = serialize(history)
-#     messages.append({"role": "user", "content": message})
-
-#     completion = openai.ChatCompletion.create(
-#         model="gpt-3.5-turbo",
-#         messages=messages,
-#     )
-
-#     return completion["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")
@@ -63,8 +47,18 @@ def transcribe(audio_file):
 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

credentials.json

@@ -0,0 +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"]}}

google_manager/auth.py

@@ -0,0 +1,28 @@
+import os.path
+from google.auth.transport.requests import Request
+from google.oauth2.credentials import Credentials
+from google_auth_oauthlib.flow import InstalledAppFlow
+
+
+def authenticate(SCOPES):
+    """
+    Request access for the google docs api
+    """
+    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=7860)
+
+        # Save the credentials for the next run
+        with open("token.json", "w") as token:
+            token.write(creds.to_json())
+
+    return creds

google_manager/constants.py

@@ -0,0 +1,7 @@
+SCOPES = [
+    "https://www.googleapis.com/auth/documents",
+    "https://www.googleapis.com/auth/drive.file",
+]
+
+
+FOLDER_NAME = "GptSummary"

google_manager/docs.py

@@ -0,0 +1,71 @@
+import datetime
+from googleapiclient.discovery import build
+from googleapiclient.errors import HttpError
+
+
+def save_doc(creds, title, content):
+    try:
+        service = build("docs", "v1", credentials=creds)
+
+        # create a document
+        title = title
+        body = {"title": title}
+        doc = service.documents().create(body=body).execute()
+        print("Created document with title: {0}".format(doc.get("title")))
+
+        # Get the ID of the new file
+        doc_id = doc["documentId"]
+
+        # Write "Hello World" in the new file
+        doc_content = content
+        service.documents().get(documentId=doc_id).execute()
+        requests = [{"insertText": {"location": {"index": 1}, "text": doc_content}}]
+        result = (
+            service.documents()
+            .batchUpdate(documentId=doc_id, body={"requests": requests})
+            .execute()
+        )
+
+        print(f"A new Google Doc file has been created with ID: {doc_id}")
+        return result
+
+    except HttpError as err:
+        print(err)
+
+
+def move_doc(creds, document_id, folder_id):
+    service = build("drive", "v3", credentials=creds)
+
+    try:
+        # Get the current parents of the document
+        file = service.files().get(fileId=document_id, fields="parents").execute()
+        current_parents = ",".join(file.get("parents"))
+
+        # Move the document to the new folder
+        file = (
+            service.files()
+            .update(
+                fileId=document_id,
+                addParents=folder_id,
+                removeParents=current_parents,
+                fields="id, parents",
+            )
+            .execute()
+        )
+
+        print(
+            f'The document with ID {file.get("id")} was moved to the folder with ID {folder_id}.'
+        )
+
+    except HttpError as error:
+        print(f"An error occurred: {error}")
+
+
+def name_doc():
+    """
+    Gets and format the time to generate document name
+    """
+    now = datetime.datetime.now()
+    timestamp = now.strftime("%Y-%m-%d_%H-%M-%S")
+
+    return f"summary_{timestamp}.txt"

google_manager/docs_api.py

@@ -0,0 +1,61 @@
+# import os.path
+
+# from google.auth.transport.requests import Request
+# from google.oauth2.credentials import Credentials
+# from google_auth_oauthlib.flow import InstalledAppFlow
+# from googleapiclient.discovery import build
+# from googleapiclient.errors import HttpError
+
+# # If modifying these scopes, delete the file token.json.
+# SCOPES = [
+#     "https://www.googleapis.com/auth/documents",
+#     "https://www.googleapis.com/auth/drive.file",
+# ]
+
+
+# def get_auth():
+#     """
+#     Request access for the google docs api
+#     """
+#     creds = None
+#     # The file token.json stores the user's access and refresh tokens, and is
+#     # created automatically when the authorization flow completes for the first
+#     # time.
+#     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=7860)
+#         # Save the credentials for the next run
+#         with open("token.json", "w") as token:
+#             token.write(creds.to_json())
+
+
+# def save_doc(creds, title):
+#     try:
+#         service = build("docs", "v1", credentials=creds)
+
+#         # create a document
+#         title = "My Document"
+#         body = {"title": title}
+#         doc = service.documents().create(body=body).execute()
+#         print("Created document with title: {0}".format(doc.get("title")))
+
+#     except HttpError as err:
+#         print(err)
+
+
+# import datetime
+
+# # Get the current date and time
+# now = datetime.datetime.now()
+
+# # Format the date and time as a string
+# timestamp = now.strftime("%Y-%m-%d_%H-%M-%S")
+
+# # Define the filename with the timestamp
+# filename = f"file_{timestamp}.txt"

google_manager/drive.py

@@ -0,0 +1,59 @@
+from googleapiclient.discovery import build
+from googleapiclient.errors import HttpError
+
+
+def create_folder(creds, folder_name, parent_folder_id=None) -> str:
+    """ """
+    drive_service = build("drive", "v3", credentials=creds)
+
+    # Create a metadata dict with the folder name and parent folder ID (if any)
+    folder_metadata = {
+        "name": folder_name,
+        "mimeType": "application/vnd.google-apps.folder",
+    }
+    if parent_folder_id:
+        folder_metadata["parents"] = [parent_folder_id]
+
+    try:
+        folder = (
+            drive_service.files().create(body=folder_metadata, fields="id").execute()
+        )
+        print(f'Folder "{folder_name}" with ID "{folder["id"]}" created successfully')
+        return folder["id"]
+    except HttpError as error:
+        print(f"An error occurred while creating the Google Drive folder: {error}")
+        return None
+
+
+def search_folder(creds, folder_name):
+    """ """
+    try:
+        # create drive api client
+        service = build("drive", "v3", credentials=creds)
+        files = []
+        page_token = None
+
+        while True:
+            response = (
+                service.files()
+                .list(
+                    q="mimeType='application/vnd.google-apps.folder' and trashed=false",
+                    spaces="drive",
+                    fields="nextPageToken, files(id, name)",
+                    pageToken=page_token,
+                )
+                .execute()
+            )
+            for file in response.get("files", []):
+                # Process change
+                print(f'Found file: {file.get("name")}, {file.get("id")}')
+            files.extend(response.get("files", []))
+            page_token = response.get("nextPageToken", None)
+            if page_token is None:
+                break
+
+    except HttpError as error:
+        print(f"An error occurred: {error}")
+        files = None
+
+    return files

google_manager/fassade.py

@@ -0,0 +1,22 @@
+from google_manager.auth import authenticate
+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 upload_to_drive(content, FOLDER_NAME=FOLDER_NAME):
+        FOLDER_NAME = "GptSummary"
+        creds = authenticate(SCOPES)
+        files = search_folder(creds, FOLDER_NAME)
+
+        if not files:
+            folder_id = create_folder(creds, FOLDER_NAME)
+
+        else:
+            folder_id = files.pop()["id"]
+
+        doc_name = name_doc()
+        doc_response = save_doc(creds, doc_name, content)
+
+        move_doc(creds, doc_response["documentId"], folder_id)

google_manager/test.py

@@ -0,0 +1,9 @@
+from fassade import Fassade
+
+CONTENT = """
+Cars were invented in 1886, when German inventor Carl Benz patented his Benz Patent-Motorwagen.[3][4][5] Cars became widely available during the 20th century. One of the first cars affordable by the masses was the 1908 Model T, an American car manufactured by the Ford Motor Company. Cars were rapidly adopted in the US, where they replaced horse-drawn carriages.[6] In Europe and other parts of the world, demand for automobiles did not increase until after World War II.[7] The car is considered an essential part of the developed economy
+"""
+
+
+if __name__ == "__main__":
+    Fassade.upload_to_drive(CONTENT)

token.json

@@ -0,0 +1 @@
+{"token": "ya29.a0AVvZVsovFtXFOt_y0MPPJ3Ezf3Sgnps6rqvbNu2rM9xG31Hqgnd0vRATIxRBOAp2VHtKCfprshTzijyWLSjYcVDLkxdNxnfUvSSQNi9dAIb-CDXln-fkrG9rhOjIUR-1LtXjvs4X3uV2EuEhTfH8R7rgl_Pg-SkaCgYKAZMSARASFQGbdwaIJJ8fbDLgh97SY8g0zKL0hw0166", "refresh_token": "1//03P_ayTe4UkoXCgYIARAAGAMSNwF-L9IribM55gnIuAF39QKGPVCOLhzhyXBe_4znQdrZr5g6SnVcbk3h11WlxPlh5YR_QRDGmSg", "token_uri": "https://oauth2.googleapis.com/token", "client_id": "802510675416-0929ppk1vqugs5nnuiakamt1ugec6qa7.apps.googleusercontent.com", "client_secret": "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", "scopes": ["https://www.googleapis.com/auth/documents", "https://www.googleapis.com/auth/drive.file"], "expiry": "2023-03-06T17:44:39.266964Z"}