Update app.py

app.py

@@ -1,20 +1,32 @@
 import streamlit as st
 import os
 from together import Together
+import PyPDF2
 
-# Initialize the Llama-3 client
 client = Together(api_key=os.environ.get("TOGETHER_API_KEY"))
 
-def get_response(question):
+def get_response(question, pdf_files):
+    texts = [get_pdf_content(pdf) for pdf in pdf_files]
+    combined_text = "\n".join(texts)
     response = client.chat.completions.create(
         model="meta-llama/Llama-3-8b-chat-hf",
         messages=[{"role": "user", "content": question}],
+        context=combined_text
     )
     return response.choices[0].message.content
 
-# Setting up Streamlit interface
-st.title('RAG based PDF Chatbot')
-user_input = st.text_input("Ask me anything!")
-if user_input:
-    answer = get_response(user_input)
-    st.text_area("Answer:", value=answer, height=300)
+def get_pdf_content(pdf_file):
+    reader = PyPDF2.PdfReader(pdf_file)
+    text = ""
+    for page in reader.pages:
+        text += page.extract_text()
+    return text
+
+st.title('Llama-3 Chatbot with PDF Input')
+uploaded_files = st.file_uploader("Upload PDF", accept_multiple_files=True, type=['pdf'])
+question = st.text_input("Ask me anything based on the uploaded PDFs!")
+
+if st.button("Answer"):
+    if uploaded_files and question:
+        answer = get_response(question, uploaded_files)
+        st.text_area("Answer:", value=answer, height=300)