update to HF dataset and secrets for login

app.py

@@ -6,6 +6,7 @@ import asyncio
 import logging
 from copy import deepcopy
 import json
+import uuid
 
 import gradio as gr
 
@@ -49,7 +50,8 @@ async def respond(
     inp: str,
     state: Optional[Tuple[List,
                           ConversationTokenBufferMemory,
-                          ConversationChain]],
+                          ConversationChain, 
+                          str]],
     request: gr.Request
 ):
     """Execute the chat functionality."""
@@ -82,8 +84,9 @@ async def respond(
                 max_token_limit=GPT_3_5_CONTEXT_LENGTH,
                 return_messages=True)
             chain = ConversationChain(memory=memory, prompt=template, llm=llm)
-            state = ([], memory, chain)
-        history, memory, chain = state
+            session_id = str(uuid.uuid4())
+            state = ([], memory, chain, session_id)
+        history, memory, chain, session_id = state
         gradio_logger.info(f"""[{request.username}] STARTING CHAIN""")
         gradio_logger.debug(f"History: {history}")
         gradio_logger.debug(f"User input: {inp}")
@@ -101,14 +104,16 @@ async def respond(
             user, bot = history[-1]
             bot += tok
             history[-1] = (user, bot)
-            yield history, (history, memory, chain)
+            yield history, (history, memory, chain, session_id)
         await run
         gradio_logger.info(f"""[{request.username}] ENDING CHAIN""")
         gradio_logger.debug(f"History: {history}")
         gradio_logger.debug(f"Memory: {memory.json()}")
         data_to_flag = {
             "history": deepcopy(history),
-            "username": request.username
+            "username": request.username,
+            "timestamp": datetime.datetime.now(datetime.timezone.utc).isoformat(),
+            "session_id": session_id
         },
         gradio_logger.debug(f"Data to flag: {data_to_flag}")
         gradio_flagger.flag(flag_data=data_to_flag, username=request.username)
@@ -117,6 +122,7 @@ async def respond(
         raise e
 
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+HF_TOKEN = os.getenv("HF_TOKEN")
 
 llm = ChatOpenAI(model_name="gpt-3.5-turbo",
                  temperature=1,
@@ -129,9 +135,9 @@ template = make_template()
 
 theme = gr.themes.Soft()
 
-creds = [(os.getenv("USERNAME"), os.getenv("PASSWORD"))]
+creds = [(os.getenv("CHAT_USERNAME"), os.getenv("CHAT_PASSWORD"))]
 
-gradio_flagger = gr.CSVLogger()
+gradio_flagger = gr.HuggingFaceDatasetSaver(HF_TOKEN, "chats")
 title = "Chat with ChatGPT"
 
 with gr.Blocks(css="""#col_container { margin-left: auto; margin-right: auto;} #chatbot {height: 520px; overflow: auto;}""",
@@ -139,7 +145,6 @@ with gr.Blocks(css="""#col_container { margin-left: auto; margin-right: auto;} #
                analytics_enabled=False,
                title=title) as demo:
     gr.HTML(title)
-    
     with gr.Column(elem_id="col_container"):
         state = gr.State()
         chatbot = gr.Chatbot(label='ChatBot', elem_id="chatbot")
@@ -148,7 +153,7 @@ with gr.Blocks(css="""#col_container { margin-left: auto; margin-right: auto;} #
         b1 = gr.Button(value="Submit", variant="secondary").style(
             full_width=False)
 
-    gradio_flagger.setup([chatbot], "flagged_data_points")
+    gradio_flagger.setup([chatbot], "chats")
 
     inputs.submit(respond, [inputs, state], [chatbot, state],)
     b1.click(respond, [inputs, state], [chatbot, state],)