Upload 9 files

.streamlit/config.toml

@@ -0,0 +1,29 @@
+[theme]
+primaryColor = "#3B82F6"
+backgroundColor = "#0F172A"
+secondaryBackgroundColor = "#1E293B"
+textColor = "#E2E8F0"
+font = "sans serif"
+
+[server]
+maxUploadSize = 200
+enableXsrfProtection = true
+enableCORS = false
+
+[browser]
+gatherUsageStats = false
+
+[client]
+toolbarMode = "minimal"
+showSidebarNavigation = true
+
+[runner]
+magicEnabled = true
+fastRerenderEnabled = true
+
+[logger]
+<<<<<<< HEAD
+level = "info"
+=======
+level = "info"
+>>>>>>> 1bc20a0d3edc7f88f03e506f84b01a7303d403b2

pages/1_📝_Document_Analysis.py

@@ -0,0 +1,263 @@
+import streamlit as st
+from PIL import Image
+import datetime
+from theme import apply_dark_theme, show_page_header, show_footer
+from utils import (
+    process_image, 
+    process_pdf, 
+    initialize_session_state, 
+    create_chat_engine,
+    save_to_history,
+    generate_pdf_analysis,
+    process_captured_image,
+    format_analysis_results
+)
+
+# Page config
+st.set_page_config(
+    page_title="Document Analysis |  CiviDoc AI",
+    page_icon="📝",
+    layout="centered",
+    initial_sidebar_state="collapsed",
+)
+
+# Apply dark theme
+st.markdown(apply_dark_theme(), unsafe_allow_html=True)
+
+def document_analysis_page():
+    # Initialize states
+    initialize_session_state()
+    
+    # Header
+    st.markdown(show_page_header(
+        "📝 Document Analysis",
+        "Upload or capture documents for instant analysis"
+    ), unsafe_allow_html=True)
+    
+    # Main content tabs - Mobile friendly
+    tabs = st.tabs([
+        "📤 Upload ",
+        "📸 Capture",
+        "📊 Results"
+    ])
+    
+    with tabs[0]:
+        st.markdown(
+            "<div class='card'>"
+            "<h3>Upload Documents</h3>"
+            "<p style='margin-bottom: 2rem;'>Support for PDF, JPG, JPEG, and PNG files.</p>"
+            "<div class='status-badge status-success'>Multiple files supported</div>"
+            "</div>",
+            unsafe_allow_html=True
+        )
+        
+        # Mobile-friendly file uploader
+        uploaded_files = st.file_uploader(
+            "Drop files or tap to browse",
+            type=["jpg", "jpeg", "png", "pdf"],
+            accept_multiple_files=True,
+            key="doc_uploader"
+        )
+        
+        if uploaded_files:
+            st.markdown(
+                f"<div class='status-badge status-success' style='margin: 1rem 0;'>"
+                f"📎 {len(uploaded_files)} file(s) uploaded"
+                f"</div>",
+                unsafe_allow_html=True
+            )
+            
+            # File list - Mobile friendly
+            st.markdown(
+                "<div class='card'>"
+                "<h4>Selected Files:</h4>"
+                "<div class='touch-spacing'>",
+                unsafe_allow_html=True
+            )
+            
+            for file in uploaded_files:
+                st.markdown(
+                    f"<div style='display: flex; align-items: center; padding: 0.5rem 0;'>"
+                    f"<span style='margin-right: 0.5rem;'>📄</span>{file.name}"
+                    f"</div>",
+                    unsafe_allow_html=True
+                )
+            
+            st.markdown("</div></div>", unsafe_allow_html=True)
+            
+            # Analysis button - Touch friendly
+            if st.button("🔍 Analyze Documents", use_container_width=True):
+                process_uploaded_files(uploaded_files)
+    
+    with tabs[1]:
+        st.markdown(
+            "<div class='card'>"
+            "<h3>Capture Document</h3>"
+            "<p>Take a clear photo of your document using your camera.</p>"
+            "</div>",
+            unsafe_allow_html=True
+        )
+        
+        # Mobile-optimized camera input
+        picture = st.camera_input(
+            "📸 Tap to capture",
+            help="Make sure the document is well-lit and clearly visible"
+        )
+        
+        if picture:
+            st.markdown(
+                "<div class='status-badge status-success' style='margin: 1rem 0;'>"
+                "📸 Image captured successfully"
+                "</div>",
+                unsafe_allow_html=True
+            )
+            
+            # Analysis button - Touch friendly
+            if st.button("🔍 Analyze Photo", use_container_width=True):
+                process_captured_image(picture)
+    
+    with tabs[2]:
+        display_analysis_results()
+
+def process_uploaded_files(files):
+    """Process multiple uploaded files with mobile-friendly progress tracking"""
+    progress_text = "Processing documents..."
+    total_files = len(files)
+    
+    # Progress tracking
+    progress_placeholder = st.empty()
+    progress_bar = st.progress(0)
+    status_text = st.empty()
+    
+    for idx, uploaded_file in enumerate(files, 1):
+        try:
+            # Update progress
+            progress_placeholder.markdown(
+                f"<div style='text-align: center; margin: 1rem 0;'>"
+                f"Processing file {idx} of {total_files}"
+                f"</div>",
+                unsafe_allow_html=True
+            )
+            
+            status_text.markdown(
+                f"<div class='status-badge status-warning'>"
+                f"📝 Processing: {uploaded_file.name}"
+                f"</div>",
+                unsafe_allow_html=True
+            )
+            
+            if uploaded_file.type in ['image/jpeg', 'image/png']:
+                # Process image
+                image = Image.open(uploaded_file)
+                analysis = process_image(image)
+                
+                # Save results
+                st.session_state.analyses[uploaded_file.name] = {
+                    'type': uploaded_file.type,
+                    'analysis': analysis,
+                    'timestamp': datetime.datetime.now()
+                }
+                
+                # Create chat engine
+                st.session_state.chat_engines[uploaded_file.name] = create_chat_engine(analysis)
+                
+            elif uploaded_file.type == 'application/pdf':
+                # Process PDF
+                documents = process_pdf(uploaded_file)
+                analysis = generate_pdf_analysis(documents)
+                
+                # Save results
+                st.session_state.analyses[uploaded_file.name] = {
+                    'type': uploaded_file.type,
+                    'analysis': analysis,
+                    'timestamp': datetime.datetime.now()
+                }
+                
+                # Create chat engine
+                st.session_state.chat_engines[uploaded_file.name] = create_chat_engine(documents)
+            
+            # Update progress
+            progress_bar.progress(idx/total_files)
+            
+            # Save to history
+            save_to_history(
+                uploaded_file.name,
+                uploaded_file.type.split('/')[1].upper(),
+                analysis,
+                datetime.datetime.now()
+            )
+            
+        except Exception as e:
+            st.error(
+                f"❌ Error processing {uploaded_file.name}\n"
+                f"Details: {str(e)}"
+            )
+    
+    # Clear progress indicators
+    progress_placeholder.empty()
+    progress_bar.empty()
+    
+    # Show completion message
+    status_text.markdown(
+        "<div class='status-badge status-success' style='margin: 1rem 0;'>"
+        "✅ All documents processed successfully!"
+        "</div>",
+        unsafe_allow_html=True
+    )
+    
+    # Mobile-friendly action buttons
+    st.markdown("<div class='touch-spacing'>", unsafe_allow_html=True)
+    if st.button("💬 Start Chat", use_container_width=True):
+        st.switch_page("pages/2_💬_Document_Chat.py")
+    if st.button("📤 Process More", use_container_width=True):
+        st.rerun()
+    st.markdown("</div>", unsafe_allow_html=True)
+
+def display_analysis_results():
+    """Display analysis results with mobile-friendly layout"""
+    if st.session_state.analyses:
+        # Mobile-friendly filters
+        st.markdown(
+            "<div class='card' style='margin-bottom: 1rem;'>"
+            "<h4>Filter Results</h4>",
+            unsafe_allow_html=True
+        )
+        
+        # Touch-friendly filter options
+        filter_type = st.multiselect(
+            "Document Type",
+            ["Image", "PDF", "Captured"],
+            default=["Image", "PDF", "Captured"]
+        )
+        
+        
+        st.markdown("</div>", unsafe_allow_html=True)
+        
+        # Display results
+        if st.session_state.analyses:
+            for filename, data in st.session_state.analyses.items():
+                st.markdown(
+                    f"<div class='card'>"
+                    f"<div style='display: flex; justify-content: space-between; align-items: center;'>"
+                    f"<h4>{filename}</h4>"
+                    f"<div class='status-badge status-success'>{data['type'].split('/')[1].upper()}</div>"
+                    f"</div>"
+                    f"<hr style='margin: 0.5rem 0;'>"
+                    f"{format_analysis_results(data['analysis'])}"  # Use the formatting function
+                    f"<div class='touch-spacing'>"
+                    f"</div>"
+                    f"</div>",
+                    unsafe_allow_html=True
+                )
+    else:
+        st.markdown(
+            "<div class='card' style='text-align: center;'>"
+            "<h3>No Documents Analyzed</h3>"
+            "<p>Upload or capture documents to see analysis results here.</p>"
+            "</div>",
+            unsafe_allow_html=True
+        )
+        
+
+if __name__ == "__main__":
+    document_analysis_page()

pages/2_💬_Document_Chat.py

@@ -0,0 +1,165 @@
+import streamlit as st
+from theme import apply_dark_theme, show_page_header, show_footer
+from utils import initialize_session_state
+
+# Page config
+st.set_page_config(
+    page_title="Document Chat |  CiviDoc AI",
+    page_icon="💬",
+    layout="centered",
+    initial_sidebar_state="collapsed",
+
+)
+
+# Apply dark theme
+st.markdown(apply_dark_theme(), unsafe_allow_html=True)
+
+def document_chat_page():
+    # Initialize states
+    initialize_session_state()
+    
+    # Header
+    st.markdown(show_page_header(
+        "💬 Document Chat",
+        "Get instant answers about your documents"
+    ), unsafe_allow_html=True)
+    
+    if st.session_state.chat_engines:
+        # Document selector - Mobile friendly
+        st.markdown(
+            "<div class='card'>"
+            "<h4>Select Document</h4>",
+            unsafe_allow_html=True
+        )
+        
+        doc_names = list(st.session_state.chat_engines.keys())
+        selected_doc = st.selectbox(
+            "Choose a document to discuss:",
+            doc_names,
+            key="doc_selector",
+            format_func=lambda x: f"📄 {x}"
+        )
+        
+        st.markdown("</div>", unsafe_allow_html=True)
+        
+        if selected_doc:
+            # Document preview
+            with st.expander("📄 Document Content", expanded=False):
+                if selected_doc in st.session_state.analyses:
+                    st.markdown(
+                        f"<div class='card'>{st.session_state.analyses[selected_doc]['analysis']}</div>",
+                        unsafe_allow_html=True
+                    )
+            
+            # Initialize chat history for selected document
+            if 'messages' not in st.session_state:
+                st.session_state.messages = [
+                    {"role": "assistant", "content": f"Hello! I'm here to help you understand {selected_doc}. What would you like to know?"}
+                ]
+            
+            # Display chat messages
+            for message in st.session_state.messages:
+                with st.chat_message(message["role"]):
+                    st.write(message["content"])
+            
+            # Suggested questions
+            if len(st.session_state.messages) <= 1:
+                st.markdown(
+                    "<div class='card'>"
+                    "<h4>Suggested Questions</h4>"
+                    "<div class='touch-spacing'>",
+                    unsafe_allow_html=True
+                )
+                
+                suggestions = [
+                    "What is this document about?",
+                    "What are the main requirements?",
+                    "Are there any deadlines I should know about?",
+                    "Can you explain the technical terms?",
+                    "What actions do I need to take?"
+                ]
+                
+                for suggestion in suggestions:
+                    if st.button(f"🔍 {suggestion}", use_container_width=True):
+                        handle_user_input(suggestion, selected_doc)
+                
+                st.markdown("</div></div>", unsafe_allow_html=True)
+            
+            # Chat input
+            if prompt := st.chat_input("Type your question here..."):
+                handle_user_input(prompt, selected_doc)
+            
+            # Clear chat button
+            if len(st.session_state.messages) > 1:
+                if st.button("🗑️ Clear Chat", use_container_width=True):
+                    st.session_state.messages = [st.session_state.messages[0]]
+                    st.rerun()
+    
+    else:
+        # No documents message
+        st.markdown(
+            "<div class='card' style='text-align: center;'>"
+            "<h3>No Documents Available</h3>"
+            "<p>Please analyze some documents first to start chatting.</p>"
+            "<div style='margin-top: 1rem;'>",
+            unsafe_allow_html=True
+        )
+        
+        if st.button("📝 Go to Document Analysis", use_container_width=True):
+            st.switch_page("pages/1_📝_Document_Analysis.py")
+        
+        st.markdown("</div></div>", unsafe_allow_html=True)
+
+def handle_user_input(prompt, selected_doc):
+    """Handle user input with loading states and error handling"""
+    # Add user message
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    
+    # Display user message
+    with st.chat_message("user"):
+        st.write(prompt)
+    
+    # Generate and display assistant response
+    with st.chat_message("assistant"):
+        with st.spinner("Thinking..."):
+            try:
+                chat_engine = st.session_state.chat_engines[selected_doc]
+                response = chat_engine.chat(prompt)
+                assistant_response = response.response
+                
+                # Add assistant response to messages
+                st.session_state.messages.append({
+                    "role": "assistant",
+                    "content": assistant_response
+                })
+                
+                # Display response
+                st.write(assistant_response)
+                
+                # Show related questions
+                with st.expander("Related Questions", expanded=False):
+                    suggestions = generate_related_questions(prompt, assistant_response)
+                    for suggestion in suggestions:
+                        if st.button(f"🔍 {suggestion}", key=f"related_{suggestion}", use_container_width=True):
+                            handle_user_input(suggestion, selected_doc)
+                
+            except Exception as e:
+                error_message = f"Sorry, I encountered an error: {str(e)}"
+                st.error(error_message)
+                st.session_state.messages.append({
+                    "role": "assistant",
+                    "content": error_message
+                })
+
+def generate_related_questions(prompt, response):
+    """Generate related questions based on the conversation context"""
+    # Add your logic to generate related questions
+    return [
+        "Can you explain this in simpler terms?",
+        "What are the next steps?",
+        "Are there any requirements I should know about?",
+        "Where can I find more information?"
+    ]
+
+if __name__ == "__main__":
+    document_chat_page()

pages/3_✍️_Writing_Assistant.py

@@ -0,0 +1,375 @@
+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__":
+    writing_assistant_page()

pages/4_📚_History.py

@@ -0,0 +1,147 @@
+import streamlit as st
+import pandas as pd
+from datetime import datetime
+from utils import initialize_session_state, get_document_history, delete_from_history, format_timestamp, format_analysis_results
+
+def display_document_content(content):
+    """Display formatted document content"""
+    st.markdown(
+        f"<div class='card'>"
+        f"<h4>Document Analysis</h4>"
+        f"{format_analysis_results(content)}"
+        f"</div>",
+        unsafe_allow_html=True
+    )
+def document_history_page():
+    st.title("📚 Document History")
+    initialize_session_state()
+    
+    # Create DataFrame from document history
+    if st.session_state.document_history:
+        history_data = []
+        for doc_name, details in get_document_history().items():
+            history_data.append({
+                'Document Name': doc_name,
+                'Type': details['type'],
+                'Date': format_timestamp(details['timestamp']),
+                'Status': details['status']
+            })
+        
+        df = pd.DataFrame(history_data)
+        
+        # Filters
+        col1, col2 = st.columns(2)
+        with col1:
+            doc_type_filter = st.multiselect(
+                "Filter by Document Type",
+                options=df['Type'].unique(),
+                default=[]
+            )
+        
+        with col2:
+            date_range = st.date_input(
+                "Filter by Date Range",
+                value=(datetime.now().date(), datetime.now().date()),
+                key="date_range"
+            )
+        
+        # Apply filters
+        filtered_df = df.copy()
+        if doc_type_filter:
+            filtered_df = filtered_df[filtered_df['Type'].isin(doc_type_filter)]
+        if len(date_range) == 2:
+            filtered_df['Date'] = pd.to_datetime(filtered_df['Date'])
+            mask = (filtered_df['Date'].dt.date >= date_range[0]) & (filtered_df['Date'].dt.date <= date_range[1])
+            filtered_df = filtered_df[mask]
+        
+        # Display interactive table
+        st.dataframe(
+            filtered_df,
+            column_config={
+                "Document Name": st.column_config.TextColumn(
+                    "Document Name",
+                    width="medium",
+                ),
+                "Type": st.column_config.TextColumn(
+                    "Type",
+                    width="small",
+                ),
+                "Date": st.column_config.TextColumn(
+                    "Processing Date",
+                    width="small",
+                ),
+                "Status": st.column_config.TextColumn(
+                    "Status",
+                    width="small",
+                ),
+            },
+            hide_index=True,
+        )
+        
+        # Document Details Section
+        st.subheader("Document Details")
+        selected_doc = st.selectbox(
+            "Select a document to view details",
+            options=df['Document Name'].tolist()
+        )
+        
+        if selected_doc:
+            doc_details = st.session_state.document_history[selected_doc]
+            
+            col1, col2, col3 = st.columns([2,2,1])
+            with col1:
+                st.markdown(f"**Type:** {doc_details['type']}")
+                st.markdown(f"**Processed on:** {format_timestamp(doc_details['timestamp'])}")
+            
+            with col2:
+                st.markdown(f"**Status:** {doc_details['status']}")
+            
+            with col3:
+                if st.button("Delete Document", key=f"delete_{selected_doc}"):
+                    delete_from_history(selected_doc)
+                    st.rerun()
+            
+            # Display document content
+            display_document_content(doc_details['content'])
+            #with st.expander("Document Content", expanded=True):
+                #st.markdown(doc_details['content'])
+            
+            # Actions
+            st.subheader("Actions")
+            col1, col2, col3 = st.columns(3)
+            
+            with col1:
+                if st.button("Chat about Document", use_container_width=True):
+                    st.session_state.current_doc = selected_doc
+                    st.switch_page("pages/2_💬_Document_Chat.py")
+            
+            with col2:
+                if st.button("Download Document", use_container_width=True):
+                    # Create downloadable version
+                    content = doc_details['content']
+                    st.download_button(
+                        label="Download",
+                        data=content.encode(),
+                        file_name=f"{selected_doc}.txt",
+                        mime="text/plain"
+                    )
+            
+            with col3:
+                if st.button("Share Document", use_container_width=True):
+                    # Generate shareable link or copy to clipboard
+                    st.info("Document sharing functionality coming soon!")
+    
+    else:
+        st.info("No documents in history. Start by analyzing or creating documents!")
+        
+        col1, col2 = st.columns(2)
+        with col1:
+            if st.button("Analyze Documents", use_container_width=True):
+                st.switch_page("pages/1_📝_Document_Analysis.py")
+        
+        with col2:
+            if st.button("Create New Document", use_container_width=True):
+                st.switch_page("pages/3_✍️_Writing_Assistant.py")
+
+if __name__ == "__main__":
+    document_history_page()

requirements.txt

@@ -0,0 +1,51 @@
+<<<<<<< HEAD
+groq
+python-dotenv
+langchain
+langchain-community
+sentence-transformers
+transformers
+llama-cloud==0.1.4
+llama-index==0.11.22
+llama-index-agent-openai==0.3.4
+llama-index-cli==0.3.1
+llama-index-core==0.11.22
+llama-index-embeddings-langchain==0.2.1
+llama-index-embeddings-openai==0.2.5
+llama-index-indices-managed-llama-cloud==0.4.0
+llama-index-legacy==0.9.48.post4
+llama-index-llms-groq==0.2.0
+llama-index-llms-openai==0.2.16
+llama-index-llms-openai-like==0.2.0
+llama-index-multi-modal-llms-openai==0.2.3
+llama-index-program-openai==0.2.0
+llama-index-question-gen-openai==0.2.0
+llama-index-readers-file==0.2.2
+llama-index-readers-llama-parse==0.3.0
+llama-parse==0.5.13
+=======
+groq
+python-dotenv
+langchain
+langchain-community
+sentence-transformers
+transformers
+llama-cloud==0.1.4
+llama-index==0.11.22
+llama-index-agent-openai==0.3.4
+llama-index-cli==0.3.1
+llama-index-core==0.11.22
+llama-index-embeddings-langchain==0.2.1
+llama-index-embeddings-openai==0.2.5
+llama-index-indices-managed-llama-cloud==0.4.0
+llama-index-legacy==0.9.48.post4
+llama-index-llms-groq==0.2.0
+llama-index-llms-openai==0.2.16
+llama-index-llms-openai-like==0.2.0
+llama-index-multi-modal-llms-openai==0.2.3
+llama-index-program-openai==0.2.0
+llama-index-question-gen-openai==0.2.0
+llama-index-readers-file==0.2.2
+llama-index-readers-llama-parse==0.3.0
+llama-parse==0.5.13
+>>>>>>> 1bc20a0d3edc7f88f03e506f84b01a7303d403b2

theme.py

@@ -0,0 +1,426 @@
+<<<<<<< HEAD
+
+def apply_dark_theme():
+    return (
+        "<style>"
+        # Base styles
+        "body { background-color: #0F172A; color: #E2E8F0; }"
+        ".main { padding: 0rem 1rem; }"
+        
+        # Button styles
+        ".stButton>button {"
+        "    width: 100%;"
+        "    padding: 1rem;"
+        "    font-size: 1.1em;"
+        "    border-radius: 10px;"
+        "    background-color: #1E293B;"
+        "    color: #E2E8F0;"
+        "    border: 1px solid #3B82F6;"
+        "    box-shadow: 0 2px 5px rgba(0,0,0,0.3);"
+        "    transition: all 0.2s ease;"
+        "}"
+        ".stButton>button:hover {"
+        "    transform: translateY(-2px);"
+        "    box-shadow: 0 4px 15px rgba(59, 130, 246, 0.3);"
+        "    border-color: #60A5FA;"
+        "    background-color: #2D3748;"
+        "}"
+        
+        # Card styles
+        ".card {"
+        "    padding: 1.5rem;"
+        "    border-radius: 10px;"
+        "    background-color: #1E293B;"
+        "    margin-bottom: 1rem;"
+        "    border: 1px solid #2D3748;"
+        "    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);"
+        "    transition: all 0.2s ease;"
+        "}"
+        ".card:hover {"
+        "    border-color: #3B82F6;"
+        "    box-shadow: 0 4px 12px rgba(59, 130, 246, 0.2);"
+        "    transform: translateY(-2px);"
+        "}"
+        
+        # Typography
+        "h1, h2, h3, h4, h5, h6 { color: #60A5FA; }"
+        "h1 { font-size: 2.5em; font-weight: 700; }"
+        "h2 { font-size: 2em; font-weight: 600; }"
+        "h3 { font-size: 1.5em; font-weight: 600; }"
+        "p { line-height: 1.6; }"
+        
+        # Form elements
+        ".stTextInput>div>div>input,"
+        ".stTextArea>div>div>textarea,"
+        ".stSelectbox>div>div>select {"
+        "    background-color: #1E293B;"
+        "    color: #E2E8F0;"
+        "    border: 1px solid #3B82F6;"
+        "    border-radius: 8px;"
+        "}"
+        
+        # File uploader
+        ".stFileUploader>div>button {"
+        "    background-color: #1E293B;"
+        "    color: #E2E8F0;"
+        "    border: 1px dashed #3B82F6;"
+        "    border-radius: 8px;"
+        "}"
+        ".stFileUploader>div>button:hover {"
+        "    border-color: #60A5FA;"
+        "    background-color: #2D3748;"
+        "}"
+        
+        # Expander
+        ".streamlit-expanderHeader {"
+        "    background-color: #1E293B;"
+        "    color: #E2E8F0;"
+        "    border-radius: 8px;"
+        "}"
+        ".streamlit-expanderContent {"
+        "    background-color: #1E293B;"
+        "    border: 1px solid #2D3748;"
+        "    border-radius: 0 0 8px 8px;"
+        "}"
+        
+        # Tabs
+        ".stTabs [data-baseweb='tab-list'] {"
+        "    gap: 30px;"
+        "    background-color: #1E293B;"
+        "    padding: 1.5rem;"
+        "    border-radius: 8px;"
+        "}"
+        ".stTabs [data-baseweb='tab'] {"
+        "    height: 40px;"
+        "    padding: 1.5rem;"
+        "    background-color: #2D3748;"
+        "    border-radius: 8px;"
+        "    color: #E2E8F0;"
+        "    border: 1px solid #3B82F6;"
+        "}"
+        ".stTabs [aria-selected='true'] {"
+        "    background-color: #3B82F6;"
+        "}"
+        
+        # Chat elements
+        ".chat-message {"
+        "    padding: 1rem;"
+        "    margin: 0.5rem 0;"
+        "    border-radius: 8px;"
+        "    background-color: #1E293B;"
+        "    border: 1px solid #2D3748;"
+        "}"
+        ".user-message {"
+        "    background-color: #2D3748;"
+        "    border-color: #3B82F6;"
+        "}"
+        
+        # Status indicators
+        ".status-badge {"
+        "    display: inline-block;"
+        "    padding: 0.25rem 0.75rem;"
+        "    border-radius: 9999px;"
+        "    font-size: 0.875rem;"
+        "    font-weight: 500;"
+        "    text-align: center;"
+        "}"
+        ".status-success { background-color: #065F46; color: #6EE7B7; }"
+        ".status-warning { background-color: #92400E; color: #FCD34D; }"
+        ".status-error { background-color: #991B1B; color: #FCA5A5; }"
+        
+        # Tooltips
+        ".tooltip {"
+        "    position: relative;"
+        "    display: inline-block;"
+        "}"
+        ".tooltip .tooltiptext {"
+        "    visibility: hidden;"
+        "    background-color: #2D3748;"
+        "    color: #E2E8F0;"
+        "    text-align: center;"
+        "    padding: 0.5rem;"
+        "    border-radius: 6px;"
+        "    border: 1px solid #3B82F6;"
+        "    position: absolute;"
+        "    z-index: 1;"
+        "    bottom: 125%;"
+        "    left: 50%;"
+        "    transform: translateX(-50%);"
+        "    opacity: 0;"
+        "    transition: opacity 0.2s;"
+        "}"
+        ".tooltip:hover .tooltiptext {"
+        "    visibility: visible;"
+        "    opacity: 1;"
+        "}"
+        
+        # Progress indicators
+        ".progress-bar {"
+        "    width: 100%;"
+        "    height: 8px;"
+        "    background-color: #2D3748;"
+        "    border-radius: 4px;"
+        "    overflow: hidden;"
+        "}"
+        ".progress-bar-fill {"
+        "    height: 100%;"
+        "    background-color: #3B82F6;"
+        "    transition: width 0.3s ease;"
+        "}"
+        
+        # Custom scrollbar
+        "::-webkit-scrollbar { width: 8px; height: 8px; }"
+        "::-webkit-scrollbar-track { background: #1E293B; }"
+        "::-webkit-scrollbar-thumb { background: #3B82F6; border-radius: 4px; }"
+        "::-webkit-scrollbar-thumb:hover { background: #60A5FA; }"
+        
+        # Animations
+        "@keyframes fadeIn { from { opacity: 0; } to { opacity: 1; } }"
+        ".fade-in { animation: fadeIn 0.3s ease-in; }"
+        
+        # Footer
+        ".footer {"
+        "    text-align: center;"
+        "    color: #94A3B8;"
+        "    padding: 2rem 1rem;"
+        "    margin-top: 2rem;"
+        "    border-top: 1px solid #2D3748;"
+        "}"
+        "</style>"
+    )
+
+def show_page_header(title, description=None):
+    header_html = (
+        "<div style='background-color: #1E293B; padding: 2rem; "+
+        "border-radius: 10px; margin-bottom: 2rem;'>"+
+        "<h1 style='color: #60A5FA; margin-bottom: 0.5rem;'>" + title + "</h1>"
+    )
+    if description:
+        header_html += (
+            "<p style='color: #E2E8F0; font-size: 1.1em;'>" + description + "</p>"
+        )
+    header_html += "</div>"
+    return header_html
+
+def show_footer():
+    return (
+        "<div class='footer'>"
+        "    <p>Need help? Contact our support team</p>"
+        "    <p style='font-size: 0.8em; margin-top: 0.5rem;'>"
+        "        CiviDoc AI © 2024"
+        "    </p>"
+        "</div>"
+=======
+# utils/theme.py
+
+def apply_dark_theme():
+    return (
+        "<style>"
+        # Base styles
+        "body { background-color: #0F172A; color: #E2E8F0; }"
+        ".main { padding: 0rem 1rem; }"
+        
+        # Button styles
+        ".stButton>button {"
+        "    width: 100%;"
+        "    padding: 1rem;"
+        "    font-size: 1.1em;"
+        "    border-radius: 10px;"
+        "    background-color: #1E293B;"
+        "    color: #E2E8F0;"
+        "    border: 1px solid #3B82F6;"
+        "    box-shadow: 0 2px 5px rgba(0,0,0,0.3);"
+        "    transition: all 0.2s ease;"
+        "}"
+        ".stButton>button:hover {"
+        "    transform: translateY(-2px);"
+        "    box-shadow: 0 4px 15px rgba(59, 130, 246, 0.3);"
+        "    border-color: #60A5FA;"
+        "    background-color: #2D3748;"
+        "}"
+        
+        # Card styles
+        ".card {"
+        "    padding: 1.5rem;"
+        "    border-radius: 10px;"
+        "    background-color: #1E293B;"
+        "    margin-bottom: 1rem;"
+        "    border: 1px solid #2D3748;"
+        "    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);"
+        "    transition: all 0.2s ease;"
+        "}"
+        ".card:hover {"
+        "    border-color: #3B82F6;"
+        "    box-shadow: 0 4px 12px rgba(59, 130, 246, 0.2);"
+        "    transform: translateY(-2px);"
+        "}"
+        
+        # Typography
+        "h1, h2, h3, h4, h5, h6 { color: #60A5FA; }"
+        "h1 { font-size: 2.5em; font-weight: 700; }"
+        "h2 { font-size: 2em; font-weight: 600; }"
+        "h3 { font-size: 1.5em; font-weight: 600; }"
+        "p { line-height: 1.6; }"
+        
+        # Form elements
+        ".stTextInput>div>div>input,"
+        ".stTextArea>div>div>textarea,"
+        ".stSelectbox>div>div>select {"
+        "    background-color: #1E293B;"
+        "    color: #E2E8F0;"
+        "    border: 1px solid #3B82F6;"
+        "    border-radius: 8px;"
+        "}"
+        
+        # File uploader
+        ".stFileUploader>div>button {"
+        "    background-color: #1E293B;"
+        "    color: #E2E8F0;"
+        "    border: 1px dashed #3B82F6;"
+        "    border-radius: 8px;"
+        "}"
+        ".stFileUploader>div>button:hover {"
+        "    border-color: #60A5FA;"
+        "    background-color: #2D3748;"
+        "}"
+        
+        # Expander
+        ".streamlit-expanderHeader {"
+        "    background-color: #1E293B;"
+        "    color: #E2E8F0;"
+        "    border-radius: 8px;"
+        "}"
+        ".streamlit-expanderContent {"
+        "    background-color: #1E293B;"
+        "    border: 1px solid #2D3748;"
+        "    border-radius: 0 0 8px 8px;"
+        "}"
+        
+        # Tabs
+        ".stTabs [data-baseweb='tab-list'] {"
+        "    gap: 8px;"
+        "    background-color: #1E293B;"
+        "    padding: 0.5rem;"
+        "    border-radius: 8px;"
+        "}"
+        ".stTabs [data-baseweb='tab'] {"
+        "    height: 50px;"
+        "    background-color: #2D3748;"
+        "    border-radius: 8px;"
+        "    color: #E2E8F0;"
+        "    border: 1px solid #3B82F6;"
+        "}"
+        ".stTabs [aria-selected='true'] {"
+        "    background-color: #3B82F6;"
+        "}"
+        
+        # Chat elements
+        ".chat-message {"
+        "    padding: 1rem;"
+        "    margin: 0.5rem 0;"
+        "    border-radius: 8px;"
+        "    background-color: #1E293B;"
+        "    border: 1px solid #2D3748;"
+        "}"
+        ".user-message {"
+        "    background-color: #2D3748;"
+        "    border-color: #3B82F6;"
+        "}"
+        
+        # Status indicators
+        ".status-badge {"
+        "    display: inline-block;"
+        "    padding: 0.25rem 0.75rem;"
+        "    border-radius: 9999px;"
+        "    font-size: 0.875rem;"
+        "    font-weight: 500;"
+        "    text-align: center;"
+        "}"
+        ".status-success { background-color: #065F46; color: #6EE7B7; }"
+        ".status-warning { background-color: #92400E; color: #FCD34D; }"
+        ".status-error { background-color: #991B1B; color: #FCA5A5; }"
+        
+        # Tooltips
+        ".tooltip {"
+        "    position: relative;"
+        "    display: inline-block;"
+        "}"
+        ".tooltip .tooltiptext {"
+        "    visibility: hidden;"
+        "    background-color: #2D3748;"
+        "    color: #E2E8F0;"
+        "    text-align: center;"
+        "    padding: 0.5rem;"
+        "    border-radius: 6px;"
+        "    border: 1px solid #3B82F6;"
+        "    position: absolute;"
+        "    z-index: 1;"
+        "    bottom: 125%;"
+        "    left: 50%;"
+        "    transform: translateX(-50%);"
+        "    opacity: 0;"
+        "    transition: opacity 0.2s;"
+        "}"
+        ".tooltip:hover .tooltiptext {"
+        "    visibility: visible;"
+        "    opacity: 1;"
+        "}"
+        
+        # Progress indicators
+        ".progress-bar {"
+        "    width: 100%;"
+        "    height: 8px;"
+        "    background-color: #2D3748;"
+        "    border-radius: 4px;"
+        "    overflow: hidden;"
+        "}"
+        ".progress-bar-fill {"
+        "    height: 100%;"
+        "    background-color: #3B82F6;"
+        "    transition: width 0.3s ease;"
+        "}"
+        
+        # Custom scrollbar
+        "::-webkit-scrollbar { width: 8px; height: 8px; }"
+        "::-webkit-scrollbar-track { background: #1E293B; }"
+        "::-webkit-scrollbar-thumb { background: #3B82F6; border-radius: 4px; }"
+        "::-webkit-scrollbar-thumb:hover { background: #60A5FA; }"
+        
+        # Animations
+        "@keyframes fadeIn { from { opacity: 0; } to { opacity: 1; } }"
+        ".fade-in { animation: fadeIn 0.3s ease-in; }"
+        
+        # Footer
+        ".footer {"
+        "    text-align: center;"
+        "    color: #94A3B8;"
+        "    padding: 2rem 1rem;"
+        "    margin-top: 2rem;"
+        "    border-top: 1px solid #2D3748;"
+        "}"
+        "</style>"
+    )
+
+def show_page_header(title, description=None):
+    header_html = (
+        "<div style='background-color: #1E293B; padding: 2rem; "+
+        "border-radius: 10px; margin-bottom: 2rem;'>"+
+        "<h1 style='color: #60A5FA; margin-bottom: 0.5rem;'>" + title + "</h1>"
+    )
+    if description:
+        header_html += (
+            "<p style='color: #E2E8F0; font-size: 1.1em;'>" + description + "</p>"
+        )
+    header_html += "</div>"
+    return header_html
+
+def show_footer():
+    return (
+        "<div class='footer'>"
+        "    <p>Need help? Contact our support team</p>"
+        "    <p style='font-size: 0.8em; margin-top: 0.5rem;'>"
+        "        CiviDoc AI © 2024"
+        "    </p>"
+        "</div>"
+>>>>>>> 1bc20a0d3edc7f88f03e506f84b01a7303d403b2
+    )

utils.py

@@ -0,0 +1,763 @@
+<<<<<<< HEAD
+import streamlit as st
+from groq import Groq
+import io
+import base64
+import re
+import os
+from dotenv import load_dotenv
+from llama_index.core import VectorStoreIndex, Settings, Document
+from llama_index.readers.file import PDFReader
+from llama_index.llms.groq import Groq as LlamaGroq
+from llama_index.embeddings.langchain import LangchainEmbedding
+from langchain_community.embeddings import HuggingFaceEmbeddings
+from datetime import datetime
+from PIL import Image
+import gettext
+
+# Load environment variables and configure
+load_dotenv()
+groq_api_key = os.getenv("GROQ_API_KEY")
+client = Groq(api_key=groq_api_key)
+
+# Configure LlamaIndex
+Settings.llm = LlamaGroq(api_key=groq_api_key, model="llama-3.1-70b-versatile")
+lc_embed_model = HuggingFaceEmbeddings(
+    model_name="sentence-transformers/all-mpnet-base-v2"
+)
+Settings.embed_model = LangchainEmbedding(lc_embed_model)
+
+def initialize_session_state():
+    """Initialize all session state variables"""
+    if 'chat_engines' not in st.session_state:
+        st.session_state.chat_engines = {}
+    if 'analyses' not in st.session_state:
+        st.session_state.analyses = {}
+    if 'documents' not in st.session_state:
+        st.session_state.documents = {}
+    if 'current_doc' not in st.session_state:
+        st.session_state.current_doc = None
+    if 'messages' not in st.session_state:
+        st.session_state.messages = []
+    if 'document_history' not in st.session_state:
+        st.session_state.document_history = {}
+
+def encode_image_to_base64(image):
+    """Convert PIL Image to base64 string"""
+    buffered = io.BytesIO()
+    image.save(buffered, format="JPEG")
+    return base64.b64encode(buffered.getvalue()).decode()
+
+def process_image(image):
+    """Process image using Llama vision model"""
+    img_base64 = encode_image_to_base64(image)
+    img_url = f"data:image/jpeg;base64,{img_base64}"
+    
+    completion = client.chat.completions.create(
+        model="llama-3.2-11b-vision-preview",
+        messages=[
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": """Please analyze this government document and provide:
+                        1. Document type and purpose
+                        2. Key requirements and deadlines
+                        3. Complex terms explained simply
+                        4. Required actions or next steps
+                        5. Important contact information or submission details"""
+                    },
+                    {
+                        "type": "image_url",
+                        "image_url": {
+                            "url": img_url
+                        }
+                    }
+                ]
+            }
+        ],
+        temperature=0.1,
+        max_tokens=1024,
+        top_p=1,
+        stream=False
+    )
+    
+    return completion.choices[0].message.content
+
+def generate_pdf_analysis(documents):
+    """Generate analysis from PDF documents using Groq"""
+    try:
+        # Combine all document content
+        full_text = "\n".join([doc.text for doc in documents])
+        
+        # Generate analysis using Groq
+        completion = client.chat.completions.create(
+            model="llama-3.1-70b-versatile",
+            messages=[
+                {
+                    "role": "user",
+                    "content": (
+                        "Please analyze this government document and provide:\n"
+                        "1. Document Type and Purpose:\n"
+                        "   - What kind of document is this?\n"
+                        "   - What is its main purpose?\n\n"
+                        "2. Key Requirements:\n"
+                        "   - What are the main requirements or conditions?\n"
+                        "   - What documents or information are needed?\n\n"
+                        "3. Important Deadlines:\n"
+                        "   - What are the key dates and deadlines?\n"
+                        "   - Are there any time-sensitive requirements?\n\n"
+                        "4. Complex Terms Explained:\n"
+                        "   - Explain any technical or legal terms in simple language\n"
+                        "   - Clarify any complex procedures\n\n"
+                        "5. Required Actions:\n"
+                        "   - What steps need to be taken?\n"
+                        "   - What is the process to follow?\n\n"
+                        "6. Contact Information:\n"
+                        "   - Who to contact for queries?\n"
+                        "   - Where to submit the documents?\n\n"
+                        "Document content:\n" + full_text
+                    )
+                }
+            ],
+            temperature=0.1,
+            max_tokens=2048,
+            top_p=1
+        )
+        
+        # Format the analysis with proper styling
+        analysis = completion.choices[0].message.content
+
+        completionsum = client.chat.completions.create(
+            model="llama-3.1-8b-instant",
+            messages=[
+                {
+                    "role": "user",
+                    "content": (
+                        "Summarize the following content: " + analysis
+                    )
+                }
+            ],
+            temperature=0.1,
+            max_tokens=2048,
+            top_p=1
+        )
+
+        analysissum = completionsum.choices[0].message.content
+        
+        return analysissum
+    except Exception as e:
+        error_msg = "Error generating PDF analysis: " + str(e)
+        raise Exception(error_msg)
+
+def clean_llm_output(output):
+    """Clean LLM output by removing HTML tags and formatting symbols"""
+    # Remove HTML tags
+    cleaned_text = re.sub(r'<[^>]+>', '', output)
+    # Remove double asterisks
+    cleaned_text = cleaned_text.replace('**', '')
+    cleaned_text = cleaned_text.replace('*', '')
+    # Remove extra whitespace
+    cleaned_text = re.sub(r'\s+', ' ', cleaned_text)
+    return cleaned_text.strip()
+
+def format_analysis_results(text):
+    """Format analysis results into structured HTML"""
+    # First clean the text
+    cleaned_text = clean_llm_output(text)
+    
+    # Split into sections
+    sections = []
+    current_section = ""
+    current_title = ""
+    
+    for line in cleaned_text.split('\n'):
+        line = line.strip()
+        if ':' in line and not line.startswith('*'):
+            # If we have a previous section, save it
+            if current_title:
+                sections.append((current_title, current_section.strip()))
+            # Start new section
+            parts = line.split(':', 1)
+            current_title = parts[0].strip()
+            current_section = parts[1].strip() if len(parts) > 1 else ""
+        else:
+            current_section += " " + line
+    
+    # Add the last section
+    if current_title:
+        sections.append((current_title, current_section.strip()))
+    
+    # Generate HTML
+    html = "<div class='analysis-results'>"
+    for title, content in sections:
+        html += f"""
+            <div class='analysis-section card' style='margin-bottom: 1rem;'>
+                <h4 style='color: #60A5FA; margin-bottom: 0.5rem;'>{title}</h4>
+                <p style='margin: 0;'>{content}</p>
+            </div>
+        """
+    html += "</div>"
+    
+    return html
+
+def process_captured_image(picture):
+    """Process image captured from camera with mobile-friendly UI"""
+    try:
+        # Show processing status
+        status_placeholder = st.empty()
+        status_placeholder.markdown(
+            "<div class='status-badge status-warning'>"
+            "📸 Processing captured image..."
+            "</div>",
+            unsafe_allow_html=True
+        )
+        
+        # Process the image
+        image = Image.open(picture)
+        
+        # Display the captured image with proper mobile sizing
+        st.image(
+            image,
+            caption="Captured Document",
+            use_column_width=True  # Makes image responsive
+        )
+        
+        # Process image with AI
+        with st.spinner("Analyzing document..."):
+            analysis = process_image(image)
+        
+        # Generate filename with timestamp
+        timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+        filename = f"captured_image_{timestamp}"
+        
+        # Save results
+        st.session_state.analyses[filename] = {
+            'type': 'image/jpeg',
+            'analysis': analysis,
+            'timestamp': datetime.datetime.now()
+        }
+        
+        # Create chat engine
+        st.session_state.chat_engines[filename] = create_chat_engine(analysis)
+        
+        # Save to history
+        save_to_history(
+            filename,
+            'Captured Image',
+            analysis,
+            datetime.datetime.now()
+        )
+        
+        # Update status to success
+        status_placeholder.markdown(
+            "<div class='status-badge status-success'>"
+            "✅ Image analyzed successfully!"
+            "</div>",
+            unsafe_allow_html=True
+        )
+        
+        # Display analysis results
+        st.markdown(
+            "<div class='card'>"
+            "<h4>Analysis Results</h4>"
+            f"<div style='margin: 1rem 0;'>{analysis}</div>"
+            "</div>",
+            unsafe_allow_html=True
+        )
+        
+        # Mobile-friendly action buttons
+        st.markdown("<div class='touch-spacing'>", unsafe_allow_html=True)
+        
+        col1, col2 = st.columns(2)
+        with col1:
+            if st.button("💬 Start Chat", use_container_width=True):
+                st.session_state.current_doc = filename
+                st.switch_page("pages/2_💬_Document_Chat.py")
+        with col2:
+            if st.button("📸 New Capture", use_container_width=True):
+                st.rerun()
+                
+        st.markdown("</div>", unsafe_allow_html=True)
+        
+    except Exception as e:
+        st.error(
+            "❌ Error processing image\n"
+            f"Details: {str(e)}"
+        )
+
+def process_pdf(pdf_file):
+    """Process PDF document using LlamaIndex"""
+    temp_dir = "temp_docs"
+    os.makedirs(temp_dir, exist_ok=True)
+    temp_path = os.path.join(temp_dir, "temp.pdf")
+    
+    with open(temp_path, "wb") as f:
+        f.write(pdf_file.getvalue())
+    
+    try:
+        reader = PDFReader()
+        documents = reader.load_data(temp_path)
+        return documents
+    finally:
+        if os.path.exists(temp_path):
+            os.remove(temp_path)
+        if os.path.exists(temp_dir) and not os.listdir(temp_dir):
+            os.rmdir(temp_dir)
+
+def create_chat_engine(content):
+    """Create chat engine from document content"""
+    if isinstance(content, str):
+        documents = [Document(text=content)]
+    else:
+        documents = content
+    
+    index = VectorStoreIndex.from_documents(documents)
+    return index.as_chat_engine(chat_mode="condense_question", verbose=True)
+
+def generate_document(doc_type, fields):
+    """Generate government documents based on type and fields"""
+    prompt = f"""Generate a formal {doc_type} with the following details:
+    
+    {fields}
+    
+    Please format this as a proper official document following standard government formatting."""
+    
+    completion = client.chat.completions.create(
+        model="llama-3.1-70b-versatile",
+        messages=[
+            {
+                "role": "user",
+                "content": prompt
+            }
+        ],
+        temperature=0.7,
+        max_tokens=2048,
+        top_p=1
+    )
+    
+    return completion.choices[0].message.content
+
+def save_to_history(doc_name, doc_type, content, timestamp=None):
+    """Save document to history with metadata"""
+    if timestamp is None:
+        timestamp = datetime.now()
+    
+    st.session_state.document_history[doc_name] = {
+        'type': doc_type,
+        'content': content,
+        'timestamp': timestamp,
+        'status': 'Processed'
+    }
+
+def get_document_history():
+    """Retrieve document history sorted by timestamp"""
+    history = st.session_state.document_history
+    return dict(sorted(
+        history.items(),
+        key=lambda x: x[1]['timestamp'],
+        reverse=True
+    ))
+
+def delete_from_history(doc_name):
+    """Delete document from history"""
+    if doc_name in st.session_state.document_history:
+        del st.session_state.document_history[doc_name]
+        if doc_name in st.session_state.chat_engines:
+            del st.session_state.chat_engines[doc_name]
+        if doc_name in st.session_state.analyses:
+            del st.session_state.analyses[doc_name]
+        if st.session_state.current_doc == doc_name:
+            st.session_state.current_doc = None
+
+def format_timestamp(timestamp):
+    """Format timestamp for display"""
+    return timestamp.strftime("%Y-%m-%d %H:%M:%S")
+=======
+# utils.py
+import streamlit as st
+from groq import Groq
+import io
+import base64
+import re
+import os
+from dotenv import load_dotenv
+from llama_index.core import VectorStoreIndex, Settings, Document
+from llama_index.readers.file import PDFReader
+from llama_index.llms.groq import Groq as LlamaGroq
+from llama_index.embeddings.langchain import LangchainEmbedding
+from langchain_community.embeddings import HuggingFaceEmbeddings
+from datetime import datetime
+from PIL import Image
+
+# Load environment variables and configure
+load_dotenv()
+groq_api_key = os.getenv("GROQ_API_KEY")
+client = Groq(api_key=groq_api_key)
+
+# Configure LlamaIndex
+Settings.llm = LlamaGroq(api_key=groq_api_key, model="llama-3.1-70b-versatile")
+lc_embed_model = HuggingFaceEmbeddings(
+    model_name="sentence-transformers/all-mpnet-base-v2"
+)
+Settings.embed_model = LangchainEmbedding(lc_embed_model)
+
+def initialize_session_state():
+    """Initialize all session state variables"""
+    if 'chat_engines' not in st.session_state:
+        st.session_state.chat_engines = {}
+    if 'analyses' not in st.session_state:
+        st.session_state.analyses = {}
+    if 'documents' not in st.session_state:
+        st.session_state.documents = {}
+    if 'current_doc' not in st.session_state:
+        st.session_state.current_doc = None
+    if 'messages' not in st.session_state:
+        st.session_state.messages = []
+    if 'document_history' not in st.session_state:
+        st.session_state.document_history = {}
+
+def encode_image_to_base64(image):
+    """Convert PIL Image to base64 string"""
+    buffered = io.BytesIO()
+    image.save(buffered, format="JPEG")
+    return base64.b64encode(buffered.getvalue()).decode()
+
+def process_image(image):
+    """Process image using Llama vision model"""
+    img_base64 = encode_image_to_base64(image)
+    img_url = f"data:image/jpeg;base64,{img_base64}"
+    
+    completion = client.chat.completions.create(
+        model="llama-3.2-11b-vision-preview",
+        messages=[
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": """Please analyze this government document and provide:
+                        1. Document type and purpose
+                        2. Key requirements and deadlines
+                        3. Complex terms explained simply
+                        4. Required actions or next steps
+                        5. Important contact information or submission details"""
+                    },
+                    {
+                        "type": "image_url",
+                        "image_url": {
+                            "url": img_url
+                        }
+                    }
+                ]
+            }
+        ],
+        temperature=0.1,
+        max_tokens=1024,
+        top_p=1,
+        stream=False
+    )
+    
+    return completion.choices[0].message.content
+
+def generate_pdf_analysis(documents):
+    """Generate analysis from PDF documents using Groq"""
+    try:
+        # Combine all document content
+        full_text = "\n".join([doc.text for doc in documents])
+        
+        # Generate analysis using Groq
+        completion = client.chat.completions.create(
+            model="llama-3.1-70b-versatile",
+            messages=[
+                {
+                    "role": "user",
+                    "content": (
+                        "Please analyze this government document and provide:\n"
+                        "1. Document Type and Purpose:\n"
+                        "   - What kind of document is this?\n"
+                        "   - What is its main purpose?\n\n"
+                        "2. Key Requirements:\n"
+                        "   - What are the main requirements or conditions?\n"
+                        "   - What documents or information are needed?\n\n"
+                        "3. Important Deadlines:\n"
+                        "   - What are the key dates and deadlines?\n"
+                        "   - Are there any time-sensitive requirements?\n\n"
+                        "4. Complex Terms Explained:\n"
+                        "   - Explain any technical or legal terms in simple language\n"
+                        "   - Clarify any complex procedures\n\n"
+                        "5. Required Actions:\n"
+                        "   - What steps need to be taken?\n"
+                        "   - What is the process to follow?\n\n"
+                        "6. Contact Information:\n"
+                        "   - Who to contact for queries?\n"
+                        "   - Where to submit the documents?\n\n"
+                        "Document content:\n" + full_text
+                    )
+                }
+            ],
+            temperature=0.1,
+            max_tokens=2048,
+            top_p=1
+        )
+        
+        # Format the analysis with proper styling
+        analysis = completion.choices[0].message.content
+
+        completionsum = client.chat.completions.create(
+            model="llama-3.1-8b-instant",
+            messages=[
+                {
+                    "role": "user",
+                    "content": (
+                        "Summarize the following content: " + analysis
+                    )
+                }
+            ],
+            temperature=0.1,
+            max_tokens=2048,
+            top_p=1
+        )
+
+        analysis = completionsum.choices[0].message.content
+        
+        # Add formatting for better readability
+        formatted_analysis = (
+            "<div class='analysis-container'>"
+            "<div class='analysis-section'>" +
+            analysis.replace('\n\n', '</div><div class="analysis-section">') +
+            "</div>"
+            "</div>"
+        )
+        
+        return formatted_analysis
+        
+    except Exception as e:
+        error_msg = "Error generating PDF analysis: " + str(e)
+        raise Exception(error_msg)
+
+def clean_llm_output(output):
+    """Clean LLM output by removing HTML tags and formatting symbols"""
+    # Remove HTML tags
+    cleaned_text = re.sub(r'<[^>]+>', '', output)
+    # Remove double asterisks
+    cleaned_text = cleaned_text.replace('**', '')
+    cleaned_text = cleaned_text.replace('*', '')
+    # Remove extra whitespace
+    cleaned_text = re.sub(r'\s+', ' ', cleaned_text)
+    return cleaned_text.strip()
+
+def format_analysis_results(text):
+    """Format analysis results into structured HTML"""
+    # First clean the text
+    cleaned_text = clean_llm_output(text)
+    
+    # Split into sections
+    sections = []
+    current_section = ""
+    current_title = ""
+    
+    for line in cleaned_text.split('\n'):
+        line = line.strip()
+        if ':' in line and not line.startswith('*'):
+            # If we have a previous section, save it
+            if current_title:
+                sections.append((current_title, current_section.strip()))
+            # Start new section
+            parts = line.split(':', 1)
+            current_title = parts[0].strip()
+            current_section = parts[1].strip() if len(parts) > 1 else ""
+        else:
+            current_section += " " + line
+    
+    # Add the last section
+    if current_title:
+        sections.append((current_title, current_section.strip()))
+    
+    # Generate HTML
+    html = "<div class='analysis-results'>"
+    for title, content in sections:
+        html += f"""
+            <div class='analysis-section card' style='margin-bottom: 1rem;'>
+                <h4 style='color: #60A5FA; margin-bottom: 0.5rem;'>{title}</h4>
+                <p style='margin: 0;'>{content}</p>
+            </div>
+        """
+    html += "</div>"
+    
+    return html
+
+def process_captured_image(picture):
+    """Process image captured from camera with mobile-friendly UI"""
+    try:
+        # Show processing status
+        status_placeholder = st.empty()
+        status_placeholder.markdown(
+            "<div class='status-badge status-warning'>"
+            "📸 Processing captured image..."
+            "</div>",
+            unsafe_allow_html=True
+        )
+        
+        # Process the image
+        image = Image.open(picture)
+        
+        # Display the captured image with proper mobile sizing
+        st.image(
+            image,
+            caption="Captured Document",
+            use_column_width=True  # Makes image responsive
+        )
+        
+        # Process image with AI
+        with st.spinner("Analyzing document..."):
+            analysis = process_image(image)
+        
+        # Generate filename with timestamp
+        timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+        filename = f"captured_image_{timestamp}"
+        
+        # Save results
+        st.session_state.analyses[filename] = {
+            'type': 'image/jpeg',
+            'analysis': analysis,
+            'timestamp': datetime.datetime.now()
+        }
+        
+        # Create chat engine
+        st.session_state.chat_engines[filename] = create_chat_engine(analysis)
+        
+        # Save to history
+        save_to_history(
+            filename,
+            'Captured Image',
+            analysis,
+            datetime.datetime.now()
+        )
+        
+        # Update status to success
+        status_placeholder.markdown(
+            "<div class='status-badge status-success'>"
+            "✅ Image analyzed successfully!"
+            "</div>",
+            unsafe_allow_html=True
+        )
+        
+        # Display analysis results
+        st.markdown(
+            "<div class='card'>"
+            "<h4>Analysis Results</h4>"
+            f"<div style='margin: 1rem 0;'>{analysis}</div>"
+            "</div>",
+            unsafe_allow_html=True
+        )
+        
+        # Mobile-friendly action buttons
+        st.markdown("<div class='touch-spacing'>", unsafe_allow_html=True)
+        
+        col1, col2 = st.columns(2)
+        with col1:
+            if st.button("💬 Start Chat", use_container_width=True):
+                st.session_state.current_doc = filename
+                st.switch_page("pages/Document_Chat.py")
+        with col2:
+            if st.button("📸 New Capture", use_container_width=True):
+                st.rerun()
+                
+        st.markdown("</div>", unsafe_allow_html=True)
+        
+    except Exception as e:
+        st.error(
+            "❌ Error processing image\n"
+            f"Details: {str(e)}"
+        )
+
+def process_pdf(pdf_file):
+    """Process PDF document using LlamaIndex"""
+    temp_dir = "temp_docs"
+    os.makedirs(temp_dir, exist_ok=True)
+    temp_path = os.path.join(temp_dir, "temp.pdf")
+    
+    with open(temp_path, "wb") as f:
+        f.write(pdf_file.getvalue())
+    
+    try:
+        reader = PDFReader()
+        documents = reader.load_data(temp_path)
+        return documents
+    finally:
+        if os.path.exists(temp_path):
+            os.remove(temp_path)
+        if os.path.exists(temp_dir) and not os.listdir(temp_dir):
+            os.rmdir(temp_dir)
+
+def create_chat_engine(content):
+    """Create chat engine from document content"""
+    if isinstance(content, str):
+        documents = [Document(text=content)]
+    else:
+        documents = content
+    
+    index = VectorStoreIndex.from_documents(documents)
+    return index.as_chat_engine(chat_mode="condense_question", verbose=True)
+
+def generate_document(doc_type, fields):
+    """Generate government documents based on type and fields"""
+    prompt = f"""Generate a formal {doc_type} with the following details:
+    
+    {fields}
+    
+    Please format this as a proper official document following standard government formatting."""
+    
+    completion = client.chat.completions.create(
+        model="llama-3.1-70b-versatile",
+        messages=[
+            {
+                "role": "user",
+                "content": prompt
+            }
+        ],
+        temperature=0.7,
+        max_tokens=2048,
+        top_p=1
+    )
+    
+    return completion.choices[0].message.content
+
+def save_to_history(doc_name, doc_type, content, timestamp=None):
+    """Save document to history with metadata"""
+    if timestamp is None:
+        timestamp = datetime.now()
+    
+    st.session_state.document_history[doc_name] = {
+        'type': doc_type,
+        'content': content,
+        'timestamp': timestamp,
+        'status': 'Processed'
+    }
+
+def get_document_history():
+    """Retrieve document history sorted by timestamp"""
+    history = st.session_state.document_history
+    return dict(sorted(
+        history.items(),
+        key=lambda x: x[1]['timestamp'],
+        reverse=True
+    ))
+
+def delete_from_history(doc_name):
+    """Delete document from history"""
+    if doc_name in st.session_state.document_history:
+        del st.session_state.document_history[doc_name]
+        if doc_name in st.session_state.chat_engines:
+            del st.session_state.chat_engines[doc_name]
+        if doc_name in st.session_state.analyses:
+            del st.session_state.analyses[doc_name]
+        if st.session_state.current_doc == doc_name:
+            st.session_state.current_doc = None
+
+def format_timestamp(timestamp):
+    """Format timestamp for display"""
+    return timestamp.strftime("%Y-%m-%d %H:%M:%S")
+>>>>>>> 1bc20a0d3edc7f88f03e506f84b01a7303d403b2

🏛️_CiviDoc_AI.py

@@ -0,0 +1,202 @@
+import streamlit as st
+from theme import apply_dark_theme, show_page_header, show_footer
+
+# Page config
+st.set_page_config(
+    page_title="CiviDoc AI",
+    page_icon="🏛️",
+    layout="centered",
+    initial_sidebar_state="collapsed"
+)
+
+# Apply dark theme
+st.markdown(apply_dark_theme(), unsafe_allow_html=True)
+
+def main():
+    # Header
+    st.markdown(show_page_header(
+        "<div style='text-align: center;'>"
+        "🏛️ CiviDoc AI",
+        "<div style='text-align: center;'>"
+        "Your AI-powered companion for all government document needs",
+    ), unsafe_allow_html=True)
+    
+    # Quick Access Section - Mobile Friendly Cards
+    st.markdown(
+        "<div class='grid'>"  # Uses the responsive grid system
+        
+        "<div class='card' onclick='void(0)'>"  # Added onclick for better touch feedback
+        "<div style='text-align: center;'>"
+        "<h3 style='margin-bottom: 0.5rem;'>📝 Document Analysis</h3>"
+        "<p style='margin-bottom: 1rem;'>Upload and understand government documents instantly</p>"
+        "<div class='status-badge status-success'>Ready to Use</div>"
+        "</div>"
+        "</div>"
+        
+        "<div class='card' onclick='void(0)'>"
+        "<div style='text-align: center;'>"
+        "<h3 style='margin-bottom: 0.5rem;'>✍️ Writing Assistant</h3>"
+        "<p style='margin-bottom: 1rem;'>Create professional government documents effortlessly</p>"
+        "<div class='status-badge status-success'>Ready to Use</div>"
+        "</div>"
+        "</div>"
+        
+        "<div class='card' onclick='void(0)'>"
+        "<div style='text-align: center;'>"
+        "<h3 style='margin-bottom: 0.5rem;'>💬 Document Chat</h3>"
+        "<p style='margin-bottom: 1rem;'>Get instant answers about your documents</p>"
+        "<div class='status-badge status-success'>Ready to Use</div>"
+        "</div>"
+        "</div>"
+        
+        "</div>",
+        unsafe_allow_html=True
+    )
+    
+    # Quick Action Buttons - Touch Friendly
+    st.markdown("<div class='touch-spacing'>", unsafe_allow_html=True)
+    if st.button("📝 Start Document Analysis", use_container_width=True):
+        st.switch_page("pages/1_📝_Document_Analysis.py")
+    if st.button("✍️ Create New Document", use_container_width=True):
+        st.switch_page("pages/3_✍️_Writing_Assistant.py")
+    if st.button("💬 Open Document Chat", use_container_width=True):
+        st.switch_page("pages/2_💬_Document_Chat.py")
+    st.markdown("</div>", unsafe_allow_html=True)
+    
+    # Features Section - Responsive Grid
+    st.markdown("<h3 style='margin: 1.5rem 0 1rem;'>🌟 Key Features</h3>", unsafe_allow_html=True)
+    
+    # Features grid with improved mobile layout
+    st.markdown(
+        "<div class='grid'>"
+        
+        # Document Analysis Card
+        "<div class='card'>"
+        "<h3 style='margin-bottom: 0.75rem;'>📑 Document Analysis</h3>"
+        "<div class='touch-spacing'>"
+        "<p>✓ Instant document understanding</p>"
+        "<p>✓ Complex term explanations</p>"
+        "<p>✓ Form filling guidance</p>"
+        "<p>✓ Requirement extraction</p>"
+        "<p>✓ Deadline tracking</p>"
+        "</div>"
+        "</div>"
+        
+        # Writing Assistant Card
+        "<div class='card'>"
+        "<h3 style='margin-bottom: 0.75rem;'>✍️ Writing Assistant</h3>"
+        "<div class='touch-spacing'>"
+        "<p>✓ RTI application generator</p>"
+        "<p>✓ Complaint letter creator</p>"
+        "<p>✓ Legal notice drafting</p>"
+        "<p>✓ Appeal letter formatting</p>"
+        "<p>✓ Custom document templates</p>"
+        "</div>"
+        "</div>"
+        
+        # Interactive Help Card
+        "<div class='card'>"
+        "<h3 style='margin-bottom: 0.75rem;'>💬 Interactive Help</h3>"
+        "<div class='touch-spacing'>"
+        "<p>✓ Real-time document chat</p>"
+        "<p>✓ Context-aware responses</p>"
+        "<p>✓ Procedure explanations</p>"
+        "<p>✓ Multi-document support</p>"
+        "<p>✓ Instant clarifications</p>"
+        "</div>"
+        "</div>"
+        
+        # Document Management Card
+        "<div class='card'>"
+        "<h3 style='margin-bottom: 0.75rem;'>📚 Document Management</h3>"
+        "<div class='touch-spacing'>"
+        "<p>✓ Secure document storage</p>"
+        "<p>✓ Version tracking</p>"
+        "<p>✓ Easy organization</p>"
+        "<p>✓ Quick retrieval</p>"
+        "<p>✓ Status monitoring</p>"
+        "</div>"
+        "</div>"
+        
+        "</div>",
+        unsafe_allow_html=True
+    )
+    
+    # How It Works Section - Mobile Friendly Steps
+    st.markdown("<h3 style='margin: 1.5rem 0 1rem;'>🔄 How It Works</h3>", unsafe_allow_html=True)
+    
+    st.markdown(
+        "<div class='grid'>"  # Responsive grid
+        
+        # Step 1
+        "<div class='card' style='text-align: center;'>"
+        "<h4 style='margin-bottom: 0.5rem;'>1. Upload</h4>"
+        "<p style='margin-bottom: 0.75rem;'>Upload your government documents or start creating new ones</p>"
+        "<div class='progress-bar'><div class='progress-bar-fill' style='width: 25%;'></div></div>"
+        "</div>"
+        
+        # Step 2
+        "<div class='card' style='text-align: center;'>"
+        "<h4 style='margin-bottom: 0.5rem;'>2. Process</h4>"
+        "<p style='margin-bottom: 0.75rem;'>Our AI analyzes and processes your documents instantly</p>"
+        "<div class='progress-bar'><div class='progress-bar-fill' style='width: 50%;'></div></div>"
+        "</div>"
+        
+        # Step 3
+        "<div class='card' style='text-align: center;'>"
+        "<h4 style='margin-bottom: 0.5rem;'>3. Understand</h4>"
+        "<p style='margin-bottom: 0.75rem;'>Get clear explanations and guidance for your documents</p>"
+        "<div class='progress-bar'><div class='progress-bar-fill' style='width: 75%;'></div></div>"
+        "</div>"
+        
+        # Step 4
+        "<div class='card' style='text-align: center;'>"
+        "<h4 style='margin-bottom: 0.5rem;'>4. Act</h4>"
+        "<p style='margin-bottom: 0.75rem;'>Take action with confidence using our recommendations</p>"
+        "<div class='progress-bar'><div class='progress-bar-fill' style='width: 100%;'></div></div>"
+        "</div>"
+        
+        "</div>",
+        unsafe_allow_html=True
+    )
+    
+    # Additional Information - Mobile Friendly Layout
+    st.markdown("<div class='grid'>", unsafe_allow_html=True)
+    
+    # Who Is This For Section
+    st.markdown(
+        "<div class='card'>"
+        "<h3 style='margin-bottom: 0.75rem;'>🎯 Who Is This For?</h3>"
+        "<div class='touch-spacing'>"
+        "<p>• Citizens dealing with government procedures</p>"
+        "<p>• RTI applicants and activists</p>"
+        "<p>• Legal professionals</p>"
+        "<p>• Government service seekers</p>"
+        "<p>• Anyone needing document assistance</p>"
+        "</div>"
+        "</div>",
+        unsafe_allow_html=True
+    )
+    
+    # Security & Privacy Section
+    st.markdown(
+        "<div class='card'>"
+        "<h3 style='margin-bottom: 0.75rem;'>🛡️ Security & Privacy</h3>"
+        "<div class='touch-spacing'>"
+        "<p>• End-to-end encryption</p>"
+        "<p>• Secure document processing</p>"
+        "<p>• No permanent storage</p>"
+        "<p>• Privacy-first approach</p>"
+        "<p>• Regular security updates</p>"
+        "</div>"
+        "</div>",
+        unsafe_allow_html=True
+    )
+    
+    st.markdown("</div>", unsafe_allow_html=True)
+    
+    # Footer
+    st.markdown(show_footer(), unsafe_allow_html=True)
+
+if __name__ == "__main__":
+    main()