fix storing message pair

app.py

@@ -1,5 +1,4 @@
 import os
-import time
 import uuid
 from typing import Dict, List, Optional, Tuple
 
@@ -65,11 +64,14 @@ def store_chat(
     msg_pair_medium: Tuple[str, str],
     medium_lang: str,
 ):
-    store_message_pair(chat_id, msg_pair_bo, LANG_BO)
-    store_message_pair(chat_id, msg_pair_medium, medium_lang)
+    msg_pair = {
+        "bo": msg_pair_bo,
+        medium_lang: msg_pair_medium,
+    }
+    store_message_pair(chat_id, msg_pair)
 
 
-def bot(history_bo: list, request: gr.Request):
+def bot(history_bo: list, chat_id: str):
     """Translate user input to English, send to OpenAI, translate response to Tibetan, and return to user.
 
     Args:
@@ -96,7 +98,7 @@ def bot(history_bo: list, request: gr.Request):
         print("------------------------")
 
     store_chat(
-        chat_id=request.client.host,
+        chat_id=chat_id,
         msg_pair_bo=(input_bo, resopnse_bo),
         msg_pair_medium=(input_, response),
         medium_lang=LANG_MEDIUM,
@@ -104,11 +106,17 @@ def bot(history_bo: list, request: gr.Request):
     return history_bo
 
 
+def get_chat_id():
+    return str(uuid.uuid4())
+
+
 with gr.Blocks() as demo:
+    chat_id = gr.State(value=get_chat_id)
     history_en = gr.State(value=[])
     history_bo = gr.Chatbot(label="Tibetan Chatbot").style(height=650)
     input_bo = gr.Textbox(
-        show_label=False, placeholder="Type a message here and press enter"
+        show_label=False,
+        placeholder=f"Type a message here and press enter",
     )
     input_bo.submit(
         fn=user,
@@ -117,11 +125,11 @@ with gr.Blocks() as demo:
         queue=False,
     ).then(
         fn=bot,
-        inputs=[history_bo],
+        inputs=[history_bo, chat_id],
         outputs=[history_bo],
     )
 
     clear = gr.Button("New Chat")
-    clear.click(lambda: ("", []), None, [input_bo, history_bo], queue=False)
+    clear.click(lambda: [], None, history_bo, queue=False)
 
 demo.launch()

store.py

@@ -1,7 +1,7 @@
 import json
 import uuid
 from datetime import datetime
-from typing import Tuple
+from typing import Dict, Tuple
 
 import boto3
 
@@ -9,7 +9,7 @@ dynamodb = boto3.resource("dynamodb", region_name="us-east-1")
 all_chats_table = dynamodb.Table("ChatbotTibetanAllChats")
 
 
-def store_message_pair(chat_id: str, msg_pair: Tuple[str, str], lang: str):
+def store_message_pair(chat_id: str, msg_pair: Dict[str, Tuple[str, str]]):
     """Store the chat history to DynamoDB
 
     Args:
@@ -25,7 +25,6 @@ def store_message_pair(chat_id: str, msg_pair: Tuple[str, str], lang: str):
         Item={
             "msg_pair_id": msg_pair_id,
             "msg_pair": json.dumps(msg_pair, ensure_ascii=False),
-            "lang": lang,
             "created_at": datetime.now().isoformat(),
             "chat_id": chat_id,
         }
@@ -35,12 +34,9 @@ def store_message_pair(chat_id: str, msg_pair: Tuple[str, str], lang: str):
 
 if __name__ == "__main__":
     # Replace with your own DynamoDB table name and chat ID
-    chat_id = uuid.uuid4().hex[:4]
+    chat_id = str(uuid.uuid4())
 
     # Replace with your own chat history (list of tuples or list of dictionaries)
-    chat_history = [
-        ("User", "Hello, how are you?"),
-        ("Chatbot", "I am fine, thank you!"),
-    ]
-    for msg_pair in chat_history:
-        store_message_pair(chat_id, msg_pair, "en")
+    msg_pair = {"bo": ("hello", "hello"), "en": ("hello", "hello")}
+    response = store_message_pair(chat_id, msg_pair)
+    print(response)