Update app.py

app.py

@@ -1,5 +1,8 @@
 import gradio as gr
 import os
+import csv
+import requests
+from datetime import datetime
 
 # Install the groq package if it is not installed
 try:
@@ -46,13 +49,38 @@ class SimpleChatBot:
 
 chatbot = SimpleChatBot()
 
+# Function to log chat history to Google Sheet
+def log_to_sheet(user_message, bot_response):
+    # Prepare the URL for Google Sheets API using the shared link
+    sheet_url = "https://docs.google.com/spreadsheets/d/1BQ23fiiGu5wN4RmRC4XmrKS6qCWHuymlbsoLEEmEN_4/gviz/tq?tqx=out:csv"
+    
+    # Prepare the data row
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    row = [timestamp, user_message, bot_response]
+    
+    # Append the row to the sheet using HTTP requests
+    try:
+        response = requests.post(sheet_url, data={'append': row})
+        if response.status_code == 200:
+            print("Logged to Google Sheet successfully.")
+        else:
+            print(f"Failed to log to Google Sheet: {response.status_code}")
+    except Exception as e:
+        print(f"Error while logging to Google Sheet: {e}")
+
+# Chatbot response function with logging
 def respond(message, chat_history):
     chat_history = [{"role": entry["role"], "content": entry["content"]} for entry in chat_history]
     response = chatbot.get_response(message, chat_history)
     chat_history.append({"role": "user", "content": message})
     chat_history.append({"role": "assistant", "content": response})
+    
+    # Log to Google Sheet
+    log_to_sheet(message, response)
+    
     return chat_history, ""
 
+# Gradio interface
 with gr.Blocks(title="簡單的Gradio聊天機器人") as demo:
     gr.Markdown("# 簡單的Gradio聊天機器人")