| import streamlit as st
|
| import requests
|
| import os
|
| import json
|
| import sys
|
| from dotenv import load_dotenv
|
|
|
|
|
| sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
| from utils import load_css
|
|
|
| load_dotenv()
|
|
|
| API_URL = os.getenv("API_URL", "http://localhost:8000/")
|
|
|
| st.set_page_config(page_title="Chat App", page_icon="💬", layout="wide")
|
| load_css()
|
|
|
| col_title, col_btn = st.columns([0.8, 0.2])
|
| with col_title:
|
| st.markdown('<h1 class="gradient-text">💬 Chat Interface</h1>', unsafe_allow_html=True)
|
| with col_btn:
|
| if st.button("Clear History", type="primary"):
|
| st.session_state.messages = []
|
| st.rerun()
|
|
|
| if "messages" not in st.session_state:
|
| st.session_state.messages = []
|
|
|
|
|
| chat_container = st.container()
|
|
|
| with chat_container:
|
| for message in st.session_state.messages:
|
| with st.chat_message(message["role"]):
|
| st.markdown(message["content"])
|
|
|
| if prompt := st.chat_input("Ask a question about your documents..."):
|
| st.session_state.messages.append(
|
| {"role": "user", "content": prompt}
|
| )
|
|
|
| with st.chat_message("user"):
|
| st.markdown(prompt)
|
|
|
| with st.chat_message("assistant"):
|
| with st.spinner("Thinking..."):
|
| res = None
|
| try:
|
| res = requests.post(
|
| API_URL+"chat",
|
| json={"question": prompt, "history": json.dumps(st.session_state.messages)}
|
| )
|
| except requests.exceptions.RequestException:
|
| st.error("⚠️ Could not connect to the backend. Please try again later.")
|
| st.stop()
|
|
|
| if res is None:
|
| st.stop()
|
|
|
| if res.status_code == 200:
|
| reply = res.json()["response"]
|
| else:
|
| reply = "Sorry, something went wrong. Please try again later."
|
|
|
| st.session_state.messages.append(
|
| {"role": "assistant", "content": reply}
|
| )
|
| st.markdown(reply)
|
|
|
|
|
