Update app.py

app.py

@@ -1,171 +1,102 @@
 import streamlit as st
 import requests
-import os
 
-# ✅ Hugging Face fix: use same-origin backend routing
+# ✅ HF-safe same-origin routing
 BACKEND_URL = ""
 
 st.set_page_config(page_title="PDF Assistant", page_icon="📘", layout="wide")
 
-# ---------------- CSS (Dark Theme) ----------------
+# ---------------- CSS ----------------
 st.markdown("""
 <style>
-:root {
-    --primary-color: #1e3a8a;
-    --background-color: #0e1117;
-    --secondary-background-color: #1a1d29;
-    --text-color: #f0f2f6;
-}
-
 .chat-user {
     background: #2d3748;
     padding: 12px;
-    border-radius: 10px 10px 2px 10px;
-    margin: 6px 0 6px auto;
-    max-width: 85%;
+    border-radius: 10px;
     text-align: right;
-    color: var(--text-color);
 }
 .chat-bot {
-    background: var(--primary-color);
+    background: #1e3a8a;
     padding: 12px;
-    border-radius: 10px 10px 10px 2px;
-    margin: 6px auto 6px 0;
-    max-width: 85%;
-    text-align: left;
-    color: #ffffff;
+    border-radius: 10px;
+    color: white;
 }
-
 .sources {
     font-size: 0.8em;
     opacity: 0.7;
-    margin-top: 10px;
-    border-top: 1px solid rgba(255, 255, 255, 0.1);
-    padding-top: 5px;
-}
-
-.footer {
-    position: fixed;
-    left: 0;
-    bottom: 0;
-    width: 100%;
-    background-color: var(--secondary-background-color);
-    color: var(--text-color);
-    text-align: center;
-    padding: 10px;
-    font-size: 0.85em;
-    border-top: 1px solid rgba(255, 255, 255, 0.1);
-}
-.footer a {
-    color: var(--primary-color);
-    text-decoration: none;
-    font-weight: bold;
-}
-.footer a:hover {
-    text-decoration: underline;
 }
 </style>
 """, unsafe_allow_html=True)
 
-# ---------------- SESSION STATE ----------------
+# ---------------- STATE ----------------
 if "chat" not in st.session_state:
     st.session_state.chat = []
-
-if "uploaded_file_name" not in st.session_state:
-    st.session_state.uploaded_file_name = None
-
+if "pdf" not in st.session_state:
+    st.session_state.pdf = None
 if "uploader_key" not in st.session_state:
     st.session_state.uploader_key = 0
 
 st.title("📘 PDF Assistant")
 
 # ---------------- FUNCTIONS ----------------
-def clear_chat_history():
-    st.session_state.chat = []
-
-
 def clear_memory():
-    res = requests.post(f"{BACKEND_URL}/clear")
-    if res.status_code == 200:
-        st.session_state.uploaded_file_name = None
-        st.session_state.uploader_key += 1
-        st.success("Memory cleared. Please upload a new PDF.")
-    else:
-        st.error(res.json().get("detail", "Failed to clear memory"))
+    requests.post(f"{BACKEND_URL}/api/clear")
+    st.session_state.chat = []
+    st.session_state.pdf = None
+    st.session_state.uploader_key += 1
+    st.success("Memory cleared")
 
 
 # ---------------- SIDEBAR ----------------
 with st.sidebar:
-    st.header("Controls")
-    st.button("🗑️ Clear Chat History", on_click=clear_chat_history, use_container_width=True)
-    st.button("🔥 Clear PDF Memory", on_click=clear_memory, use_container_width=True)
-
-    st.markdown("---")
-    if st.session_state.uploaded_file_name:
-        st.success(f"✅ Active PDF:\n`{st.session_state.uploaded_file_name}`")
-    else:
-        st.warning("⬆️ Upload a PDF to start chatting!")
+    st.button("🔥 Clear PDF Memory", on_click=clear_memory)
 
-# ---------------- UPLOAD PDF ----------------
+# ---------------- UPLOAD ----------------
 uploaded = st.file_uploader(
-    "Upload your PDF",
+    "Upload PDF",
     type=["pdf"],
     key=st.session_state.uploader_key
 )
 
-if uploaded and uploaded.name != st.session_state.uploaded_file_name:
-    st.session_state.uploaded_file_name = None
-    st.session_state.chat = []
-
-    with st.spinner(f"Processing '{uploaded.name}'..."):
+if uploaded and uploaded.name != st.session_state.pdf:
+    with st.spinner("Processing PDF..."):
         files = {"file": (uploaded.name, uploaded.getvalue(), "application/pdf")}
-        res = requests.post(f"{BACKEND_URL}/upload", files=files)
+        res = requests.post(f"{BACKEND_URL}/api/upload", files=files)
 
         if res.status_code == 200:
-            chunks = res.json().get("chunks", 0)
-            st.success(f"PDF processed successfully! {chunks} chunks created.")
-            st.session_state.uploaded_file_name = uploaded.name
+            st.session_state.pdf = uploaded.name
+            st.success("PDF processed")
         else:
-            st.error(res.json().get("detail", "Upload failed"))
+            st.error(res.text)
 
     st.rerun()
 
-# ---------------- CHAT INPUT ----------------
-disabled_input = st.session_state.uploaded_file_name is None
+# ---------------- CHAT ----------------
 question = st.text_input(
-    "Ask a question about the loaded PDF:",
-    disabled=disabled_input
+    "Ask a question",
+    disabled=st.session_state.pdf is None
 )
 
-if st.button("Send", disabled=disabled_input) and question:
+if st.button("Send") and question:
     st.session_state.chat.append(("user", question))
 
-    with st.spinner("Thinking..."):
-        res = requests.post(f"{BACKEND_URL}/ask", json={"question": question})
+    res = requests.post(
+        f"{BACKEND_URL}/api/ask",
+        json={"question": question}
+    )
 
-        if res.status_code == 200:
-            data = res.json()
-            answer = data.get("answer", "")
-            sources = data.get("sources", 0)
-            bot_msg = f"{answer}<div class='sources'>Context Chunks Used: {sources}</div>"
-            st.session_state.chat.append(("bot", bot_msg))
-        else:
-            st.session_state.chat.append(("bot", "❌ Error generating answer"))
+    if res.status_code == 200:
+        data = res.json()
+        msg = f"{data['answer']}<div class='sources'>Chunks used: {data['sources']}</div>"
+        st.session_state.chat.append(("bot", msg))
+    else:
+        st.session_state.chat.append(("bot", "Error"))
 
     st.rerun()
 
 # ---------------- CHAT HISTORY ----------------
-st.markdown("## Chat History")
 for role, msg in st.session_state.chat:
     if role == "user":
         st.markdown(f"<div class='chat-user'>{msg}</div>", unsafe_allow_html=True)
     else:
         st.markdown(f"<div class='chat-bot'>{msg}</div>", unsafe_allow_html=True)
-
-# ---------------- FOOTER ----------------
-st.markdown("""
-<div class="footer">
-    Created by <a href="https://www.linkedin.com/in/abhishek-iitr/" target="_blank">
-    Abhishek Saxena</a>
-</div>
-""", unsafe_allow_html=True)