File size: 4,864 Bytes
6a91298 5fc13d5 6a91298 5fc13d5 6a91298 5fc13d5 6a91298 5fc13d5 6a91298 5fc13d5 6a91298 5fc13d5 6a91298 5fc13d5 6a91298 5fc13d5 6a91298 5fc13d5 6a91298 5fc13d5 6a91298 5fc13d5 6a91298 5fc13d5 bac1636 5fc13d5 | 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 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 | import streamlit as st
import requests
import os
import sys
from dotenv import load_dotenv
# Add parent directory to path to import utils
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="Documents", page_icon="📄", layout="wide")
load_css()
st.markdown('<h1 class="gradient-text">📄 Document Management</h1>', unsafe_allow_html=True)
def fetch_documents():
try:
res = requests.get(API_URL + "documents")
if res.status_code == 200:
return res.json().get("documents", [])
except Exception:
st.error("Failed to fetch documents from the server.")
return []
def upload_document(file):
files = {"file": (file.name, file.getvalue(), file.type)}
try:
res=requests.post(API_URL + "document", files=files)
except requests.exceptions.RequestException:
st.error("⚠️ Could not connect to the backend. Please try again later.")
return None
return res
def delete_document(name):
try:
res=requests.delete(API_URL + f"document", json={"source": name})
except requests.exceptions.RequestException:
st.error("⚠️ Could not connect to the backend. Please try again later.")
return None
return res
# Upload Section
st.subheader("📤 Upload Document")
st.markdown("""
<div style="background: rgba(255, 255, 255, 0.05); padding: 20px; border-radius: 12px; border: 1px dashed rgba(255, 255, 255, 0.2); margin-bottom: 20px;">
<p style="margin: 0; color: #aaa;">Supported formats: PDF, DOCX, TXT</p>
</div>
""", unsafe_allow_html=True)
uploaded_file = st.file_uploader(
"Choose a file",
type=["pdf", "docx", "txt"],
label_visibility="collapsed"
)
if uploaded_file:
if st.button("Upload and Ingest", type="primary"):
with st.spinner("Uploading and ingesting document..."):
res = upload_document(uploaded_file)
if res is None:
pass # Error already shown by upload_document
elif res.status_code == 200:
st.success("✅ Document uploaded successfully")
else:
st.error("❌ Failed to upload document")
st.divider()
# Documents List
st.subheader("📄 Available Documents")
search_col, _ = st.columns([1, 1])
with search_col:
search_query = st.text_input(
"",
placeholder="🔍 Search documents...",
label_visibility="collapsed"
)
with st.spinner("Fetching documents..."):
documents = fetch_documents()
if search_query:
documents = [
doc for doc in documents
if search_query.lower() in doc[0].lower()
]
if not documents:
st.info("No documents available.")
else:
# Header
st.markdown("""
<div style="display: flex; font-weight: bold; color: #e0e0e0; padding: 10px 0; border-bottom: 2px solid rgba(67, 97, 238, 0.5); background: rgba(0,0,0,0.2); margin-bottom: 10px;">
<div style="flex: 3; padding-left: 10px;">Filename</div>
<div style="flex: 2;">Status</div>
<div style="flex: 2;">Uploaded At</div>
<div style="flex: 1; text-align: right; padding-right: 10px;">Actions</div>
</div>
""", unsafe_allow_html=True)
for idx, doc in enumerate(documents):
filename, status, timestamp, path = doc
status_class = "status-ingested" if status == "ingested" else "status-pending"
status_display = f"✅ {status.capitalize()}" if status == "ingested" else f"⏳ {status.capitalize()}"
col1, col2, col3, col4 = st.columns([3, 2, 2, 1])
with col1:
st.markdown(f"<div style='padding-top: 5px; padding-left: 10px;'>{filename}</div>", unsafe_allow_html=True)
with col2:
st.markdown(f"<span class='status-badge {status_class}'>{status_display}</span>", unsafe_allow_html=True)
with col3:
st.markdown(f"<div style='padding-top: 5px; color: #aaa;'>{timestamp}</div>", unsafe_allow_html=True)
with col4:
if st.button("🗑️", key=f"delete_{idx}", help="Delete Document"):
with st.spinner("Deleting..."):
res = delete_document(path)
if res and res.status_code == 200:
st.success(f"Deleted {filename}")
st.rerun()
else:
st.error("Failed to delete")
st.markdown("<hr style='margin: 5px 0; border-color: rgba(255,255,255,0.05);'>", unsafe_allow_html=True)
|