Update streamlit_app.py

streamlit_app.py

@@ -1,76 +1,76 @@
-
-import streamlit as st
-import requests
-import re  # For space cleanup
-
-st.set_page_config(page_title="AI Chatbot", layout="centered")
-st.title("🤖 AI Chatbot")
-
-if "messages" not in st.session_state:
-    st.session_state.messages = []
-
-# Function to query AI API and stream response
-def query_ai(question):
-    url = "http://127.0.0.1:8000/query/"
-    params = {"input_text": question}
-    
-    with requests.get(url, params=params, stream=True,timeout=120) as response:
-        if response.status_code == 200:
-            full_response = ""
-            for chunk in response.iter_content(chunk_size=1024):
-                if chunk:
-                    text_chunk = chunk.decode("utf-8")
-                    full_response += text_chunk
-                    yield full_response  # Streamed response
-
-# Custom CSS for spacing fix
-st.markdown("""
-    <style>
-        .chat-box {
-            background-color: #1e1e1e;
-            padding: 12px;
-            border-radius: 10px;
-            margin-top: 5px;
-            font-size: 154x;
-            font-family: monospace;
-            white-space: pre-wrap;
-            word-wrap: break-word;
-            line-height: 1.2;
-            color: #ffffff;
-        }
-    </style>
-""", unsafe_allow_html=True)
-
-user_input = st.text_input("Ask a question:", "", key="user_input")
-submit_button = st.button("Submit")
-
-if submit_button and user_input:
-    st.session_state.messages.append({"role": "user", "content": user_input})
-
-    # Placeholder for streaming
-    response_container = st.empty()
-    full_response = ""
-
-    with st.spinner("🤖 AI is thinking..."):
-        for chunk in query_ai(user_input):
-            full_response = chunk
-            response_container.markdown(f'<div class="chat-box">{full_response}</div>', unsafe_allow_html=True)
-
-    response_container.empty()  # Hides the streamed "Thinking" response after completion
-
-    # Extract refined answer after "</think>"
-    if "</think>" in full_response:
-        refined_response = full_response.split("</think>", 1)[-1].strip()
-    else:
-        refined_response = full_response  # Fallback if </think> is missing
-
-    # Remove extra newlines and excessive spaces
-    refined_response = re.sub(r'\n\s*\n', '\n', refined_response.strip())
-
-    # Expandable AI Thought Process Box
-    with st.expander("🤖 AI's Thought Process (Click to Expand)"):
-        st.markdown(f'<div class="chat-box">{full_response}</div>', unsafe_allow_html=True)
-
-    # Display refined answer with clean formatting
-    st.write("Answer:")
-    st.markdown(refined_response, unsafe_allow_html=True)
+
+import streamlit as st
+import requests
+import re  # For space cleanup
+
+st.set_page_config(page_title="AI Chatbot", layout="centered")
+st.title("🤖 AI Chatbot")
+
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Function to query AI API and stream response
+def query_ai(question):
+    url = "http://127.0.0.1:8000/query/"
+    params = {"input_text": question}
+    
+    with requests.get(url, params=params, stream=True,timeout=None) as response:
+        if response.status_code == 200:
+            full_response = ""
+            for chunk in response.iter_content(chunk_size=1024):
+                if chunk:
+                    text_chunk = chunk.decode("utf-8")
+                    full_response += text_chunk
+                    yield full_response  # Streamed response
+
+# Custom CSS for spacing fix
+st.markdown("""
+    <style>
+        .chat-box {
+            background-color: #1e1e1e;
+            padding: 12px;
+            border-radius: 10px;
+            margin-top: 5px;
+            font-size: 154x;
+            font-family: monospace;
+            white-space: pre-wrap;
+            word-wrap: break-word;
+            line-height: 1.2;
+            color: #ffffff;
+        }
+    </style>
+""", unsafe_allow_html=True)
+
+user_input = st.text_input("Ask a question:", "", key="user_input")
+submit_button = st.button("Submit")
+
+if submit_button and user_input:
+    st.session_state.messages.append({"role": "user", "content": user_input})
+
+    # Placeholder for streaming
+    response_container = st.empty()
+    full_response = ""
+
+    with st.spinner("🤖 AI is thinking..."):
+        for chunk in query_ai(user_input):
+            full_response = chunk
+            response_container.markdown(f'<div class="chat-box">{full_response}</div>', unsafe_allow_html=True)
+
+    response_container.empty()  # Hides the streamed "Thinking" response after completion
+
+    # Extract refined answer after "</think>"
+    if "</think>" in full_response:
+        refined_response = full_response.split("</think>", 1)[-1].strip()
+    else:
+        refined_response = full_response  # Fallback if </think> is missing
+
+    # Remove extra newlines and excessive spaces
+    refined_response = re.sub(r'\n\s*\n', '\n', refined_response.strip())
+
+    # Expandable AI Thought Process Box
+    with st.expander("🤖 AI's Thought Process (Click to Expand)"):
+        st.markdown(f'<div class="chat-box">{full_response}</div>', unsafe_allow_html=True)
+
+    # Display refined answer with clean formatting
+    st.write("Answer:")
+    st.markdown(refined_response, unsafe_allow_html=True)