Update app.py

app.py

@@ -139,60 +139,15 @@ for chat in st.session_state.chat_history:
 st.markdown("</div>", unsafe_allow_html=True)
 
 # Custom input field
-st.markdown(
-    """
-    <div class="chat-input">
-        <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>
-    function sendChat() {
-        const inputField = document.getElementById('chat-input-field');
-        const message = inputField.value;
-        if (message) {
-            inputField.value = '';
-            fetch('/', {
-                method: 'POST',
-                headers: {
-                    'Content-Type': 'application/json'
-                },
-                body: JSON.stringify({ 'user_input': message })
-            }).then(response => {
-                if (response.ok) {
-                    location.reload();
-                }
-            });
-        }
-    }
-    </script>
-    """,
-    unsafe_allow_html=True
-)
+user_input = st.text_input("Enter your question here:")
 
-# 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
-)
+# Button to trigger response
+if st.button("Get Response"):
+    response = fetch_response(user_input)
+    st.write("Response:", response)
 
-# Handling Save Session in Streamlit
-if st.button("Trigger Save Session", key="trigger_save_session", help="Hidden button to handle session saving"):
+# Save session button
+if st.button("Save Session", key="save_session", help="Save the current chat session", class_="save-session"):
     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."}
@@ -213,3 +168,4 @@ st.markdown(
 
 
 
+