Spaces:
Build error
Build error
File size: 1,874 Bytes
a9e6c3b 51225e7 aa31b3b 8858519 b604a12 9476a94 3371395 324a26b b604a12 a9e6c3b f172bb5 afad2ef a620e89 f172bb5 d38433c b604a12 bca3677 44e6288 bca3677 a620e89 b604a12 324a26b eb428fa 5a1233f bca3677 44e6288 a620e89 bca3677 5a1233f bca3677 bba0424 |
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 |
# Set API Keys
os.environ["GROQ_API_KEY"] = st.secrets.get("GROQ_API_KEY", "")
# Load LLM models
llm_judge = ChatGroq(model="deepseek-r1-distill-llama-70b")
rag_llm = ChatGroq(model="mixtral-8x7b-32768")
llm_judge.verbose = True
rag_llm.verbose = True
# Clear ChromaDB cache to fix tenant issue
chromadb.api.client.SharedSystemClient.clear_system_cache()
st.title("Blah - 1")
# Step 1: Choose PDF Source
pdf_source = st.radio("Upload or provide a link to a PDF:", ["Upload a PDF file", "Enter a PDF URL"], index=0, horizontal=True)
if pdf_source == "Upload a PDF file":
uploaded_file = st.file_uploader("Upload your PDF file", type="pdf")
if uploaded_file:
st.session_state.pdf_path = "temp.pdf"
with open(st.session_state.pdf_path, "wb") as f:
f.write(uploaded_file.getbuffer())
st.session_state.pdf_loaded = False
st.session_state.chunked = False
st.session_state.vector_created = False
elif pdf_source == "Enter a PDF URL":
pdf_url = st.text_input("Enter PDF URL:")
if pdf_url and not st.session_state.get("pdf_loaded", False):
with st.spinner("Downloading PDF..."):
try:
response = requests.get(pdf_url)
if response.status_code == 200:
st.session_state.pdf_path = "temp.pdf"
with open(st.session_state.pdf_path, "wb") as f:
f.write(response.content)
st.session_state.pdf_loaded = False
st.session_state.chunked = False
st.session_state.vector_created = False
st.success("✅ PDF Downloaded Successfully!")
else:
st.error("❌ Failed to download PDF. Check the URL.")
except Exception as e:
st.error(f"Error downloading PDF: {e}")
|