DrishtiSharma commited on
Commit
b604a12
Β·
verified Β·
1 Parent(s): aa31b3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +31 -0
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