Update app.py

app.py

@@ -27,17 +27,29 @@ if uploaded_file:
 
     st.success("PDF Uploaded and Processed Successfully!")
     
-    # Display a text input box for the question
-    user_question = st.text_input("Ask a question about the Constitution:")
-    
+    # Split text into chunks to handle token limits
+    chunk_size = 5000  # Number of characters per chunk
+    chunks = [constitution_text[i:i+chunk_size] for i in range(0, len(constitution_text), chunk_size)]
+
+    # Let user select a chunk to ask questions about
+    st.write("The document is too large, so it's split into smaller parts. Select one to ask questions:")
+    selected_chunk = st.selectbox("Select a part of the document:", range(len(chunks)))
+    current_chunk = chunks[selected_chunk]
+
+    st.write("You selected part:", selected_chunk + 1)
+    st.text_area("Selected Text Segment", current_chunk, height=200)
+
+    # Ask a question about the selected chunk
+    user_question = st.text_input("Ask a question about the selected part:")
+
     if user_question:
         st.write("Processing your question...")
         
-        # Send the question to Groq API
+        # Send the selected chunk and question to Groq API
         try:
             chat_completion = client.chat.completions.create(
                 messages=[
-                    {"role": "user", "content": f"Context: {constitution_text}\nQuestion: {user_question}"}
+                    {"role": "user", "content": f"Context: {current_chunk}\nQuestion: {user_question}"}
                 ],
                 model="llama3-8b-8192",
             )