Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,6 +1,11 @@
 import streamlit as st
 import requests
 import uuid
+import os
+
+# Grab the token from Hugging Face secrets
+HF_TOKEN = os.getenv("HF_TOKEN")
+HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
 
 # 1. PAGE CONFIGURATION
 st.set_page_config(page_title="Text2SQL Agent", page_icon="🤖", layout="centered")
@@ -40,7 +45,8 @@ with st.sidebar:
                         # (Check your FastAPI main.py to confirm this endpoint name!)
                         response = requests.post(
                             "https://lightrt-text2sql-backend.hf.space/upload_url", 
-                            json=payload_data
+                            json=payload_data,
+                            headers=HEADERS
                         )
                         
                         if response.status_code == 200:
@@ -94,7 +100,8 @@ if st.session_state.is_db_connected:
                 # Send the question to your LangGraph backend
                 chat_response = requests.post(
                     "https://lightrt-text2sql-backend.hf.space/chat", 
-                    json=payload
+                    json=payload,
+                    headers=HEADERS
                 )
                 
                 if chat_response.status_code == 200: