Update app.py

app.py

@@ -104,54 +104,61 @@ st.sidebar.caption(f"Session ID: {genuuid()}")
 # Main chat interface
 st.header("Chat with the Agents")
 
-# Initialize chat history in session state
-if "messages" not in st.session_state:
-    st.session_state.messages = []
-
-# Display chat messages from history on app rerun
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"])
-
-# Collect user input
-if user_input := st.chat_input("Write your message here:"):
-    # Add user message to the chat history
-    st.session_state.messages.append({"role": "user", "content": user_input})
-    st.chat_message("user").markdown(user_input)
-
-    # Prepare data for API call
-    data = ChatRequestClient(
-        user_id=user_id,
-        user_input=user_input,
-        numberOfQuestions=numberOfQuestions,
-        welcomeMessage="",
-        llm1=llm1,
-        tokens1=tokens1,
-        temperature1=temp1,
-        persona1SystemMessage=persona1SystemMessage,
-        persona2SystemMessage=persona2SystemMessage,
-        userMessage2=userMessage2,
-        llm2=llm2,
-        tokens2=tokens2,
-        temperature2=temp2
-    )
-
-    # Call the API
-    response = call_chat_api(data)
-
-    # Process the API response
-    agent_message = response.get("content", "No response received from the agent.")
-    elapsed_time = response.get("elapsed_time", 0)
-    count = response.get("count", 0)
-
-    # Add agent response to the chat history
-    st.session_state.messages.append({"role": "assistant", "content": agent_message})
-    with st.chat_message("assistant"):
-        st.markdown(agent_message)
-
-    # Display additional metadata
-    st.markdown(f"##### Time taken: {format_elapsed_time(elapsed_time)} seconds")
-    st.markdown(f"##### Question Count: {count} of {numberOfQuestions}")
+# User ID Input
+user_id = st.text_input("User ID:", key="user_id")
+
+# Ensure user_id is defined or fallback to a default value
+if not user_id:
+    st.warning("Please provide a User ID to start the chat.")
+else:
+    # Initialize chat history in session state
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
+
+    # Display chat messages from history on app rerun
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+
+    # Collect user input
+    if user_input := st.chat_input("Write your message here:"):
+        # Add user message to the chat history
+        st.session_state.messages.append({"role": "user", "content": user_input})
+        st.chat_message("user").markdown(user_input)
+
+        # Prepare data for API call
+        data = ChatRequestClient(
+            user_id=user_id,  # Ensure user_id is passed correctly
+            user_input=user_input,
+            numberOfQuestions=numberOfQuestions,
+            welcomeMessage="",
+            llm1=llm1,
+            tokens1=tokens1,
+            temperature1=temp1,
+            persona1SystemMessage=persona1SystemMessage,
+            persona2SystemMessage=persona2SystemMessage,
+            userMessage2=userMessage2,
+            llm2=llm2,
+            tokens2=tokens2,
+            temperature2=temp2
+        )
+
+        # Call the API
+        response = call_chat_api(data)
+
+        # Process the API response
+        agent_message = response.get("content", "No response received from the agent.")
+        elapsed_time = response.get("elapsed_time", 0)
+        count = response.get("count", 0)
+
+        # Add agent response to the chat history
+        st.session_state.messages.append({"role": "assistant", "content": agent_message})
+        with st.chat_message("assistant"):
+            st.markdown(agent_message)
+
+        # Display additional metadata
+        st.markdown(f"##### Time taken: {format_elapsed_time(elapsed_time)} seconds")
+        st.markdown(f"##### Question Count: {count} of {numberOfQuestions}")