Update app.py

app.py

@@ -236,6 +236,19 @@ Now, generate the {document_type}:
     State('specific-document-name', 'children'),
     prevent_initial_call=True
 )
+@app.callback(
+    Output('document-preview', 'children'),
+    Output('loading-output', 'children'),
+    Output('status-bar', 'children', allow_duplicate=True),
+    Output('document-upload', 'style'),
+    Output('document-choice', 'style'),
+    Output('btn-generate-document', 'style'),
+    [Input(f'btn-{doc_type.lower().replace(" ", "-")}', 'n_clicks') for doc_type in document_types.keys()],
+    Input('btn-generate-document', 'n_clicks'),
+    State('document-source', 'value'),
+    State('specific-document-name', 'children'),
+    prevent_initial_call=True
+)
 def generate_document_preview(*args):
     global current_document, document_type, shredded_document, generated_documents
     ctx = dash.callback_context
@@ -245,12 +258,12 @@ def generate_document_preview(*args):
     triggered_id = ctx.triggered[0]['prop_id'].split('.')[0]
     
     if triggered_id.startswith('btn-'):
-        document_type = triggered_id.replace('btn-', '').replace('-', '_').title()
+        document_type = triggered_id.replace('btn-', '').replace('-', ' ').title()
         if document_type == "Shred":
             return html.Div("Ready to shred. Click 'Generate Document' to proceed."), "", "Ready to shred.", {'display': 'none'}, {'display': 'none'}, {'display': 'block'}
         elif document_type in ["Pink", "Red", "Gold"]:
             return html.Div(f"Ready to generate {document_type} document."), "", f"Ready to generate {document_type} document.", {'display': 'block'}, {'display': 'flex'}, {'display': 'block'}
-        elif document_type in ["Pink_Review", "Red_Review", "Gold_Review", "Virtual_Board", "LOE"]:
+        elif document_type in ["Pink Review", "Red Review", "Gold Review", "Virtual Board", "Loe"]:
             return html.Div(f"Ready to generate {document_type}."), "", f"Ready to generate {document_type}.", {'display': 'block'}, {'display': 'flex'}, {'display': 'block'}
     
     if triggered_id == 'btn-generate-document':
@@ -279,28 +292,28 @@ def generate_document_preview(*args):
                 else:
                     current_document = generate_document(document_type, [shredded_document])
             elif document_type == "Red":
-                if 'Pink_Review' not in generated_documents:
+                if 'Pink Review' not in generated_documents:
                     return html.Div("Please complete Pink Review first."), "", "Please complete Pink Review first.", {'display': 'none'}, {'display': 'none'}, {'display': 'none'}
                 if document_source == 'uploaded' and specific_document:
-                    current_document = generate_document(document_type, [specific_document, generated_documents['Pink_Review'], shredded_document])
+                    current_document = generate_document(document_type, [specific_document, generated_documents['Pink Review'], shredded_document])
                 else:
-                    current_document = generate_document(document_type, [generated_documents['Pink_Review'], shredded_document])
+                    current_document = generate_document(document_type, [generated_documents['Pink Review'], shredded_document])
             elif document_type == "Gold":
-                if 'Red_Review' not in generated_documents:
+                if 'Red Review' not in generated_documents:
                     return html.Div("Please complete Red Review first."), "", "Please complete Red Review first.", {'display': 'none'}, {'display': 'none'}, {'display': 'none'}
                 if document_source == 'uploaded' and specific_document:
-                    current_document = generate_document(document_type, [specific_document, generated_documents['Red_Review'], shredded_document])
+                    current_document = generate_document(document_type, [specific_document, generated_documents['Red Review'], shredded_document])
                 else:
-                    current_document = generate_document(document_type, [generated_documents['Red_Review'], shredded_document])
-            elif document_type in ["Pink_Review", "Red_Review", "Gold_Review"]:
-                previous_document = generated_documents.get(document_type.split('_')[0], None)
+                    current_document = generate_document(document_type, [generated_documents['Red Review'], shredded_document])
+            elif document_type in ["Pink Review", "Red Review", "Gold Review"]:
+                previous_document = generated_documents.get(document_type.split(' ')[0], None)
                 if not previous_document:
-                    return html.Div(f"Please generate {document_type.split('_')[0]} document first."), "", f"Please generate {document_type.split('_')[0]} document first.", {'display': 'none'}, {'display': 'none'}, {'display': 'none'}
+                    return html.Div(f"Please generate {document_type.split(' ')[0]} document first."), "", f"Please generate {document_type.split(' ')[0]} document first.", {'display': 'none'}, {'display': 'none'}, {'display': 'none'}
                 if document_source == 'uploaded' and specific_document:
                     current_document = generate_document(document_type, [specific_document, previous_document, shredded_document])
                 else:
                     current_document = generate_document(document_type, [previous_document, shredded_document])
-            elif document_type in ["Virtual_Board", "LOE"]:
+            elif document_type in ["Virtual Board", "Loe"]:
                 if 'Gold' not in generated_documents:
                     return html.Div("Please complete Gold document first."), "", "Please complete Gold document first.", {'display': 'none'}, {'display': 'none'}, {'display': 'none'}
                 if document_source == 'uploaded' and specific_document:
@@ -318,7 +331,6 @@ def generate_document_preview(*args):
             return html.Div(f"Error generating document: {str(e)}"), "Error", "An error occurred while generating the document.", {'display': 'none'}, {'display': 'none'}, {'display': 'none'}
 
     raise dash.exceptions.PreventUpdate
-
 @app.callback(
     Output('specific-document-name', 'children'),
     Input('upload-specific-document', 'contents'),