File size: 3,696 Bytes
6ed23c5 73fc345 6ed23c5 73fc345 6ed23c5 a41f426 6ed23c5 8dcddc4 6ed23c5 8dcddc4 6ed23c5 8dcddc4 6ed23c5 8dcddc4 6ed23c5 8dcddc4 6ed23c5 8dcddc4 6ed23c5 8dcddc4 |
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 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 |
import streamlit as st
from groq import Groq
import pdfplumber
import os
import gc
# No need for load_dotenv() since Hugging Face automatically loads secrets as environment variables
api_key = os.getenv("GROQ_API_KEY")
if not api_key:
st.error("Groq API key not found. Please check your environment variables.")
st.stop() # Stop execution if API key is missing
# Initialize the Groq client with the API key
client = Groq(api_key=api_key)
def generate_response_groq(context, query):
"""Generate response using Groq API."""
prompt = f"You are a text generating model, given you a context and query, you have to answer the query in detail. Context: {context}\nQuestion: {query}\nAnswer: Your answer must be descriptive, into 2 lines atleast, also if you don't know any questions answer just say :I don't know"
chat_completion = client.chat.completions.create(
messages=[
{
"role": "user",
"content": prompt,
}
],
model="llama3-8b-8192",
)
response = chat_completion.choices[0].message.content
return response
def extract_text_from_pdf(pdf_file):
"""Extract text from PDF file using pdfplumber."""
text = ""
with pdfplumber.open(pdf_file) as pdf:
for page in pdf.pages:
text += page.extract_text() or ""
return text
# Set the page layout to wide for better UI space
st.set_page_config(page_title="PDF Query Application", layout="wide")
# Sidebar
st.sidebar.title("PDF Query Assistant")
st.sidebar.image("https://upload.wikimedia.org/wikipedia/commons/thumb/a/af/PDF_file_icon.svg/1024px-PDF_file_icon.svg.png", use_column_width=True) # Adding an image in the sidebar
st.sidebar.markdown("### Navigation")
st.sidebar.markdown("Use this app to upload a PDF and ask questions about its content.")
st.sidebar.markdown("")
# Main UI layout
st.title("📄 PDF Query Application")
st.markdown("""
<style>
.main-content {background-color: #f0f2f6; padding: 20px; border-radius: 10px;}
.stButton>button {background-color: #4CAF50; color: white; font-size: 16px; border-radius: 10px;}
.stTextInput>div>div>input {background-color: #f0f2f6; color: black; border-radius: 5px;}
</style>
""", unsafe_allow_html=True)
st.markdown("<div class='main-content'>", unsafe_allow_html=True)
uploaded_file = st.file_uploader("Upload a PDF file", type="pdf")
if uploaded_file:
st.success("PDF uploaded successfully! 📄")
document_text = extract_text_from_pdf(uploaded_file)
st.text_area("📜 Extracted Text", document_text, height=200)
query = st.text_input("🔍 Enter your query")
if st.button("💬 Get Answer"):
if query:
with st.spinner("Generating response..."):
response = generate_response_groq(document_text, query)
st.write("**Response:**")
st.write(response)
# Clear memory after generating response
gc.collect()
else:
st.error("Please enter a query.")
st.markdown("</div>", unsafe_allow_html=True)
# Footer
st.sidebar.markdown("### About")
st.sidebar.info("Developed with ❤️ using Streamlit and Groq API.")
st.sidebar.markdown("---")
st.sidebar.write("For more information, visit [Groq](https://www.groq.com) and [Streamlit](https://streamlit.io).")
# Customize the theme and color contrast
st.markdown("""
<style>
.css-1aumxhk {background-color: #E8EAF6;}
.stTextInput>div>div>input {border-color: #3f51b5;}
.stTextArea>div>div>textarea {border-color: #3f51b5;}
.stButton>button {background-color: #3f51b5; color: white; font-size: 16px;}
</style>
""", unsafe_allow_html=True)
|