working version :)

app.py

@@ -11,7 +11,12 @@ Notes:
 - Need to call Request from start state
 - Example URL: localhost:7860/?user=123&session=456&questionid=0&response=0
 
+TODO:
+- Test interview ending behavior: does it get triggered reliably?
+- Add password protection
 
+Pre-flight:
+- Check dotenv values match Gradio secrets
 """
 
 from __future__ import annotations
@@ -25,21 +30,24 @@ from typing import Generator, Any
 
 from pathlib import Path
 
-logger = logging.getLogger(__name__)
 
 from utils import (
     PromptTemplate,
     convert_gradio_to_openai,
     initialize_client,
-    seed_azure_key,
+    load_dotenv,
     upload_azure,
     record_chat,
+    ChatLoggerHandler,
+    auth_no_user
 )
 
 
 # %% Initialize common assets
+base_logger = logging.getLogger(__name__)
+chat_logger = ChatLoggerHandler()
 if os.environ.get("AZURE_ENDPOINT") is None:  # Set Azure credentials from local files
-    seed_azure_key()
+    load_dotenv()
 client = initialize_client()  # Shared across sessions
 question_mapping: dict[str, str] = json.loads(Path("assets/question_mapping.json").read_text())
 
@@ -63,7 +71,7 @@ def initialize_interview(request: gr.Request) -> tuple:
     request_params = request.query_params
     user_id: str = request_params.get("user", "testUser")
     session_id: str = request_params.get("session", "testSession")
-    logger.info(f"User: {user_id} (Session: {session_id})")
+    base_logger.info(f"User: {user_id} (Session: {session_id})")
 
     # Parse question
     question_id: str = request_params.get("questionid", "0")
@@ -72,13 +80,13 @@ def initialize_interview(request: gr.Request) -> tuple:
     question_wording: str = question_data["question"]
     question_choices: str = question_data["choices"]
     response_text: str = question_choices[int(response_id)]
-    logger.info(f"Question: {question_wording} ({response_text})")
+    base_logger.info(f"Question: {question_wording} ({response_text})")
 
     # Load initial and system messages
     initial_message: str = PromptTemplate.from_file("assets/initial_message.txt").format(surveyQuestion=question_wording)
     system_message: str = PromptTemplate.from_file("assets/system_message.txt").format(surveyQuestion=question_wording, responseVal=response_text)
-    logger.info(f"Initial message: {initial_message}")
-    logger.info(f"System message: {system_message}")
+    base_logger.info(f"Initial message: {initial_message}")
+    base_logger.info(f"System message: {system_message}")
 
     # Return all
     return (
@@ -171,8 +179,8 @@ def log_interaction(
     session_id: str,
 ) -> None:
     "Record last pair of interactions"
-    record_chat(session_id, "user", chat_history[-1][0])
-    record_chat(session_id, "bot", chat_history[-1][1])
+    record_chat(chat_logger, session_id, "user", chat_history[-1][0])
+    record_chat(chat_logger, session_id, "bot", chat_history[-1][1])
 
 
 def interview_end_check(
@@ -275,6 +283,7 @@ with gr.Blocks(theme="sudeepshouche/minimalist") as demo:
             icon="./arrow_icon.svg",
         )
     exitButton = gr.Button("Save and Exit", visible=False, variant="stop")
+    # testExitButton = gr.Button("Save and Exit", visible=True, variant="stop")
     # Footer
     disclaimer = gr.HTML(
         """
@@ -355,6 +364,9 @@ with gr.Blocks(theme="sudeepshouche/minimalist") as demo:
     exitButton.click(
         end_interview, inputs=[sessionId, chatDisplay], outputs=[chatDisplay]
     )
+    # testExitButton.click(
+    #     end_interview, inputs=[sessionId, chatDisplay], outputs=[chatDisplay]
+    # )
 
 
 if __name__ == "__main__":

requirements.txt

@@ -3,4 +3,5 @@ openai
 wandb
 azure-storage-blob
 azure-identity
-debugpy
+debugpy
+python-dotenv

utils.py

@@ -6,6 +6,7 @@ import os
 from configparser import ConfigParser
 from pathlib import Path
 from string import Formatter
+from dotenv import dotenv_values
 
 import openai
 from azure.storage.blob import BlobClient
@@ -24,10 +25,13 @@ class ChatLoggerHandler:
 
     def record(self, session: str, role: str, record: str):
         log_entry = {
-            "metadata": {"session": session, "timestamp": get_current_timestamp()},
-            "record": {"role": role, "message": record},
+            "session": session,
+            "timestamp": get_current_timestamp(),
+            "role": role,
+            "message": record,
         }
-        (self.logdir / f"{session}.jsonl").write_text(json.dumps(log_entry) + "\n")
+        with open(self.logdir / f"{session}.jsonl", "a+") as f:
+            f.write(json.dumps(log_entry) + "\n")
 
 
 def record_chat(
@@ -123,6 +127,12 @@ def convert_openai_to_gradio(
     return chat_history
 
 
+def load_dotenv():
+    config = dotenv_values(".env")
+    for key, value in config.items():
+        os.environ[key] = value
+
+
 def seed_azure_key(cfg: str = "~/.cfg/openai.cfg") -> None:
     config = ConfigParser()
     try: