Update app.py

app.py

@@ -925,128 +925,128 @@ def main():
                         form_response = tax_agent_response(form_question, tax_form_type=form_type)
                         st.markdown(form_response)
     # Tab 3: Upload Form
-with tab3:
-    st.header("Upload Your Tax Form")
-    st.markdown("Upload a PDF tax form to analyze, get summaries, and ask questions about it.")
-    
-    # File uploader
-    uploaded_file = st.file_uploader(
-        "Choose a PDF file", 
-        type=['pdf'], 
-        help="Upload any Pakistani tax form PDF for analysis"
-    )
-    
-    if uploaded_file is not None:
-        # Store uploaded file in session state
-        st.session_state.pdf_bytes = BytesIO(uploaded_file.read())
-        st.session_state.uploaded_file_name = uploaded_file.name
-        
-        # Display success message
-        st.success(f"✅ Successfully uploaded: {uploaded_file.name}")
-        
-        # Extract form fields from uploaded PDF
-        with st.spinner("Analyzing uploaded form..."):
-            st.session_state.form_fields = extract_form_fields(st.session_state.pdf_bytes)
-        
-        # Display PDF preview
-        st.markdown("### 📄 Form Preview")
-        display_pdf(st.session_state.pdf_bytes)
-        
-        # Generate summary using LLM
-        if GROQ_API_KEY:
-            with st.spinner("Generating form summary..."):
-                summary = generate_form_summary(st.session_state.form_fields, uploaded_file.name)
-                if summary:
-                    st.markdown("### 📋 Form Summary")
-                    st.markdown(summary)
-        
-        # Show form fields if available
-        if st.session_state.form_fields:
-            st.markdown("### 📝 Form Fields")
+    with tab3:
+        st.header("Upload Your Tax Form")
+        st.markdown("Upload a PDF tax form to analyze, get summaries, and ask questions about it.")
+        
+        # File uploader
+        uploaded_file = st.file_uploader(
+            "Choose a PDF file", 
+            type=['pdf'], 
+            help="Upload any Pakistani tax form PDF for analysis"
+        )
+        
+        if uploaded_file is not None:
+            # Store uploaded file in session state
+            st.session_state.pdf_bytes = BytesIO(uploaded_file.read())
+            st.session_state.uploaded_file_name = uploaded_file.name
             
-            # Get field explanations if LLM is available
-            field_explanations = {}
+            # Display success message
+            st.success(f"✅ Successfully uploaded: {uploaded_file.name}")
+            
+            # Extract form fields from uploaded PDF
+            with st.spinner("Analyzing uploaded form..."):
+                st.session_state.form_fields = extract_form_fields(st.session_state.pdf_bytes)
+            
+            # Display PDF preview
+            st.markdown("### 📄 Form Preview")
+            display_pdf(st.session_state.pdf_bytes)
+            
+            # Generate summary using LLM
             if GROQ_API_KEY:
-                with st.spinner("Analyzing form fields..."):
-                    explanations = explain_form_fields(
-                        st.session_state.form_fields, 
-                        "Pakistan", 
-                        uploaded_file.name
-                    )
-                    if explanations:
-                        field_explanations = explanations
+                with st.spinner("Generating form summary..."):
+                    summary = generate_form_summary(st.session_state.form_fields, uploaded_file.name)
+                    if summary:
+                        st.markdown("### 📋 Form Summary")
+                        st.markdown(summary)
             
-            # Display fields with explanations if available
-            if field_explanations and "sections" in field_explanations:
-                for section in field_explanations["sections"]:
-                    with st.expander(f"📑 {section['name']}"):
-                        st.write(section["explanation"])
-                        for field_name in section["fields"]:
-                            matching_fields = [f for f in st.session_state.form_fields 
-                                            if f["name"] == field_name]
-                            if matching_fields:
-                                field = matching_fields[0]
-                                st.write(f"**{field['name']}** ({field['type']})")
-            else:
-                # Simple field display without explanations
-                for field in st.session_state.form_fields:
-                    st.write(f"**{field['name']}** ({field['type']})")
-        
-        else:
-            st.warning("⚠️ This form doesn't have fillable fields or couldn't be analyzed.")
-        
-        # Q&A Section for uploaded form
-        st.markdown("### ❓ Ask Questions About This Form")
-        st.markdown("Get help understanding your uploaded tax form.")
-        
-        upload_question = st.text_input(
-            "Ask a question about this form:", 
-            placeholder="What is this form used for? How do I fill section X?",
-            key="upload_question_input"
-        )
-        
-        if upload_question:
-            with st.spinner("Analyzing your question..."):
-                # Generate response using the uploaded form context
-                upload_response = generate_upload_form_response(
-                    upload_question, 
-                    uploaded_file.name,
-                    st.session_state.form_fields
-                )
+            # Show form fields if available
+            if st.session_state.form_fields:
+                st.markdown("### 📝 Form Fields")
+                
+                # Get field explanations if LLM is available
+                field_explanations = {}
+                if GROQ_API_KEY:
+                    with st.spinner("Analyzing form fields..."):
+                        explanations = explain_form_fields(
+                            st.session_state.form_fields, 
+                            "Pakistan", 
+                            uploaded_file.name
+                        )
+                        if explanations:
+                            field_explanations = explanations
                 
-                st.markdown("### 💬 Response:")
-                st.markdown(upload_response)
-        
-        # Download options
-        st.markdown("### 📥 Download Options")
-        col1, col2 = st.columns(2)
-        
-        with col1:
-            # Download original form
-            st.download_button(
-                label="📥 Download Original Form",
-                data=st.session_state.pdf_bytes,
-                file_name=uploaded_file.name,
-                mime="application/pdf"
+                # Display fields with explanations if available
+                if field_explanations and "sections" in field_explanations:
+                    for section in field_explanations["sections"]:
+                        with st.expander(f"📑 {section['name']}"):
+                            st.write(section["explanation"])
+                            for field_name in section["fields"]:
+                                matching_fields = [f for f in st.session_state.form_fields 
+                                                if f["name"] == field_name]
+                                if matching_fields:
+                                    field = matching_fields[0]
+                                    st.write(f"**{field['name']}** ({field['type']})")
+                else:
+                    # Simple field display without explanations
+                    for field in st.session_state.form_fields:
+                        st.write(f"**{field['name']}** ({field['type']})")
+            
+            else:
+                st.warning("⚠️ This form doesn't have fillable fields or couldn't be analyzed.")
+            
+            # Q&A Section for uploaded form
+            st.markdown("### ❓ Ask Questions About This Form")
+            st.markdown("Get help understanding your uploaded tax form.")
+            
+            upload_question = st.text_input(
+                "Ask a question about this form:", 
+                placeholder="What is this form used for? How do I fill section X?",
+                key="upload_question_input"
             )
+            
+            if upload_question:
+                with st.spinner("Analyzing your question..."):
+                    # Generate response using the uploaded form context
+                    upload_response = generate_upload_form_response(
+                        upload_question, 
+                        uploaded_file.name,
+                        st.session_state.form_fields
+                    )
+                    
+                    st.markdown("### 💬 Response:")
+                    st.markdown(upload_response)
+            
+            # Download options
+            st.markdown("### 📥 Download Options")
+            col1, col2 = st.columns(2)
+            
+            with col1:
+                # Download original form
+                st.download_button(
+                    label="📥 Download Original Form",
+                    data=st.session_state.pdf_bytes,
+                    file_name=uploaded_file.name,
+                    mime="application/pdf"
+                )
+            
+            with col2:
+                # Option to fill form if it has fields
+                if st.session_state.form_fields:
+                    if st.button("✏️ Fill This Form"):
+                        st.session_state.is_filling = True
+                        st.rerun()
         
-        with col2:
-            # Option to fill form if it has fields
-            if st.session_state.form_fields:
-                if st.button("✏️ Fill This Form"):
-                    st.session_state.is_filling = True
-                    st.rerun()
-    
-    else:
-        st.info("👆 Please upload a PDF tax form to get started.")
-        st.markdown("""
-        **What you can do with uploaded forms:**
-        - ��� View and preview the form
-        - 📋 Get AI-generated summaries
-        - 🔍 Analyze form fields and sections
-        - ❓ Ask questions about the form
-        - ✏️ Fill out interactive forms
-        """)
+        else:
+            st.info("👆 Please upload a PDF tax form to get started.")
+            st.markdown("""
+            **What you can do with uploaded forms:**
+            - 📄 View and preview the form
+            - 📋 Get AI-generated summaries
+            - 🔍 Analyze form fields and sections
+            - ❓ Ask questions about the form
+            - ✏️ Fill out interactive forms
+            """)
 
 if __name__ == "__main__":
     main()