Update app.py

app.py

@@ -18,7 +18,9 @@ import warnings
 from PIL import Image
 from stability_sdk import client
 import stability_sdk.interfaces.gooseai.generation.generation_pb2 as generation
-import datetime
+from datetime import datetime
+from google.oauth2 import service_account
+from googleapiclient.discovery import build
 
 stability_api = client.StabilityInference(
     key=st.secrets["STABILITY_KEY"], #os.environ("STABILITY_KEY"), # key=os.environ['STABILITY_KEY'], # API Key reference.
@@ -72,12 +74,42 @@ def search_internet(question):
         presence_penalty=0)
 
     string_temp = response.choices[0].text
-
     st.write(string_temp)
     st.write(snippets)
 
 # openai.api_key = ""
 openai.api_key = st.secrets["OPENAI_KEY"] #os.environ("OPENAI_KEY") #os.environ['OPENAI_KEY']
+date_time = str(datetime.now())
+
+def g_sheet_log(myinput, output):
+  SERVICE_ACCOUNT_FILE = '/content/gsearch-376307-f16abe212f26.json'
+
+  credentials = service_account.Credentials.from_service_account_file(
+      filename=SERVICE_ACCOUNT_FILE
+  )
+
+  service_sheets = build('sheets', 'v4', credentials=credentials)
+
+  GOOGLE_SHEETS_ID = '16cM8lHm7n_X0ZVLgWfL5fcBhvKWIGO9LQz3zCl2Dn_8'
+  worksheet_name = 'Prompt_Logs!'
+  cell_range_insert = 'A:C'
+
+  values = (
+      (myinput, output, date_time),
+  )
+
+  value_range_body = {
+      'majorDimension' : 'ROWS',
+      'values' : values
+  }
+
+  service_sheets.spreadsheets().values().append(
+      spreadsheetId=GOOGLE_SHEETS_ID,
+      valueInputOption='USER_ENTERED',
+      range=worksheet_name + cell_range_insert,
+      body=value_range_body
+  ).execute()
+
 
 def openai_response(PROMPT):
     response = openai.Image.create(
@@ -173,6 +205,9 @@ if Input_type == 'TEXT':
                             img = Image.open(io.BytesIO(artifact.binary))
                             st.image(img)
                             img.save(str(artifact.seed)+ ".png") # Save our generated images with their seed number as the filename.
+                            rx = 'Image returned'
+                            g_sheet_log(mytext, rx)
+                            
             except:        
                 st.write('image is being generated please wait...')
                 def extract_image_description(input_string):
@@ -205,13 +240,18 @@ if Input_type == 'TEXT':
             OurURL = YoutubeURL + video_id
             st.write(OurURL)
             st_player(OurURL)
+            ry = 'Youtube link and video returned'
+            g_sheet_log(mytext, ry)
 
-        elif ("don't" in string_temp or "internet" in string_temp  ):
+        elif ("don't" in string_temp or "internet" in string_temp):
             st.write('searching internet ')
             search_internet(question)
+            rz = 'Internet result returned'
+            g_sheet_log(mytext, rz)
 
         else:
             st.write(string_temp)
+            g_sheet_log(mytext, string_temp)
     
 elif Input_type == 'SPEECH':
     stt_button = Button(label="Speak", width=100)