Update pages/3_✍️_Writing_Assistant.py

pages/3_✍️_Writing_Assistant.py

@@ -1,375 +1,208 @@
-import streamlit as st
-from theme import apply_dark_theme, show_page_header, show_footer
-from utils import initialize_session_state, generate_document, save_to_history
-from datetime import datetime
-
-# Page config
-st.set_page_config(
-    page_title="Writing Assistant | CiviDoc AI",
-    page_icon="✍️",
-    layout="centered",
-    initial_sidebar_state="collapsed",
-)
-
-# Apply dark theme
-st.markdown(apply_dark_theme(), unsafe_allow_html=True)
-
-def writing_assistant_page():
-    # Initialize states
-    initialize_session_state()
-    
-    # Header
-    st.markdown(show_page_header(
-        "✍️ Writing Assistant",
-        "Create professional government documents effortlessly"
-    ), unsafe_allow_html=True)
-    
-    # Document Type Selection
-    st.markdown(
-        "<div class='card'>"
-        "<h3>Select Document Type</h3>"
-        "<p>Choose the type of document you need to create</p>"
-        "</div>",
-        unsafe_allow_html=True
-    )
-    
-    doc_types = {
-        "RTI": "📝 RTI Application",
-        "COMPLAINT": "📢 Complaint Letter",
-        "LEGAL": "⚖️ Legal Notice",
-        "APPEAL": "📨 Appeal Letter",
-        "PERMISSION": "🔑 Permission Request",
-        "APPLICATION": "📋 Government Application",
-        "CUSTOM": "✒️ Custom Document"
-    }
-    
-    # Mobile-friendly document type selector with icons
-    selected_type = st.selectbox(
-        "Document Type",
-        options=list(doc_types.keys()),
-        format_func=lambda x: doc_types[x],
-        key="doc_type_selector"
-    )
-    
-    # Common Fields Section - Always visible
-    st.markdown(
-        "<div class='card'>"
-        "<h4>Personal Information</h4>",
-        unsafe_allow_html=True
-    )
-    
-    name = st.text_input("Full Name", placeholder="Enter your full name")
-    address = st.text_area("Address", placeholder="Enter your complete address")
-    contact = st.text_input("Contact Number", placeholder="Enter your contact number")
-    email = st.text_input("Email Address", placeholder="Enter your email address")
-    
-    st.markdown("</div>", unsafe_allow_html=True)
-    
-    # Dynamic Form Based on Selection
-    if selected_type == "RTI":
-        show_rti_form(name, address, contact, email)
-    elif selected_type == "COMPLAINT":
-        show_complaint_form(name, address, contact, email)
-    elif selected_type == "LEGAL":
-        show_legal_notice_form(name, address, contact, email)
-    elif selected_type == "APPEAL":
-        show_appeal_form(name, address, contact, email)
-    elif selected_type == "PERMISSION":
-        show_permission_form(name, address, contact, email)
-    elif selected_type == "APPLICATION":
-        show_application_form(name, address, contact, email)
-    else:  # CUSTOM
-        show_custom_form(name, address, contact, email)
-
-def show_rti_form(name, address, contact, email):
-    """RTI Application Form"""
-    st.markdown(
-        "<div class='card'>"
-        "<h4>RTI Application Details</h4>",
-        unsafe_allow_html=True
-    )
-    
-    department = st.text_input("Department/Authority Name", placeholder="Enter department name")
-    subject = st.text_input("Subject of Information", placeholder="Enter subject")
-    
-    st.markdown("<div class='touch-spacing'>", unsafe_allow_html=True)
-    information = st.text_area(
-        "Information Required",
-        placeholder="Clearly specify the information you are seeking...",
-        height=150
-    )
-    
-    time_period = st.text_input(
-        "Time Period",
-        placeholder="Specify the time period for which information is sought"
-    )
-    
-    st.markdown("</div></div>", unsafe_allow_html=True)
-    
-    if validate_and_generate("RTI Application", locals()):
-        st.balloons()
-
-def show_complaint_form(name, address, contact, email):
-    """Complaint Letter Form"""
-    st.markdown(
-        "<div class='card'>"
-        "<h4>Complaint Details</h4>",
-        unsafe_allow_html=True
-    )
-    
-    authority = st.text_input("Authority/Department Name", placeholder="Enter authority name")
-    
-    complaint_types = [
-        "Public Service",
-        "Infrastructure",
-        "Government Employee",
-        "Civic Issue",
-        "Other"
-    ]
-    
-    complaint_type = st.selectbox("Type of Complaint", complaint_types)
-    
-    st.markdown("<div class='touch-spacing'>", unsafe_allow_html=True)
-    description = st.text_area(
-        "Complaint Description",
-        placeholder="Describe your complaint in detail...",
-        height=150
-    )
-    
-    previous_complaints = st.text_area(
-        "Previous Complaints (if any)",
-        placeholder="Mention any previous complaints filed regarding this issue..."
-    )
-    
-    st.markdown("</div></div>", unsafe_allow_html=True)
-    
-    if validate_and_generate("Complaint Letter", locals()):
-        st.balloons()
-
-def show_legal_notice_form(name, address, contact, email):
-    """Legal Notice Form"""
-    st.markdown(
-        "<div class='card'>"
-        "<h4>Legal Notice Details</h4>",
-        unsafe_allow_html=True
-    )
-    
-    recipient = st.text_input("Notice To (Name/Department)", placeholder="Enter recipient's name")
-    recipient_address = st.text_area("Recipient's Address", placeholder="Enter recipient's address")
-    
-    st.markdown("<div class='touch-spacing'>", unsafe_allow_html=True)
-    subject = st.text_input("Subject of Notice", placeholder="Enter notice subject")
-    cause = st.text_area(
-        "Cause of Action",
-        placeholder="Describe the reason for this legal notice...",
-        height=100
-    )
-    
-    relief_sought = st.text_area(
-        "Relief Sought",
-        placeholder="Specify what action you want taken...",
-        height=100
-    )
-    
-    time_period = st.number_input(
-        "Response Time Period (in days)",
-        min_value=1,
-        value=15
-    )
-    
-    st.markdown("</div></div>", unsafe_allow_html=True)
-    
-    if validate_and_generate("Legal Notice", locals()):
-        st.balloons()
-
-def show_appeal_form(name, address, contact, email):
-    """Appeal Letter Form"""
-    st.markdown(
-        "<div class='card'>"
-        "<h4>Appeal Details</h4>",
-        unsafe_allow_html=True
-    )
-    
-    authority = st.text_input("Appellate Authority", placeholder="Enter authority name")
-    reference = st.text_input("Previous Reference/Order Number", placeholder="Enter reference number")
-    
-    st.markdown("<div class='touch-spacing'>", unsafe_allow_html=True)
-    order_date = st.date_input("Date of Previous Order")
-    
-    grounds = st.text_area(
-        "Grounds for Appeal",
-        placeholder="Explain the reasons for your appeal...",
-        height=150
-    )
-    
-    relief = st.text_area(
-        "Relief Sought",
-        placeholder="Specify what you are seeking through this appeal...",
-        height=100
-    )
-    
-    st.markdown("</div></div>", unsafe_allow_html=True)
-    
-    if validate_and_generate("Appeal Letter", locals()):
-        st.balloons()
-
-def show_permission_form(name, address, contact, email):
-    """Permission Request Form"""
-    st.markdown(
-        "<div class='card'>"
-        "<h4>Permission Request Details</h4>",
-        unsafe_allow_html=True
-    )
-    
-    authority = st.text_input("Authority Name", placeholder="Enter authority name")
-    purpose = st.text_input("Purpose of Request", placeholder="Enter the purpose")
-    
-    st.markdown("<div class='touch-spacing'>", unsafe_allow_html=True)
-    details = st.text_area(
-        "Request Details",
-        placeholder="Provide detailed information about your request...",
-        height=150
-    )
-    
-    duration = st.text_input("Duration (if applicable)", placeholder="Specify time period")
-    location = st.text_input("Location (if applicable)", placeholder="Specify location")
-    
-    undertaking = st.text_area(
-        "Undertaking/Declaration",
-        placeholder="Any declarations or undertakings...",
-        height=100
-    )
-    
-    st.markdown("</div></div>", unsafe_allow_html=True)
-    
-    if validate_and_generate("Permission Request", locals()):
-        st.balloons()
-
-def show_application_form(name, address, contact, email):
-    """Government Application Form"""
-    st.markdown(
-        "<div class='card'>"
-        "<h4>Application Details</h4>",
-        unsafe_allow_html=True
-    )
-    
-    department = st.text_input("Department Name", placeholder="Enter department name")
-    purpose = st.text_input("Purpose of Application", placeholder="Enter purpose")
-    
-    application_types = [
-        "License",
-        "Certificate",
-        "Registration",
-        "Permit",
-        "Other"
-    ]
-    
-    app_type = st.selectbox("Application Type", application_types)
-    
-    st.markdown("<div class='touch-spacing'>", unsafe_allow_html=True)
-    details = st.text_area(
-        "Application Details",
-        placeholder="Provide detailed information...",
-        height=150
-    )
-    
-    supporting_docs = st.text_area(
-        "Supporting Documents",
-        placeholder="List all supporting documents...",
-        height=100
-    )
-    
-    st.markdown("</div></div>", unsafe_allow_html=True)
-    
-    if validate_and_generate("Government Application", locals()):
-        st.balloons()
-
-def show_custom_form(name, address, contact, email):
-    """Custom Document Form"""
-    st.markdown(
-        "<div class='card'>"
-        "<h4>Custom Document Details</h4>",
-        unsafe_allow_html=True
-    )
-    
-    title = st.text_input("Document Title", placeholder="Enter document title")
-    recipient = st.text_input("Recipient/Authority", placeholder="Enter recipient name")
-    
-    st.markdown("<div class='touch-spacing'>", unsafe_allow_html=True)
-    subject = st.text_input("Subject", placeholder="Enter subject")
-    
-    content = st.text_area(
-        "Document Content",
-        placeholder="Enter the main content of your document...",
-        height=300
-    )
-    
-    st.markdown("</div></div>", unsafe_allow_html=True)
-    
-    if validate_and_generate("Custom Document", locals()):
-        st.balloons()
-
-def validate_and_generate(doc_type, fields):
-    """Validate fields and generate document"""
-    if st.button(f"Generate {doc_type}", use_container_width=True):
-        # Basic validation
-        required_fields = {k: v for k, v in fields.items() 
-                         if k not in ['st', 'contact', 'email']}
-        
-        empty_fields = [k for k, v in required_fields.items() 
-                       if not v or (isinstance(v, str) and not v.strip())]
-        
-        if empty_fields:
-            st.error(
-                f"Please fill in all required fields: "
-                f"{', '.join(empty_fields)}"
-            )
-            return False
-        
-        try:
-            with st.spinner("Generating document..."):
-                # Generate document
-                generated_content = generate_document(doc_type, fields)
-                
-                # Save to history
-                timestamp = datetime.now()
-                doc_name = f"{doc_type}_{timestamp.strftime('%Y%m%d_%H%M%S')}"
-                save_to_history(doc_name, doc_type, generated_content, timestamp)
-                
-                # Show success message
-                st.success("Document generated successfully!")
-                
-                # Display generated document
-                st.markdown(
-                    "<div class='card'>"
-                    "<h4>Generated Document</h4>"
-                    f"<pre style='white-space: pre-wrap;'>{generated_content}</pre>"
-                    "</div>",
-                    unsafe_allow_html=True
-                )
-                
-                # Action buttons
-                col1, col2 = st.columns(2)
-                with col1:
-                    st.download_button(
-                        "📥 Download Document",
-                        generated_content,
-                        file_name=f"{doc_name}.txt",
-                        mime="text/plain",
-                        use_container_width=True
-                    )
-                
-                with col2:
-                    if st.button("📋 Create Another", use_container_width=True):
-                        st.rerun()
-                
-                return True
-                
-        except Exception as e:
-            st.error(f"Error generating document: {str(e)}")
-            return False
-    
-    return False
-
-if __name__ == "__main__":
+import streamlit as st
+from theme import apply_dark_theme, show_page_header, show_footer
+from utils import initialize_session_state, generate_document, save_to_history
+from datetime import datetime
+
+# Page config
+st.set_page_config(
+    page_title="Writing Assistant | CiviDoc AI",
+    page_icon="✍️",
+    layout="centered",
+    initial_sidebar_state="collapsed",
+)
+
+# Apply dark theme
+st.markdown(apply_dark_theme(), unsafe_allow_html=True)
+
+def writing_assistant_page():
+    # Initialize states
+    initialize_session_state()
+    
+    # Header
+    st.markdown(show_page_header(
+        "✍️ Writing Assistant",
+        "Create professional government documents effortlessly"
+    ), unsafe_allow_html=True)
+    
+    # Document Type Selection
+    st.markdown(
+        "<div class='card'>"
+        "<h3>Select Document Type</h3>"
+        "<p>Choose the type of document you need to create</p>"
+        "</div>",
+        unsafe_allow_html=True
+    )
+    
+    doc_categories = {
+        "Personal Identification and Certificates": ["Aadhar Card", "Birth Certificate", "Death Certificate", "Marriage Certificate", "Name Change Certificate", "Domicile Certificate"],
+        "Financial and Tax-Related Applications": ["PAN Card", "Tax Identification Number (TIN)", "Income Certificate", "Property Tax Payment"],
+        "Licenses and Permits": ["Driver's License", "Passport Application", "Firearm License", "Fishing or Hunting Permit", "Building Permit", "Business License"],
+        "Educational and Employment Applications": ["Scholarship Applications", "Student Loan Application", "Government Job Application", "No Objection Certificate (NOC)", "Employee Provident Fund (EPF) Withdrawal"],
+        "Healthcare-Related Applications": ["Health Insurance Application", "Disability Certificate", "Medical Leave Certificate"],
+        "Social Welfare and Subsidy Applications": ["Pension Application", "Unemployment Benefits", "Ration Card", "Senior Citizen Card", "Housing Assistance or Loan Application"],
+        "Environmental and Utility Permissions": ["Electricity and Water Connection", "Gas Connection", "Sanitation Permit", "Pollution Clearance Certificate"],
+        "Legal and Civic Documents": ["Right to Information (RTI)", "Public Grievance Application", "Affidavit for Name Change or Correction", "Legal Aid Application"]
+    }
+
+    selected_category = st.selectbox(
+        "Document Category",
+        options=list(doc_categories.keys())
+    )
+    
+    selected_type = st.selectbox(
+        "Document Type",
+        options=doc_categories[selected_category]
+    )
+    
+    # Common Fields Section - Always visible
+    st.markdown(
+        "<div class='card'>"
+        "<h4>Personal Information</h4>",
+        unsafe_allow_html=True
+    )
+    
+    name = st.text_input("Full Name", placeholder="Enter your full name")
+    address = st.text_area("Address", placeholder="Enter your complete address")
+    contact = st.text_input("Contact Number", placeholder="Enter your contact number")
+    email = st.text_input("Email Address", placeholder="Enter your email address")
+    
+    st.markdown("</div>", unsafe_allow_html=True)
+    
+    # Dynamic Form Based on Selection
+    if selected_type == "Aadhar Card":
+        show_aadhar_card_form(name, address, contact, email)
+    elif selected_type == "Birth Certificate":
+        show_birth_certificate_form(name, address, contact, email)
+    elif selected_type == "Death Certificate":
+        show_death_certificate_form(name, address, contact, email)
+    elif selected_type == "Marriage Certificate":
+        show_marriage_certificate_form(name, address, contact, email)
+    # Add all additional forms following the same pattern for new documents...
+
+def show_aadhar_card_form(name, address, contact, email):
+    """Aadhar Card Application Form"""
+    st.markdown(
+        "<div class='card'>"
+        "<h4>Aadhar Card Details</h4>",
+        unsafe_allow_html=True
+    )
+    
+    uidai_center = st.text_input("UIDAI Center", placeholder="Enter UIDAI center")
+    date_of_birth = st.date_input("Date of Birth")
+    gender = st.selectbox("Gender", ["Male", "Female", "Other"])
+    
+    if validate_and_generate("Aadhar Card Application", locals()):
+        st.balloons()
+
+def show_birth_certificate_form(name, address, contact, email):
+    """Birth Certificate Application Form"""
+    st.markdown(
+        "<div class='card'>"
+        "<h4>Birth Certificate Details</h4>",
+        unsafe_allow_html=True
+    )
+    
+    birth_date = st.date_input("Date of Birth")
+    place_of_birth = st.text_input("Place of Birth", placeholder="Enter place of birth")
+    parent_name = st.text_input("Parent's Name", placeholder="Enter parent's name")
+    
+    if validate_and_generate("Birth Certificate Application", locals()):
+        st.balloons()
+
+def show_death_certificate_form(name, address, contact, email):
+    """Death Certificate Application Form"""
+    st.markdown(
+        "<div class='card'>"
+        "<h4>Death Certificate Details</h4>",
+        unsafe_allow_html=True
+    )
+    
+    date_of_death = st.date_input("Date of Death")
+    place_of_death = st.text_input("Place of Death", placeholder="Enter place of death")
+    relation = st.text_input("Relation to Deceased", placeholder="Enter relation")
+    
+    if validate_and_generate("Death Certificate Application", locals()):
+        st.balloons()
+
+def show_marriage_certificate_form(name, address, contact, email):
+    """Marriage Certificate Application Form"""
+    st.markdown(
+        "<div class='card'>"
+        "<h4>Marriage Certificate Details</h4>",
+        unsafe_allow_html=True
+    )
+    
+    spouse_name = st.text_input("Spouse Name", placeholder="Enter spouse's name")
+    marriage_date = st.date_input("Date of Marriage")
+    place_of_marriage = st.text_input("Place of Marriage", placeholder="Enter place")
+    
+    if validate_and_generate("Marriage Certificate Application", locals()):
+        st.balloons()
+
+# Add the other document forms here in the same way:
+# E.g. `show_pan_card_form`, `show_driver_license_form`, etc.
+
+def validate_and_generate(doc_type, fields):
+    """Validate fields and generate document"""
+    if st.button(f"Generate {doc_type}", use_container_width=True):
+        # Basic validation
+        required_fields = {k: v for k, v in fields.items() 
+                         if k not in ['st', 'contact', 'email']}
+        
+        empty_fields = [k for k, v in required_fields.items() 
+                       if not v or (isinstance(v, str) and not v.strip())]
+        
+        if empty_fields:
+            st.error(
+                f"Please fill in all required fields: "
+                f"{', '.join(empty_fields)}"
+            )
+            return False
+        
+        try:
+            with st.spinner("Generating document..."):
+                # Generate document
+                generated_content = generate_document(doc_type, fields)
+                
+                # Save to history
+                timestamp = datetime.now()
+                doc_name = f"{doc_type}_{timestamp.strftime('%Y%m%d_%H%M%S')}"
+                save_to_history(doc_name, doc_type, generated_content, timestamp)
+                
+                # Show success message
+                st.success("Document generated successfully!")
+                
+                # Display generated document
+                st.markdown(
+                    "<div class='card'>"
+                    "<h4>Generated Document</h4>"
+                    f"<pre style='white-space: pre-wrap;'>{generated_content}</pre>"
+                    "</div>",
+                    unsafe_allow_html=True
+                )
+                
+                # Action buttons
+                col1, col2 = st.columns(2)
+                with col1:
+                    st.download_button(
+                        "📥 Download Document",
+                        generated_content,
+                        file_name=f"{doc_name}.txt",
+                        mime="text/plain",
+                        use_container_width=True
+                    )
+                
+                with col2:
+                    if st.button("📋 Create Another", use_container_width=True):
+                        st.rerun()
+                
+                return True
+                
+        except Exception as e:
+            st.error(f"Error generating document: {str(e)}")
+            return False
+    
+    return False
+
+if __name__ == "__main__":
     writing_assistant_page()