Update app.py

app.py

@@ -1,23 +1,28 @@
 import gradio as gr
 import os
-import csv
-import requests
+import gspread
 from datetime import datetime
 
-# Install the groq package if it is not installed
+# 安裝並導入 groq 套件
 try:
     from groq import Groq
 except ImportError:
     os.system('pip install groq')
     from groq import Groq
 
-# Set up the Groq client with the secret key
+# 設定 Groq 的 API Key
 groq_key = os.getenv('groq_key')
 if not groq_key:
     raise ValueError("groq_key environment variable is not set")
 
 client = Groq(api_key=groq_key)
 
+# 使用 gspread 連接 Google Sheet
+sheet_url = "https://docs.google.com/spreadsheets/d/1BQ23fiiGu5wN4RmRC4XmrKS6qCWHuymlbsoLEEmEN_4/edit?usp=sharing"
+gc = gspread.service_account()  # 無需 Google Console 憑證時，可以跳過此步驟
+sheet = gc.open_by_url(sheet_url).sheet1  # 獲取第一個工作表
+
+# 定義簡單的 Chatbot
 class SimpleChatBot:
     def __init__(self):
         self.initial_prompt = [
@@ -49,38 +54,28 @@ class SimpleChatBot:
 
 chatbot = SimpleChatBot()
 
-# Function to log chat history to Google Sheet
+# 記錄聊天內容到 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}")
+        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        sheet.append_row([timestamp, user_message, bot_response])
+        print("Logged to Google Sheet successfully.")
     except Exception as e:
-        print(f"Error while logging to Google Sheet: {e}")
+        print(f"Error 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
+    # 記錄到 Google Sheet
     log_to_sheet(message, response)
     
     return chat_history, ""
 
-# Gradio interface
+# 建立 Gradio 界面
 with gr.Blocks(title="簡單的Gradio聊天機器人") as demo:
     gr.Markdown("# 簡單的Gradio聊天機器人")