app.py

import os
import base64
import io
import dash
from dash import dcc, html, Input, Output, State, callback_context, MATCH, ALL
import dash_bootstrap_components as dbc
import pandas as pd
import logging
from docx import Document
import mimetypes
from threading import Thread

import google.generativeai as genai

logging.basicConfig(
    level=logging.INFO,
    format='[%(asctime)s] %(levelname)s - %(message)s'
)

app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
server = app.server

GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY", "")
genai.configure(api_key=GOOGLE_API_KEY)
GEMINI_MODEL = "models/gemini-2.5-pro-preview-03-25"

MAX_INPUT_TOKENS = 1048576
MAX_OUTPUT_TOKENS = 65536

uploaded_documents = {}
uploaded_documents_fileid = {}
uploaded_proposals = {}
uploaded_proposals_fileid = {}
generated_documents = {}
shredded_documents = {}
shredded_document = None
generated_response = None

def decode_document(decoded_bytes):
    try:
        content = decoded_bytes.decode('utf-8')
        logging.info("Document decoded as UTF-8.")
        return content
    except UnicodeDecodeError as e_utf8:
        try:
            content = decoded_bytes.decode('latin-1')
            logging.warning("Document decoded as Latin-1 due to utf-8 decode error: %s", e_utf8)
            return content
        except Exception as e:
            logging.error("Document decode failed for both utf-8 and latin-1: %s", e)
            return None

def guess_mime_type(filename):
    mime_type, _ = mimetypes.guess_type(filename)
    if not mime_type:
        ext = filename.lower().split('.')[-1]
        if ext == "pdf":
            mime_type = "application/pdf"
        elif ext == "docx":
            mime_type = "application/vnd.openxmlformats-officedocument.wordprocessingml.document"
        elif ext == "doc":
            mime_type = "application/msword"
        elif ext == "xlsx":
            mime_type = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
        elif ext == "xls":
            mime_type = "application/vnd.ms-excel"
        else:
            mime_type = "application/octet-stream"
    return mime_type

def upload_to_gemini_file(decoded_bytes, filename):
    try:
        mime_type = guess_mime_type(filename)
        file_obj = io.BytesIO(decoded_bytes)
        file_obj.name = filename
        myfile = genai.upload_file(file_obj, display_name=filename, mime_type=mime_type)
        if hasattr(myfile, "name"):
            logging.info(f"File uploaded to Gemini: {filename}, file_id: {myfile.name}")
            return myfile.name
        else:
            logging.error(f"Gemini file upload did not return file name/id. Response: {myfile}")
            return None
    except Exception as e:
        logging.error(f"Exception during file upload to Gemini: {e}")
        return None

def gemini_generate_content(prompt, file_id=None, chat_input=None):
    try:
        files = []
        if file_id:
            try:
                gemini_file = genai.get_file(file_id)
                files.append(gemini_file)
            except Exception as e:
                logging.error(f"Could not fetch Gemini file for id {file_id}: {e}")
        content_list = []
        if files:
            content_list.extend(files)
            content_list.append("\n\n")
        content_list.append(prompt)
        model = genai.GenerativeModel(GEMINI_MODEL)
        response = model.generate_content(
            contents=content_list,
            generation_config=genai.types.GenerationConfig(
                max_output_tokens=MAX_OUTPUT_TOKENS
            )
        )
        result = response.text if hasattr(response, "text") else str(response)
        logging.info("Received response from Gemini.")
        return result
    except Exception as e:
        logging.error("Error during Gemini generate_content: %s", e)
        return f"Error during Gemini completion: {e}"

def save_shredded_as_docx(shredded_text, rfp_filename):
    doc = Document()
    doc.add_heading(f"Shredded Requirements for {rfp_filename}", 0)
    for line in shredded_text.split('\n'):
        doc.add_paragraph(line)
    memf = io.BytesIO()
    doc.save(memf)
    memf.seek(0)
    return memf.read()

def save_proposal_as_docx(proposal_text, base_filename):
    doc = Document()
    doc.add_heading(f"Proposal Response for {base_filename}", 0)
    for line in proposal_text.split('\n'):
        doc.add_paragraph(line)
    memf = io.BytesIO()
    doc.save(memf)
    memf.seek(0)
    return memf.read()

def process_document(action, selected_filename=None, chat_input=None):
    global shredded_document, generated_response
    logging.info(f"Process document called with action: {action}")

    doc_content = None
    doc_fileid = None
    if action in ["shred", "proposal"]:
        if selected_filename and selected_filename in uploaded_documents:
            doc_content = uploaded_documents[selected_filename]
            doc_fileid = uploaded_documents_fileid.get(selected_filename)
        elif uploaded_documents:
            doc_content = next(iter(uploaded_documents.values()))
            selected_filename = next(iter(uploaded_documents.keys()))
            doc_fileid = uploaded_documents_fileid.get(selected_filename)
        else:
            doc_content = None
            doc_fileid = None

    if action == 'shred':
        if not doc_content:
            logging.warning("No uploaded document found for shredding.")
            return "No document uploaded.", None, None, None
        prompt = (
            "Analyze the following RFP/PWS/SOW/RFI and generate a requirements spreadsheet. Return ONLY the markdown spreadsheet and NOTHING else"
            "Identify requirements by action words like 'shall', 'will', 'perform', etc. Also analye the document and put a column for recommended win theme for each requirement identified. Organize by PWS section and requirement. "
            "Do not write as if responding to the proposal Return the ONLY markdown spreadsheet and NOTHING else, no intro, no outro, no comments.\n"
        )
        if chat_input:
            prompt += f"User additional instructions: {chat_input}\n"
        prompt += f"\nFile Name: {selected_filename}\n\n"
        result_holder = {"text": None, "docx_name": None}
        def thread_shred():
            global shredded_document
            shredded_document = ""
            try:
                logging.info("Starting Gemini generate_content for shredding.")
                result = gemini_generate_content(prompt, file_id=doc_fileid, chat_input=chat_input)
                shredded_document = result
                logging.info("Document shredded successfully.")
                docx_bytes = save_shredded_as_docx(result, selected_filename)
                generated_docx_name = f"{os.path.splitext(selected_filename)[0]}_shredded.docx"
                shredded_documents[generated_docx_name] = docx_bytes
                result_holder["text"] = result
                result_holder["docx_name"] = generated_docx_name
            except Exception as e:
                shredded_document = f"Error during shredding: {e}"
                logging.error("Error in thread_shred: %s", e)
                result_holder["text"] = shredded_document
        shredded_document = "Shredding in progress..."
        t = Thread(target=thread_shred)
        t.start()
        t.join()
        if result_holder["docx_name"] and result_holder["text"]:
            if result_holder["docx_name"] not in uploaded_documents:
                uploaded_documents[result_holder["docx_name"]] = result_holder["text"]
        return result_holder["text"], None, result_holder["docx_name"], result_holder["text"]

    elif action == 'proposal':
        if not doc_content:
            logging.warning("No RFP/SOW/PWS/RFI document selected for proposal action.")
            return "No RFP/SOW/PWS/RFI document selected.", None, None, None
        rfp_filename = selected_filename
        rfp_fileid = uploaded_documents_fileid.get(selected_filename)
        prompt = (
            "Respond to the following RFP/SOW/PWS/RFI by creating a highly detailed proposal response that follows each section and subsection header. "
            "The response to each section and subsection will be compliant and compelling, focusing on describing the approach, and how the labor category uses a specific industry standard process in a workflow described in steps, and how technology is used. "
            "You must show innovation in the approach. Refer to research that validates the approach and cite sources with measurable outcome. "
            "Be sure to respond in paragraph format only, never use numbering, or bullets, only paragraph describing in detail.Return only the proposal response in markdown and nothing else, no comments, no intro, no outro, just the proposal\n"
        )
        if chat_input:
            prompt += f"User additional instructions: {chat_input}\n"
        prompt += f"\n---\nRFP/SOW/PWS/RFI ({rfp_filename}):\n{doc_content}\n"
        logging.info(f"Sending proposal prompt to Gemini. RFP: {rfp_filename}")
        result_holder = {"text": None, "docx_name": None}
        def thread_proposal():
            try:
                logging.info("Connecting to Gemini for proposal.")
                response = gemini_generate_content(prompt, file_id=rfp_fileid, chat_input=chat_input)
                logging.info("Received proposal results from Gemini.")
                docx_bytes = save_proposal_as_docx(response, rfp_filename)
                generated_docx_name = f"{os.path.splitext(rfp_filename)[0]}_proposal.docx"
                generated_documents[generated_docx_name] = docx_bytes
                result_holder["text"] = response
                result_holder["docx_name"] = generated_docx_name
            except Exception as e:
                logging.error("Error during Gemini proposal request: %s", e)
                result_holder["text"] = f"Error during Gemini completion: {e}"
        t = Thread(target=thread_proposal)
        t.start()
        t.join()
        return result_holder["text"], None, result_holder["docx_name"], result_holder["text"]

    elif action == 'compliance':
        return "Compliance checking not implemented yet.", None, None, None
    elif action == 'recover':
        return "Recovery not implemented yet.", None, None, None
    elif action == 'board':
        return "Virtual board not implemented yet.", None, None, None
    elif action == 'loe':
        return "LOE estimation not implemented yet.", None, None, None
    return "Action not implemented yet.", None, None, None

def get_uploaded_doc_list(docdict):
    if not docdict:
        return html.Div("No documents uploaded.", style={"wordWrap": "break-word"})
    doc_list = []
    for filename in docdict:
        doc_list.append(
            dbc.ListGroupItem([
                html.Span(filename, style={"wordWrap": "break-word"}),
                dbc.Button("Delete", id={'type': 'delete-doc-btn', 'index': filename, 'group': 'rfp'}, size="sm", color="danger", className="float-end ms-2")
            ], className="d-flex justify-content-between align-items-center")
        )
    return dbc.ListGroup(doc_list, flush=True)

def get_shredded_doc_list(shreddict):
    if not shreddict:
        return html.Div("No shredded requirements yet.", style={"wordWrap": "break-word"})
    doc_list = []
    for filename in shreddict:
        b64 = base64.b64encode(shreddict[filename]).decode('utf-8')
        download_link = html.A(
            f"Download {filename}",
            href=f"data:application/vnd.openxmlformats-officedocument.wordprocessingml.document;base64,{b64}",
            download=filename,
            target="_blank",
            style={"wordWrap": "break-word", "marginRight": "10px"}
        )
        doc_list.append(
            dbc.ListGroupItem([
                html.Span(filename, style={"wordWrap": "break-word", "marginRight": "10px"}),
                download_link,
                dbc.Button("Delete", id={'type': 'delete-shredded-btn', 'index': filename, 'group': 'shredded'}, size="sm", color="danger", className="float-end ms-2")
            ], className="d-flex justify-content-between align-items-center")
        )
    return dbc.ListGroup(doc_list, flush=True)

def get_uploaded_proposal_list(docdict):
    if not docdict:
        return html.Div("No proposal documents uploaded.", style={"wordWrap": "break-word"})
    doc_list = []
    for filename in docdict:
        doc_list.append(
            dbc.ListGroupItem([
                html.Span(filename, style={"wordWrap": "break-word"}),
                dbc.Button("Delete", id={'type': 'delete-proposal-btn', 'index': filename, 'group': 'proposal'}, size="sm", color="danger", className="float-end ms-2")
            ], className="d-flex justify-content-between align-items-center")
        )
    return dbc.ListGroup(doc_list, flush=True)

def get_generated_doc_list(docdict):
    if not docdict:
        return html.Div("No generated documents yet.", style={"wordWrap": "break-word"})
    doc_list = []
    for filename in docdict:
        doc_list.append(
            dbc.ListGroupItem([
                html.Span(filename, style={"wordWrap": "break-word"}),
                dbc.Button("Delete", id={'type': 'delete-generated-btn', 'index': filename, 'group': 'generated'}, size="sm", color="danger", className="float-end ms-2")
            ], className="d-flex justify-content-between align-items-center")
        )
    return dbc.ListGroup(doc_list, flush=True)

def get_generated_rfp_download_section(selected_value):
    if not selected_value or selected_value not in generated_documents:
        return html.Div("No generated document selected.", style={"wordWrap": "break-word"})
    docx_bytes = generated_documents[selected_value]
    b64 = base64.b64encode(docx_bytes).decode('utf-8')
    download_link = html.A(
        f"Download {selected_value}",
        href=f"data:application/vnd.openxmlformats-officedocument.wordprocessingml.document;base64,{b64}",
        download=selected_value,
        target="_blank",
        style={"wordWrap": "break-word"}
    )
    return html.Div([
        html.H6("RFP/SOW/PWS/RFI", style={"marginTop": "10px"}),
        html.Div(download_link, style={"marginBottom": "15px"}),
        html.Div("Preview not available for docx. Download to view.", style={"wordWrap": "break-word"})
    ])

app.layout = dbc.Container([
    dcc.Store(id='shred-store', data={'text': None, 'docx_name': None}),
    dcc.Store(id='proposal-store', data={'text': None, 'docx_name': None}),
    dbc.Row([
        dbc.Col([
            dbc.Card([
                dbc.CardHeader(html.H5("RFP/SOW/PWS/RFI")),
                dbc.CardBody([
                    html.H6("Uploaded Documents"),
                    html.Div(get_uploaded_doc_list(uploaded_documents), id='uploaded-doc-list'),
                    dcc.Dropdown(
                        id='select-document-dropdown',
                        options=[{'label': fn, 'value': fn} for fn in uploaded_documents.keys()],
                        placeholder="Select a document to work with",
                        value=next(iter(uploaded_documents), None),
                        style={"marginBottom": "10px"}
                    ),
                    dcc.Upload(
                        id='upload-document',
                        children=html.Div([
                            'Drag and Drop or ',
                            html.A('Select Files')
                        ]),
                        style={
                            'width': '100%',
                            'height': '60px',
                            'lineHeight': '60px',
                            'borderWidth': '1px',
                            'borderStyle': 'dashed',
                            'borderRadius': '5px',
                            'textAlign': 'center',
                            'margin': '10px'
                        },
                        multiple=False
                    ),
                    html.Hr(style={"marginTop": "20px", "marginBottom": "10px"}),
                    html.H6("Shredded Requirements"),
                    html.Div(get_shredded_doc_list(shredded_documents), id='shredded-doc-list')
                ])
            ], className="mb-3"),

            dbc.Card([
                dbc.CardHeader(html.H5("Proposal")),
                dbc.CardBody([
                    html.H6("Uploaded Documents"),
                    html.Div(get_uploaded_proposal_list(uploaded_proposals), id='uploaded-proposal-list'),
                    dcc.Dropdown(
                        id='select-proposal-dropdown',
                        options=[{'label': fn, 'value': fn} for fn in uploaded_proposals.keys()],
                        placeholder="Select a proposal document",
                        value=next(iter(uploaded_proposals), None),
                        style={"marginBottom": "10px"}
                    ),
                    dcc.Upload(
                        id='upload-proposal',
                        children=html.Div([
                            'Drag and Drop or ',
                            html.A('Select Files')
                        ]),
                        style={
                            'width': '100%',
                            'height': '60px',
                            'lineHeight': '60px',
                            'borderWidth': '1px',
                            'borderStyle': 'dashed',
                            'borderRadius': '5px',
                            'textAlign': 'center',
                            'margin': '10px'
                        },
                        multiple=False
                    )
                ])
            ], className="mb-3"),

            dbc.Card([
                dbc.CardHeader(html.H5("Generated Documents")),
                dbc.CardBody([
                    dcc.Dropdown(
                        id='select-generated-dropdown',
                        options=[{'label': fn, 'value': fn} for fn in generated_documents.keys()],
                        placeholder="Select a generated document",
                        value=next(iter(generated_documents), None),
                        style={"marginBottom": "10px"}
                    ),
                    html.Div(id='generated-rfp-section'),
                    html.Hr(style={"marginTop": "20px", "marginBottom": "10px"}),
                    html.Div(get_generated_doc_list(generated_documents), id='generated-doc-list'),
                ])
            ], className="mb-3")
        ], style={'minWidth': '260px', 'width':'30vw','maxWidth':'30vw'}, width=3),

        dbc.Col([
            dbc.Card([
                dbc.CardHeader(html.H2("RFP Proposal Assistant", style={'wordWrap': 'break-word'})),
                dbc.CardBody([
                    dbc.Form([
                        dbc.Textarea(id="chat-input", placeholder="Enter additional instructions...", style={"width":"100%", "wordWrap": "break-word"}, className="mb-2"),
                    ]),
                    html.Div([
                        dbc.Button("Shred", id="shred-action-btn", className="me-3 mb-2 btn-primary"),
                        dbc.Button("Proposal", id="proposal-action-btn", className="me-3 mb-2 btn-secondary"),
                        dbc.Button("Compliance", id="compliance-action-btn", className="me-3 mb-2 btn-tertiary"),
                        dbc.Button("Recover", id="recover-action-btn", className="me-3 mb-2 btn-tertiary"),
                        dbc.Button("Virtual Board", id="board-action-btn", className="me-3 mb-2 btn-tertiary"),
                        dbc.Button("LOE", id="loe-action-btn", className="mb-2 btn-tertiary"),
                    ], className="mt-3 mb-3 d-flex flex-wrap"),
                    dcc.Loading(
                        id="loading",
                        type="default",
                        children=html.Div(id="output-data-upload"),
                        style={"textAlign": "center"}
                    )
                ])
            ], style={'backgroundColor': 'white'})
        ], style={'width':'70vw','maxWidth':'70vw'}, width=9)
    ], style={'marginTop':'20px'})
], fluid=True)

@app.callback(
    Output('shred-store', 'data'),
    Output('proposal-store', 'data'),
    Output('output-data-upload', 'children'),
    Output('uploaded-doc-list', 'children'),
    Output('select-document-dropdown', 'options'),
    Output('select-document-dropdown', 'value'),
    Output('shredded-doc-list', 'children'),
    Output('generated-doc-list', 'children'),
    Output('select-generated-dropdown', 'options'),
    Output('select-generated-dropdown', 'value'),
    Output('generated-rfp-section', 'children'),
    [
        Input('shred-action-btn', 'n_clicks'),
        Input('proposal-action-btn', 'n_clicks'),
        Input('upload-document', 'contents'),
        State('upload-document', 'filename'),
        Input({'type': 'delete-doc-btn', 'index': ALL, 'group': 'rfp'}, 'n_clicks'),
        State('select-document-dropdown', 'value'),
        Input('upload-proposal', 'contents'),
        State('upload-proposal', 'filename'),
        Input({'type': 'delete-proposal-btn', 'index': ALL, 'group': 'proposal'}, 'n_clicks'),
        State('select-proposal-dropdown', 'value'),
        Input({'type': 'delete-generated-btn', 'index': ALL, 'group': 'generated'}, 'n_clicks'),
        State('select-generated-dropdown', 'value'),
        Input({'type': 'delete-shredded-btn', 'index': ALL, 'group': 'shredded'}, 'n_clicks'),
        State('shredded-doc-list', 'children'),
        Input('select-generated-dropdown', 'value'),
        State('chat-input', 'value'),
        State('select-document-dropdown', 'value'),
        State('select-proposal-dropdown', 'value'),
        State('select-generated-dropdown', 'value'),
    ],
    prevent_initial_call=True
)
def master_callback(
    shred_clicks, proposal_clicks,
    rfp_content, rfp_filename, rfp_delete_clicks, selected_doc,
    proposal_content, proposal_filename, proposal_delete_clicks, selected_proposal,
    generated_delete_clicks, selected_generated,
    shredded_delete_clicks, shredded_doc_children,
    select_generated_value,
    chat_input, selected_filename, selected_proposal_dropdown, selected_generated_dropdown_state
):
    ctx = callback_context
    triggered_id = ctx.triggered[0]['prop_id'].split('.')[0] if ctx.triggered else None

    def safe_get_n_clicks(ctx, idx):
        try:
            return ctx.inputs_list[idx]
        except Exception:
            return []

    upload_triggered = False

    shred_store = {'text': None, 'docx_name': None}
    proposal_store = {'text': None, 'docx_name': None}

    rfp_delete_clicks = safe_get_n_clicks(ctx, 4)
    proposal_delete_clicks = safe_get_n_clicks(ctx, 8)
    generated_delete_clicks = safe_get_n_clicks(ctx, 10)
    shredded_delete_clicks = safe_get_n_clicks(ctx, 12)

    if triggered_id == 'upload-document' and rfp_content is not None and rfp_filename:
        content_type, content_string = rfp_content.split(',')
        decoded = base64.b64decode(content_string)
        text = decode_document(decoded)
        fileid = None
        if rfp_filename.lower().endswith(('.pdf', '.docx', '.xlsx', '.xls')):
            fileid = upload_to_gemini_file(decoded, rfp_filename)
        if text is not None:
            uploaded_documents[rfp_filename] = text
            if fileid:
                uploaded_documents_fileid[rfp_filename] = fileid
            logging.info(f"Document uploaded: {rfp_filename}")
        else:
            logging.error(f"Failed to decode uploaded document: {rfp_filename}")
        upload_triggered = True

    if triggered_id == 'upload-proposal' and proposal_content is not None and proposal_filename:
        content_type, content_string = proposal_content.split(',')
        decoded = base64.b64decode(content_string)
        text = decode_document(decoded)
        fileid = None
        if proposal_filename.lower().endswith(('.pdf', '.docx', '.xlsx', '.xls')):
            fileid = upload_to_gemini_file(decoded, proposal_filename)
        if text is not None:
            uploaded_proposals[proposal_filename] = text
            if fileid:
                uploaded_proposals_fileid[proposal_filename] = fileid
            logging.info(f"Proposal uploaded: {proposal_filename}")
        else:
            logging.error(f"Failed to decode uploaded proposal: {proposal_filename}")
        upload_triggered = True

    if triggered_id and isinstance(rfp_delete_clicks, list):
        for i, n_click in enumerate(rfp_delete_clicks):
            if n_click:
                btn_id = ctx.inputs_list[4][i]['id']
                del_filename = btn_id['index']
                if del_filename in uploaded_documents:
                    del uploaded_documents[del_filename]
                    if del_filename in uploaded_documents_fileid:
                        try:
                            genai.delete_file(uploaded_documents_fileid[del_filename])
                        except Exception as e:
                            logging.warning(f"Failed to delete Gemini file {del_filename}: {e}")
                        del uploaded_documents_fileid[del_filename]
                    logging.info(f"Document deleted: {del_filename}")
                    if selected_doc == del_filename:
                        selected_doc = None
                    upload_triggered = True
                    break

    if triggered_id and isinstance(proposal_delete_clicks, list):
        for i, n_click in enumerate(proposal_delete_clicks):
            if n_click:
                btn_id = ctx.inputs_list[8][i]['id']
                del_filename = btn_id['index']
                if del_filename in uploaded_proposals:
                    del uploaded_proposals[del_filename]
                    if del_filename in uploaded_proposals_fileid:
                        try:
                            genai.delete_file(uploaded_proposals_fileid[del_filename])
                        except Exception as e:
                            logging.warning(f"Failed to delete Gemini proposal file {del_filename}: {e}")
                        del uploaded_proposals_fileid[del_filename]
                    logging.info(f"Proposal deleted: {del_filename}")
                    if selected_proposal == del_filename:
                        selected_proposal = None
                    upload_triggered = True
                    break

    if triggered_id and isinstance(generated_delete_clicks, list):
        for i, n_click in enumerate(generated_delete_clicks):
            if n_click:
                btn_id = ctx.inputs_list[10][i]['id']
                del_filename = btn_id['index']
                if del_filename in generated_documents:
                    del generated_documents[del_filename]
                    logging.info(f"Generated doc deleted: {del_filename}")
                    if selected_generated == del_filename:
                        selected_generated = None
                    upload_triggered = True
                    break

    if triggered_id and isinstance(shredded_delete_clicks, list):
        for i, n_click in enumerate(shredded_delete_clicks):
            if n_click:
                btn_id = ctx.inputs_list[12][i]['id']
                del_filename = btn_id['index']
                if del_filename in shredded_documents:
                    del shredded_documents[del_filename]
                    logging.info(f"Shredded doc deleted: {del_filename}")
                upload_triggered = True
                break

    doc_options = [{'label': fn, 'value': fn} for fn in uploaded_documents.keys()]
    doc_value = selected_doc if selected_doc in uploaded_documents else (next(iter(uploaded_documents), None) if uploaded_documents else None)
    shredded_doc_list_items = get_shredded_doc_list(shredded_documents)
    uploaded_doc_list = get_uploaded_doc_list(uploaded_documents)
    generated_doc_list = get_generated_doc_list(generated_documents)
    generated_doc_options = [{'label': fn, 'value': fn} for fn in generated_documents.keys()]
    generated_doc_value = select_generated_value if select_generated_value in generated_documents else (next(iter(generated_documents), None) if generated_documents else None)
    generated_rfp_section = get_generated_rfp_download_section(generated_doc_value)

    output_data_upload = html.Div("No action taken yet.", style={"wordWrap": "break-word"})

    if triggered_id == 'shred-action-btn':
        output_data_upload = dcc.Loading(type="default", children=html.Div("Shredding document...", style={"wordWrap": "break-word"}))
        shred_text, _, shredded_docx_name, shredded_text = process_document('shred', doc_value, chat_input)
        shred_store = {'text': shred_text, 'docx_name': shredded_docx_name}

        if shredded_docx_name:
            doc_options = [{'label': fn, 'value': fn} for fn in uploaded_documents.keys()]
            doc_value = shredded_docx_name if shredded_docx_name in uploaded_documents else (next(iter(uploaded_documents), None) if uploaded_documents else None)
            uploaded_doc_list = get_uploaded_doc_list(uploaded_documents)
            shredded_doc_list_items = get_shredded_doc_list(shredded_documents)
        output_data_upload = dcc.Markdown(shred_text, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
        generated_rfp_section = get_generated_rfp_download_section(generated_doc_value)
        return (
            shred_store, proposal_store, output_data_upload,
            uploaded_doc_list, doc_options, doc_value,
            shredded_doc_list_items,
            generated_doc_list, generated_doc_options, generated_doc_value,
            generated_rfp_section
        )

    if triggered_id == 'proposal-action-btn':
        output_data_upload = dcc.Loading(type="default", children=html.Div("Generating proposal...", style={"wordWrap": "break-word"}))
        proposal_text, _, proposal_docx_name, proposal_text_preview = process_document('proposal', doc_value, chat_input)
        proposal_store = {'text': proposal_text, 'docx_name': proposal_docx_name}
        new_generated_doc_value = generated_doc_value
        if proposal_docx_name:
            generated_doc_options = [{'label': fn, 'value': fn} for fn in generated_documents.keys()]
            new_generated_doc_value = proposal_docx_name if proposal_docx_name in generated_documents else (next(iter(generated_documents), None) if generated_documents else None)
            generated_doc_list = get_generated_doc_list(generated_documents)
            generated_rfp_section = get_generated_rfp_download_section(new_generated_doc_value)
            logging.info(f"Generated proposal docx saved: {proposal_docx_name}")
        else:
            generated_rfp_section = get_generated_rfp_download_section(generated_doc_value)
        output_data_upload = dcc.Markdown(proposal_text, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
        return (
            shred_store, proposal_store, output_data_upload,
            uploaded_doc_list, doc_options, doc_value,
            shredded_doc_list_items,
            generated_doc_list, generated_doc_options, new_generated_doc_value,
            generated_rfp_section
        )

    if triggered_id == 'select-generated-dropdown':
        generated_rfp_section = get_generated_rfp_download_section(select_generated_value)
        return (
            shred_store, proposal_store, output_data_upload,
            uploaded_doc_list, doc_options, doc_value,
            shredded_doc_list_items,
            generated_doc_list, generated_doc_options, generated_doc_value,
            generated_rfp_section
        )

    if upload_triggered:
        doc_value = doc_value if doc_value in uploaded_documents else (next(iter(uploaded_documents), None) if uploaded_documents else None)
        generated_doc_value = generated_doc_value if generated_doc_value in generated_documents else (next(iter(generated_documents), None) if generated_documents else None)
        generated_rfp_section = get_generated_rfp_download_section(generated_doc_value)
        output_data_upload = html.Div("Upload/Delete completed.", style={"wordWrap": "break-word"})
        return (
            shred_store, proposal_store, output_data_upload,
            uploaded_doc_list, doc_options, doc_value,
            shredded_doc_list_items,
            generated_doc_list, generated_doc_options, generated_doc_value,
            generated_rfp_section
        )

    doc_value = doc_value if doc_value in uploaded_documents else (next(iter(uploaded_documents), None) if uploaded_documents else None)
    generated_doc_value = generated_doc_value if generated_doc_value in generated_documents else (next(iter(generated_documents), None) if generated_documents else None)
    generated_rfp_section = get_generated_rfp_download_section(generated_doc_value)
    return (
        shred_store, proposal_store, output_data_upload,
        uploaded_doc_list, doc_options, doc_value,
        shredded_doc_list_items,
        generated_doc_list, generated_doc_options, generated_doc_value,
        generated_rfp_section
    )

if __name__ == '__main__':
    print("Starting the Dash application...")
    app.run(debug=True, host='0.0.0.0', port=7860, threaded=True)
    print("Dash application has finished running.")