Update app.py

app.py

@@ -39,32 +39,29 @@ def format_elapsed_time(time):
     return "{:.2f}".format(time)
 
 # Layout with three columns
-col1, col2, col3 = st.columns([1, 3, 1])  # Adjusted width ratios for better centering
+col1, col2, col3 = st.columns([1, 3, 1], gap="small")  # Adjusted width ratios for better centering
 
 # Left Column: Variables and Settings
 with col1:
     st.sidebar.image('cognizant_logo.jpg')
-    st.sidebar.header("Agent Personas Design")
-    st.sidebar.subheader("Intake AI")
+    st.sidebar.header("RAG Query Designer")
+    st.sidebar.subheader("Query Translation Prompt")
     numberOfQuestions = st.sidebar.slider("Number of Questions", 0, 10, 5, 1)
-    persona1SystemMessage = st.sidebar.text_area("Define Intake Persona", value=placeHolderPersona1, height=300)
+    persona1SystemMessage = st.sidebar.text_area("Define Query Translation Prompt", value=placeHolderPersona1, height=300)
     llm1 = st.sidebar.selectbox("Model Selection", ['GPT-4', 'GPT3.5'])
     temp1 = st.sidebar.slider("Temperature", 0.0, 1.0, 0.6, 0.1)
     tokens1 = st.sidebar.slider("Tokens", 0, 4000, 500, 100)
-    persona2SystemMessage = st.sidebar.text_area("Define Recommendation Persona", value=placeHolderPersona2, height=300)
-    llm2 = st.sidebar.selectbox("Model Selection", ['GPT-4', 'GPT3.5'], key="persona2")
-    temp2 = st.sidebar.slider("Temperature", 0.0, 1.0, 0.5, 0.1, key="temp2")
-    tokens2 = st.sidebar.slider("Tokens", 0, 4000, 500, 100, key="tokens2")
+
 
 # Middle Column: Chat Interface
 with col2:
-    st.markdown("<div style='text-align: center;'><h1>Chat with the Agents</h1></div>", unsafe_allow_html=True)
+    st.markdown("<div style='text-align: center;'><h1>Experiment With Querys</h1></div>", unsafe_allow_html=True)
 
     # User ID Input
     user_id = st.text_input("User ID:", key="user_id")
 
     if not user_id:
-        st.warning("Please provide a User ID to start the chat.")
+        st.warning("Please provide an experiment ID to start the chat.")
     else:
         # Initialize chat history
         if "messages" not in st.session_state:
@@ -78,7 +75,7 @@ with col2:
                     st.markdown(message["content"])
 
         # Chat input at the bottom
-        if user_input := st.chat_input("Write your message here:"):
+        if user_input := st.chat_input("Start Chat:"):
             # Add user message
             st.session_state.messages.append({"role": "user", "content": user_input})
             st.chat_message("user").markdown(user_input)
@@ -92,11 +89,11 @@ with col2:
                 persona2SystemMessage=persona2SystemMessage,
                 llm1=llm1,
                 tokens1=tokens1,
-                temperature1=temp1,
+                temperature1=0.,
                 userMessage2="",
-                llm2=llm2,
-                tokens2=tokens2,
-                temperature2=temp2,
+                llm2="GPT-4",
+                tokens2=500,
+                temperature2=0.1,
             )
 
             # Call the API