Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import requests
+import os
 import time
 
 # =============================
@@ -9,6 +10,9 @@ import time
 BACKEND_URL = "https://mohith96-researchagentbackend.hf.space/research"
 REQUEST_TIMEOUT = 300  # seconds
 
+# Hugging Face token for private backend access
+HF_BACKEND_TOKEN = os.getenv("HF_BACKEND_TOKEN")
+
 # =============================
 # PAGE SETUP
 # =============================
@@ -54,24 +58,34 @@ with col1:
 if run_clicked:
     if not query.strip():
         st.warning("You need to enter a research query. This is not optional.")
+    elif not HF_BACKEND_TOKEN:
+        st.error("Backend token missing. Set HF_BACKEND_TOKEN in Space secrets.")
     else:
         with st.spinner("Running research pipeline..."):
             try:
+                headers = {
+                    "Authorization": f"Bearer {HF_BACKEND_TOKEN}"
+                }
+
                 response = requests.post(
                     BACKEND_URL,
                     json={"query": query},
+                    headers=headers,
                     timeout=REQUEST_TIMEOUT
                 )
 
                 if response.status_code != 200:
-                    st.error(f"Backend error ({response.status_code}). Check logs.")
+                    st.error(
+                        f"Backend error ({response.status_code}). "
+                        f"Response: {response.text}"
+                    )
                 else:
                     payload = response.json()
                     st.session_state.title = payload.get("title", "Untitled")
                     st.session_state.result = payload.get("report", "")
 
             except requests.exceptions.Timeout:
-                st.error("Backend timed out. Either it's slow or dead.")
+                st.error("Backend timed out. Either it's slow or asleep.")
             except Exception as e:
                 st.error(f"Unexpected failure: {str(e)}")
 
@@ -97,4 +111,4 @@ if st.session_state.result:
 # =============================
 
 st.divider()
-st.caption("Frontend-only Space. All intelligence lives in the backend. As it should.")
+st.caption("Frontend-only Space. Backend is private and authenticated. As it should be.")