Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -14,3 +14,34 @@ os.environ["GROQ_API_KEY"] = st.secrets.get("GROQ_API_KEY", "")
|
|
14 |
# Load LLM models
|
15 |
llm_judge = ChatGroq(model="deepseek-r1-distill-llama-70b")
|
16 |
rag_llm = ChatGroq(model="mixtral-8x7b-32768")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
14 |
# Load LLM models
|
15 |
llm_judge = ChatGroq(model="deepseek-r1-distill-llama-70b")
|
16 |
rag_llm = ChatGroq(model="mixtral-8x7b-32768")
|
17 |
+
|
18 |
+
st.title("β")
|
19 |
+
|
20 |
+
# Options for PDF input
|
21 |
+
pdf_source = st.radio("How would you like to provide a PDF?", ["Upload a PDF", "Enter a PDF URL"], index=0)
|
22 |
+
|
23 |
+
if pdf_source == "Upload a PDF":
|
24 |
+
uploaded_file = st.file_uploader("Upload your PDF file", type="pdf")
|
25 |
+
if uploaded_file:
|
26 |
+
with open("temp.pdf", "wb") as f:
|
27 |
+
f.write(uploaded_file.getbuffer())
|
28 |
+
pdf_path = "temp.pdf"
|
29 |
+
|
30 |
+
elif pdf_source == "Enter a PDF URL":
|
31 |
+
pdf_url = st.text_input("Enter PDF URL:")
|
32 |
+
if pdf_url:
|
33 |
+
try:
|
34 |
+
response = requests.get(pdf_url)
|
35 |
+
if response.status_code == 200:
|
36 |
+
with open("temp.pdf", "wb") as f:
|
37 |
+
f.write(response.content)
|
38 |
+
pdf_path = "temp.pdf"
|
39 |
+
st.success("β
PDF Downloaded Successfully!")
|
40 |
+
else:
|
41 |
+
st.error("β Failed to download PDF. Check the URL.")
|
42 |
+
pdf_path = None
|
43 |
+
except Exception as e:
|
44 |
+
st.error(f"Error downloading PDF: {e}")
|
45 |
+
pdf_path = None
|
46 |
+
else:
|
47 |
+
pdf_path = None
|