Spaces:
Sleeping
Sleeping
File size: 2,592 Bytes
f8b9b73 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
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)
|