Update app.py

app.py

@@ -1,12 +1,20 @@
 import os
 import streamlit as st
 from PyPDF2 import PdfReader
-import requests
+from groq import Groq
 
 # Set up the Streamlit app
 st.title("Pakistani Constitution Q&A App")
 st.write("Upload the Pakistani Constitution PDF, explore sections, and ask questions.")
 
+# Initialize the Groq API client
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")  # Read the API key from environment variables
+
+if not GROQ_API_KEY:
+    st.error("Groq API key not found. Please set the 'GROQ_API_KEY' environment variable in Hugging Face Spaces settings.")
+else:
+    client = Groq(api_key=GROQ_API_KEY)
+
 # Upload PDF
 uploaded_file = st.file_uploader("Upload PDF", type=["pdf"])
 
@@ -17,12 +25,6 @@ sections = {
     # Add more sections with their page ranges here
 }
 
-# Groq API configuration
-GROQ_API_KEY = os.getenv("GROQ_API_KEY")  # Read the API key from environment variables
-
-if not GROQ_API_KEY:
-    st.error("Groq API key not found. Please set the 'GROQ_API_KEY' environment variable in Hugging Face Spaces settings.")
-
 # Helper function to extract text from PDF
 def extract_text_from_pdf(pdf_file, start_page, end_page):
     try:
@@ -35,7 +37,7 @@ def extract_text_from_pdf(pdf_file, start_page, end_page):
         st.error(f"Error extracting text from PDF: {e}")
         return ""
 
-# Section selection
+# Section selection and Q&A functionality
 if uploaded_file:
     st.success("PDF uploaded successfully.")
     selected_section = st.selectbox("Select a Section", list(sections.keys()))
@@ -51,21 +53,22 @@ if uploaded_file:
             question = st.text_input("Ask a question about this section:")
 
             if question:
-                # Make API call to Groq
+                # Interact with the Groq API
                 try:
                     st.info("Querying Groq API...")
-                    response = requests.post(
-                        "https://api.groq.com/v1/ask",
-                        headers={"Authorization": f"Bearer {GROQ_API_KEY}"},
-                        json={"context": section_text, "question": question},
+                    chat_completion = client.chat.completions.create(
+                        messages=[
+                            {
+                                "role": "user",
+                                "content": f"Based on this section of the Pakistani Constitution: {section_text}\nQuestion: {question}",
+                            }
+                        ],
+                        model="llama-3.3-70b-versatile",
                     )
 
-                    if response.status_code == 200:
-                        answer = response.json().get("answer", "No answer found.")
-                        st.success(f"Answer: {answer}")
-                    else:
-                        st.error(f"Error: {response.json().get('error', 'Unknown error occurred')}")
+                    answer = chat_completion.choices[0].message.content
+                    st.success(f"Answer: {answer}")
                 except Exception as e:
-                    st.error(f"API call failed: {e}")
+                    st.error(f"Error communicating with Groq API: {e}")
 else:
     st.warning("Please upload a PDF file to proceed.")