to gsheet

app.py

@@ -2,19 +2,19 @@
 from __future__ import annotations
 import asyncio
 import datetime
+import pytz
 import logging
 import os
 from enum import Enum
-import requests
 import json
 import uuid
 from pydantic import BaseModel
+import gspread
 
 from copy import deepcopy
 from typing import Any, Dict, List, Optional, Tuple, Union
 
 import gradio as gr
-import pytz
 import tiktoken
 
 # from dotenv import load_dotenv
@@ -46,6 +46,10 @@ CLAUDE_2_CONTEXT_LENGTH = 100000  # need to use claude tokenizer
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY")
 HF_TOKEN = os.getenv("HF_TOKEN")
+GS_CREDS = json.loads(rf"""{os.getenv("GSPREAD_SERVICE")}""")
+GSHEET_ID = os.getenv("GSHEET_ID")
+AUTH_GSHEET_NAME = os.getenv("AUTH_GSHEET_NAME")
+TURNS_GSHEET_NAME = os.getenv("TURNS_GSHEET_NAME")
 
 theme = gr.themes.Soft()
 
@@ -56,6 +60,26 @@ gradio_flagger = gr.HuggingFaceDatasetSaver(
 )
 
 
+def get_gsheet_rows(
+    sheet_id: str, sheet_name: str, creds: Dict[str, str]
+) -> List[Dict[str, str]]:
+    gc = gspread.service_account_from_dict(creds)
+    worksheet = gc.open_by_key(sheet_id).worksheet(sheet_name)
+    rows = worksheet.get_all_records()
+    return rows
+
+
+def append_gsheet_rows(
+    sheet_id: str,
+    rows: List[List[str]],
+    sheet_name: str,
+    creds: Dict[str, str],
+) -> None:
+    gc = gspread.service_account_from_dict(creds)
+    worksheet = gc.open_by_key(sheet_id).worksheet(sheet_name)
+    worksheet.append_rows(values=rows, insert_data_option="INSERT_ROWS")
+
+
 class ChatSystemMessage(str, Enum):
     CASE_SYSTEM_MESSAGE = """You are a helpful AI assistant for a Columbia Business School MBA student.
     Follow this message's instructions carefully. Respond using markdown.
@@ -132,17 +156,23 @@ def reset_textbox():
 
 
 def auth(username, password):
-    auth_endpoint = "https://worker.johnnyclee.com/auth"
     try:
-        auth_payload = {username: password}
-        auth_response = requests.post(
-            auth_endpoint,
-            json=auth_payload,
-            timeout=3,
+        auth_records = get_gsheet_rows(
+            sheet_id=GSHEET_ID, sheet_name=AUTH_GSHEET_NAME, creds=GS_CREDS
         )
-        auth_response.raise_for_status()
-        return auth_response.status_code == 200
+        auth_dict = {user["username"]: user["password"] for user in auth_records}
+        search_auth_user = auth_dict.get(username)
+        if search_auth_user:
+            autheticated = search_auth_user == password
+            if autheticated:
+                LOG.info(f"{username} successfully logged in.")
+                return autheticated
+        else:
+            LOG.info(f"{username} failed to login.")
+            return False
+
     except Exception as exc:
+        LOG.info(f"{username} failed to login")
         LOG.error(exc)
     return (username, password) in creds
 
@@ -431,18 +461,31 @@ async def respond(
         LOG.info(f"""[{request.username}] ENDING CHAIN""")
         LOG.debug(f"History: {state.history}")
         LOG.debug(f"Memory: {state.chain.memory.json()}")
+        current_timestamp = datetime.datetime.now(pytz.timezone("US/Eastern")).replace(
+            tzinfo=None
+        )
+        timestamp_string = current_timestamp.strftime("%Y-%m-%d %H:%M:%S")
         data_to_flag = (
             {
                 "history": deepcopy(state.history),
                 "username": request.username,
-                "timestamp": datetime.datetime.now(datetime.timezone.utc).isoformat(),
+                "timestamp": timestamp_string,
                 "session_id": state.session_id,
                 "metadata": state.chain.metadata,
                 "langsmith_url": langsmith_url,
             },
         )
-        LOG.debug(f"Data to flag: {data_to_flag}")
         gradio_flagger.flag(flag_data=data_to_flag, username=request.username)
+        (flagged_data,) = data_to_flag
+        metadata_to_gsheet = flagged_data.get("metadata").values()
+        gsheet_row = [[timestamp_string, *metadata_to_gsheet]]
+        LOG.info(f"Data to GSHEET: {gsheet_row}")
+        append_gsheet_rows(
+            sheet_id=GSHEET_ID,
+            sheet_name=TURNS_GSHEET_NAME,
+            rows=gsheet_row,
+            creds=GS_CREDS,
+        )
     except Exception as e:
         LOG.error(e)
         raise e
@@ -450,7 +493,7 @@ async def respond(
 
 class ChatbotConfig(BaseModel):
     app_title: str = "CBS Technology Strategy - Fall 2023"
-    chatbot_modes: List[ChatbotMode] = [mode for mode in ChatbotMode]
+    chatbot_modes: List[ChatbotMode] = [mode.value for mode in ChatbotMode]
     case_options: List[str] = poll_questions.get_case_names()
     default_case_option: str = "Netflix"
 
@@ -464,14 +507,24 @@ def change_chatbot_mode(
             new_session = ChatSession.new(
                 use_claude=False,
                 system_msg=ChatSystemMessage.CASE_SYSTEM_MESSAGE,
-                metadata=dict(username=request.username),
+                metadata=ChatSession.set_metadata(
+                    username=request.username,
+                    chatbot_mode=chatbot_mode,
+                    turns_completed=0,
+                    case=poll_question_name,
+                ),
                 poll_question_name=case_input,
             )
         else:
             new_session = ChatSession.new(
                 use_claude=True,
                 system_msg=ChatSystemMessage.RESEARCH_SYSTEM_MESSAGE,
-                metadata=dict(username=request.username),
+                metadata=ChatSession.set_metadata(
+                    username=request.username,
+                    chatbot_mode=chatbot_mode,
+                    turns_completed=0,
+                    case=poll_question_name,
+                ),
                 poll_question_name=None,
             )
         state = new_session

requirements.txt

@@ -1,5 +1,7 @@
 anthropic==0.3.7
 gradio==3.39.0
+gspread==5.10.0
 langchain==0.0.265
 openai==0.27.8
+pytz==2023.3
 tiktoken==0.4.0