Update app.py

app.py

@@ -29,10 +29,10 @@ document_type = None
 document_types = {
     "Shred": "Generate an outline of the Project Work Statement (PWS)",
     "Pink": "Create a Pink Team document based on the PWS outline",
-    "P.Review": "Evaluate compliance of the Pink Team document",
+    "P_Review": "Evaluate compliance of the Pink Team document",
     "Red": "Generate a Red Team document based on the P.Review",
-    "R.Review": "Evaluate compliance of the Red Team document",
-    "G.Review": "Perform a final compliance review",
+    "R_Review": "Evaluate compliance of the Red Team document",
+    "G_Review": "Perform a final compliance review",
     "LOE": "Generate a Level of Effort (LOE) breakdown"
 }
 
@@ -63,7 +63,7 @@ app.layout = dbc.Container([
             html.Div([
                 dbc.Button(
                     doc_type,
-                    id=f'btn-{doc_type.lower().replace(" ", "-")}',
+                    id=f'btn-{doc_type.lower().replace("_", "-")}',
                     color="link",
                     className="mb-2 w-100 text-left custom-button",
                     style={'overflow': 'hidden', 'text-overflow': 'ellipsis', 'white-space': 'nowrap'}
@@ -168,7 +168,7 @@ Now, generate the {document_type}:
 @app.callback(
     Output('document-preview', 'children'),
     Output('loading-output', 'children'),
-    [Input(f'btn-{doc_type.lower().replace(" ", "-")}', 'n_clicks') for doc_type in document_types.keys()],
+    [Input(f'btn-{doc_type.lower().replace("_", "-")}', 'n_clicks') for doc_type in document_types.keys()],
     prevent_initial_call=True
 )
 def generate_document_preview(*args):
@@ -177,7 +177,7 @@ def generate_document_preview(*args):
     if not ctx.triggered:
         raise dash.exceptions.PreventUpdate
     button_id = ctx.triggered[0]['prop_id'].split('.')[0]
-    document_type = button_id.replace('btn-', '').replace('-', ' ').title()
+    document_type = button_id.replace('btn-', '').replace('-', '_').title()
     
     if not uploaded_files:
         return html.Div("Please upload project artifacts before generating a document."), ""