Update app.py

app.py

@@ -111,6 +111,7 @@ st.markdown(
         color: #e1e1e1;
         padding: 0.5rem 2rem;
         font-size: 1rem;
+        cursor: pointer;
     }
     </style>
     """,
@@ -141,7 +142,7 @@ st.markdown("</div>", unsafe_allow_html=True)
 st.markdown(
     """
     <div class="chat-input">
-        <input type="text" id="chat-input-field" placeholder="Type your message here..." onkeydown="if(event.key === 'Enter'){document.getElementById('chat-submit-button').click();}">
+        <input type="text" id="chat-input-field" placeholder="Type your message here..." onkeydown="if(event.key === 'Enter'){sendChat();}">
         <button id="chat-submit-button" onclick="sendChat()">Send</button>
     </div>
     <script>
@@ -168,8 +169,30 @@ st.markdown(
     unsafe_allow_html=True
 )
 
-# Save session button
-if st.button("Save Session", key="save_session", help="Save the current chat session", class_="save-session"):
+# Custom Save Session button
+st.markdown(
+    """
+    <button class="save-session" onclick="saveSession()">Save Session</button>
+    <script>
+    function saveSession() {
+        fetch('/save-session', {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json'
+            },
+        }).then(response => {
+            if (response.ok) {
+                location.reload();
+            }
+        });
+    }
+    </script>
+    """,
+    unsafe_allow_html=True
+)
+
+# Handling Save Session in Streamlit
+if st.button("Trigger Save Session", key="trigger_save_session", help="Hidden button to handle session saving"):
     st.session_state.previous_sessions.append(st.session_state.chat_history)
     st.session_state.chat_history = [
         {"role": "system", "content": "you are a helpful assistant. Take the input from the users and try to provide as detailed response as possible. Provide proper examples to help the user. Try to mention references or provide citations to make it more detail-oriented."}
@@ -189,3 +212,4 @@ st.markdown(
 
 
 
+