Update run.py

run.py

@@ -3,7 +3,21 @@ import os
 import gradio as gr
 import json
 from huggingface_hub import InferenceClient
+import gspread
+from oauth2client.service_account import ServiceAccountCredentials
 
+# Google Sheets setup
+scope = ["https://spreadsheets.google.com/feeds", "https://www.googleapis.com/auth/drive"]
+creds = ServiceAccountCredentials.from_json_keyfile_name('/home/user/app/chromaold/credentials.json', scope)
+client = gspread.authorize(creds)
+sheet = client.open("nestolechatbot").sheet1  # Open the sheet
+
+def save_to_sheet(date,name, message):
+    # Write user input to the Google Sheet
+    sheet.append_row([date,name, message])
+    return f"Thanks {name}, your message has been saved!"
+
+    
 path='/Users/thiloid/Desktop/LSKI/ole_nest/Chatbot/LLM/chromaTS'
 if(os.path.exists(path)==False): path="/home/user/app/chromaTS"
 
@@ -75,6 +89,7 @@ def response(
     output = ""
     for response in stream:
         output += response.token.text
+        save_to_sheet("date",prompt, output)
         yield output
     #output=output+"\n\n<br><details open><summary><strong>Sources</strong></summary><br><ul>"+ "".join(["<li>" + s + "</li>" for s in combination])+"</ul></details>"
     yield output