Update app.py

app.py

@@ -8,7 +8,7 @@ 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
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")  # Read 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.")
@@ -26,13 +26,14 @@ sections = {
 }
 
 # Helper function to extract text from PDF
-def extract_text_from_pdf(pdf_file, start_page, end_page):
+def extract_text_from_pdf(pdf_file, start_page, end_page, max_chars=4000):
+    """Extracts text from a specific range of pages in the PDF and limits text length."""
     try:
         reader = PdfReader(pdf_file)
         text = ""
         for page in range(start_page, end_page + 1):
             text += reader.pages[page].extract_text()
-        return text.strip()
+        return text[:max_chars].strip()  # Limit text to prevent token overflow
     except Exception as e:
         st.error(f"Error extracting text from PDF: {e}")
         return ""
@@ -53,17 +54,19 @@ if uploaded_file:
             question = st.text_input("Ask a question about this section:")
 
             if question:
-                # Interact with the Groq API
                 try:
                     st.info("Querying Groq API...")
+
                     chat_completion = client.chat.completions.create(
                         messages=[
                             {
                                 "role": "user",
-                                "content": f"Based on this section of the Pakistani Constitution: {section_text}\nQuestion: {question}",
+                                "content": f"Based on this section of the Pakistani Constitution: {section_text}\n\nQuestion: {question}",
                             }
                         ],
                         model="llama-3.3-70b-versatile",
+                        max_tokens=500,  # Limiting output tokens
+                        temperature=0.7  # Adjust for more/less randomness
                     )
 
                     answer = chat_completion.choices[0].message.content
@@ -72,3 +75,82 @@ if uploaded_file:
                     st.error(f"Error communicating with Groq API: {e}")
 else:
     st.warning("Please upload a PDF file to proceed.")
+
+
+
+
+
+# import os
+# import streamlit as st
+# from PyPDF2 import PdfReader
+# 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"])
+
+# # Predefined sections of the Constitution
+# sections = {
+#     "Preamble": (0, 1),
+#     "Fundamental Rights": (2, 10),
+#     # Add more sections with their page ranges here
+# }
+
+# # Helper function to extract text from PDF
+# def extract_text_from_pdf(pdf_file, start_page, end_page):
+#     try:
+#         reader = PdfReader(pdf_file)
+#         text = ""
+#         for page in range(start_page, end_page + 1):
+#             text += reader.pages[page].extract_text()
+#         return text.strip()
+#     except Exception as e:
+#         st.error(f"Error extracting text from PDF: {e}")
+#         return ""
+
+# # Section selection and Q&A functionality
+# if uploaded_file:
+#     st.success("PDF uploaded successfully.")
+#     selected_section = st.selectbox("Select a Section", list(sections.keys()))
+
+#     if selected_section:
+#         start_page, end_page = sections[selected_section]
+#         section_text = extract_text_from_pdf(uploaded_file, start_page, end_page)
+
+#         if section_text:
+#             st.text_area("Selected Section Text", section_text, height=300)
+
+#             # Question input
+#             question = st.text_input("Ask a question about this section:")
+
+#             if question:
+#                 # Interact with the Groq API
+#                 try:
+#                     st.info("Querying Groq API...")
+#                     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",
+#                     )
+
+#                     answer = chat_completion.choices[0].message.content
+#                     st.success(f"Answer: {answer}")
+#                 except Exception as e:
+#                     st.error(f"Error communicating with Groq API: {e}")
+# else:
+#     st.warning("Please upload a PDF file to proceed.")