Update app.py

app.py

@@ -5,6 +5,8 @@ import uuid
 import concurrent.futures
 from requests.exceptions import ChunkedEncodingError
 import os
+from dotenv import load_dotenv
+load_dotenv()
 
 # Define the endpoints
 host = os.getenv("BACKEND_URL")
@@ -92,21 +94,23 @@ tarot_cards = [
     "Queen of Wands",
     "King of Wands",
 ]
+
 # Define the request payload structure
 class ChatRequest:
-    def __init__(self, session_id, messages, model_id, temperature, seer_name):
+    def __init__(self, session_id, messages, model_id, temperature, seer_name, seer_personality):
         self.session_id = session_id
         self.messages = messages
         self.model_id = model_id
         self.temperature = temperature
         self.seer_name = seer_name
+        self.seer_personality = seer_personality
 
 class ChatRequestWithMemory(ChatRequest):
-    def __init__(self, session_id, messages, model_id, temperature, seer_name, summary_threshold):
-        super().__init__(session_id, messages, model_id, temperature, seer_name)
+    def __init__(self, session_id, messages, model_id, temperature, seer_name, seer_personality, summary_threshold):
+        super().__init__(session_id, messages, model_id, temperature, seer_name, seer_personality)
         self.summary_threshold = summary_threshold
 
-def compare_chatbots(session_id, messages, model_id, temperature, seer_name, summary_threshold, tarot_card):
+def compare_chatbots(session_id, messages, model_id, temperature, seer_name, seer_personality, summary_threshold, tarot_card):
     # Convert messages list to a single string    
     # Prepare the payloads
     print("tarot_card", tarot_card)
@@ -117,6 +121,7 @@ def compare_chatbots(session_id, messages, model_id, temperature, seer_name, sum
         "temperature": temperature,
         "tarot_card": tarot_card,
         "seer_name": seer_name,
+        "seer_personality": seer_personality,
     })
     payload_memory = json.dumps({
         "session_id": session_id + "_memory",
@@ -124,6 +129,7 @@ def compare_chatbots(session_id, messages, model_id, temperature, seer_name, sum
         "model_id": model_id,
         "temperature": temperature,
         "seer_name": seer_name,
+        "seer_personality": seer_personality,
         "tarot_card": tarot_card,
         "summary_threshold": summary_threshold,
     })
@@ -154,8 +160,8 @@ def compare_chatbots(session_id, messages, model_id, temperature, seer_name, sum
     return response_default_text, response_memory_text
 
 # Function to handle chat interaction
-def chat_interaction(session_id, message, model_id, temperature, seer_name, summary_threshold, chat_history_default, chat_history_memory, tarot_card):
-    response_default, response_memory = compare_chatbots(session_id, message, model_id, temperature, seer_name, summary_threshold, tarot_card)
+def chat_interaction(session_id, message, model_id, temperature, seer_name, seer_personality, summary_threshold, chat_history_default, chat_history_memory, tarot_card):
+    response_default, response_memory = compare_chatbots(session_id, message, model_id, temperature, seer_name, seer_personality, summary_threshold, tarot_card)
     
     chat_history_default.append((message, response_default))
     chat_history_memory.append((message, response_memory))
@@ -214,8 +220,9 @@ with gr.Blocks() as demo:
         model_id = gr.Dropdown(label="Model ID", choices=model_id_choices, value=model_id_choices[0])
         temperature = gr.Slider(0, 1, step=0.1, label="Temperature", value=0.5)
         seer_name = gr.Textbox(label="Seer Name", value="แม่หมอแพตตี้")
+        seer_personality = gr.Textbox(label="Seer Personality", value="You are a friend who is always ready to help.")
         tarot_card = gr.Dropdown(label="Tarot Card", value=[], choices=tarot_cards, multiselect=True)
-        summary_threshold = gr.Number(label="Summary Threshold", value=5)
+        summary_threshold = gr.Number(label="Summary Threshold", value=7)
 
     with gr.Accordion("View History of Memory Chatbot", open=False):
         session_id_memory = gr.Textbox(label="Session ID", value=f"{session_id_default}_memory")
@@ -223,18 +230,18 @@ with gr.Blocks() as demo:
         chat_history_json = gr.JSON(label="Chat History")  # New JSON field
     
     submit_button.click(
-        lambda session_id, message, model_id, temperature, seer_name, summary_threshold, chatbot_default, chatbot_memory, tarot_card: chat_interaction(
-            session_id, message, model_id, temperature, seer_name, summary_threshold, chatbot_default, chatbot_memory, tarot_card
+        lambda session_id, message, model_id, temperature, seer_name, seer_personality, summary_threshold, chatbot_default, chatbot_memory, tarot_card: chat_interaction(
+            session_id, message, model_id, temperature, seer_name, seer_personality, summary_threshold, chatbot_default, chatbot_memory, tarot_card
         ),
-        inputs=[session_id, message, model_id, temperature, seer_name, summary_threshold, chatbot_default, chatbot_memory, tarot_card],
+        inputs=[session_id, message, model_id, temperature, seer_name, seer_personality, summary_threshold, chatbot_default, chatbot_memory, tarot_card],
         outputs=[message, chatbot_default, chatbot_memory, tarot_card]
     )
 
     message.submit(
-        lambda session_id, message, model_id, temperature, seer_name, summary_threshold, chatbot_default, chatbot_memory, tarot_card: chat_interaction(
-            session_id, message, model_id, temperature, seer_name, summary_threshold, chatbot_default, chatbot_memory, tarot_card
+        lambda session_id, message, model_id, temperature, seer_name, seer_personality, summary_threshold, chatbot_default, chatbot_memory, tarot_card: chat_interaction(
+            session_id, message, model_id, temperature, seer_name, seer_personality, summary_threshold, chatbot_default, chatbot_memory, tarot_card
         ),
-        inputs=[session_id, message, model_id, temperature, seer_name, summary_threshold, chatbot_default, chatbot_memory, tarot_card],
+        inputs=[session_id, message, model_id, temperature, seer_name, seer_personality, summary_threshold, chatbot_default, chatbot_memory, tarot_card],
         outputs=[message, chatbot_default, chatbot_memory, tarot_card]
     )