Update app.py

app.py

@@ -236,20 +236,15 @@ Now, generate the {document_type}:
     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'),
-    State('file-list', 'children'),
     prevent_initial_call=True
 )
-def generate_document_preview(*args):
-    global current_document, document_type, shredded_document, generated_documents
+def update_document_type(*args):
+    global document_type
     ctx = dash.callback_context
     if not ctx.triggered:
         raise dash.exceptions.PreventUpdate
     
     triggered_id = ctx.triggered[0]['prop_id'].split('.')[0]
-    file_list = args[-1]
     
     if triggered_id.startswith('btn-'):
         document_type = triggered_id.replace('btn-', '').replace('-', ' ').title()
@@ -259,74 +254,86 @@ def generate_document_preview(*args):
             return html.Div(f"Ready to generate {document_type} document."), "", dash.no_update, {'display': 'block'}, {'display': 'flex'}, {'display': 'block'}
         elif document_type in ["Pink Review", "Red Review", "Gold Review", "Virtual Board", "Loe"]:
             return html.Div(f"Ready to generate {document_type}."), "", dash.no_update, {'display': 'block'}, {'display': 'flex'}, {'display': 'block'}
-        return dash.no_update, dash.no_update, dash.no_update, dash.no_update, dash.no_update, dash.no_update
+    
+    return dash.no_update, dash.no_update, dash.no_update, dash.no_update, dash.no_update, dash.no_update
 
-    if triggered_id == 'btn-generate-document':
-        document_source = args[-3]
-        specific_document = args[-2]
-        
-        if document_type == "Shred":
-            if not uploaded_files:
-                return html.Div("Please upload a document before shredding."), "", "Please upload a document before shredding.", {'display': 'none'}, {'display': 'none'}, {'display': 'none'}
-            file_contents = list(uploaded_files.values())
-            try:
-                shredded_document = generate_document(document_type, file_contents)
-                generated_documents['Shred'] = shredded_document
-                return dcc.Markdown(shredded_document), f"{document_type} generated", "Document shredded successfully.", {'display': 'none'}, {'display': 'none'}, {'display': 'none'}
-            except Exception as e:
-                print(f"Error generating document: {str(e)}")
-                return html.Div(f"Error generating document: {str(e)}"), "Error", "An error occurred while shredding the document.", {'display': 'none'}, {'display': 'none'}, {'display': 'none'}
-        
-        if shredded_document is None:
-            return html.Div("Please shred a document first."), "", "Please shred a document first.", {'display': 'none'}, {'display': 'none'}, {'display': 'none'}
-        
+@app.callback(
+    Output('document-preview', 'children', allow_duplicate=True),
+    Output('loading-output', 'children', allow_duplicate=True),
+    Output('status-bar', 'children', allow_duplicate=True),
+    Input('btn-generate-document', 'n_clicks'),
+    State('document-source', 'value'),
+    State('specific-document-name', 'children'),
+    prevent_initial_call=True
+)
+def generate_document_preview(n_clicks, document_source, specific_document):
+    global current_document, document_type, shredded_document, generated_documents
+    
+    if document_type == "Shred":
+        if not uploaded_files:
+            return html.Div("Please upload a document before shredding."), "", "Please upload a document before shredding."
+        file_contents = list(uploaded_files.values())
         try:
-            if document_type == "Pink":
-                if document_source == 'uploaded' and specific_document:
-                    current_document = generate_document(document_type, [specific_document, shredded_document])
-                else:
-                    current_document = generate_document(document_type, [shredded_document])
-            elif document_type == "Red":
-                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])
-                else:
-                    current_document = generate_document(document_type, [generated_documents['Pink Review'], shredded_document])
-            elif document_type == "Gold":
-                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])
-                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)
-                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'}
-                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"]:
-                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:
-                    current_document = generate_document(document_type, [specific_document, generated_documents['Gold'], shredded_document])
-                else:
-                    current_document = generate_document(document_type, [generated_documents['Gold'], shredded_document])
+            shredded_document = generate_document(document_type, file_contents)
+            generated_documents['Shred'] = shredded_document
+            return dcc.Markdown(shredded_document), f"{document_type} generated", "Document shredded successfully."
+        except Exception as e:
+            print(f"Error generating document: {str(e)}")
+            return html.Div(f"Error generating document: {str(e)}"), "Error", "An error occurred while shredding the document."
+    
+    if shredded_document is None:
+        return html.Div("Please shred a document first."), "", "Please shred a document first."
+    
+    try:
+        if document_type == "Pink":
+            if document_source == 'uploaded' and specific_document:
+                current_document = generate_document(document_type, [specific_document, shredded_document])
             else:
                 current_document = generate_document(document_type, [shredded_document])
+        elif document_type == "Red":
+            if 'Pink Review' not in generated_documents:
+                return html.Div("Please complete Pink Review first."), "", "Please complete Pink Review first."
+            if document_source == 'uploaded' and specific_document:
+                current_document = generate_document(document_type, [specific_document, generated_documents['Pink Review'],
+        elif document_type == "Red":
+            if 'Pink Review' not in generated_documents:
+                return html.Div("Please complete Pink Review first."), "", "Please complete Pink Review first."
+            if document_source == 'uploaded' and specific_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])
+        elif document_type == "Gold":
+            if 'Red Review' not in generated_documents:
+                return html.Div("Please complete Red Review first."), "", "Please complete Red Review first."
+            if document_source == 'uploaded' and specific_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)
+            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."
+            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"]:
+            if 'Gold' not in generated_documents:
+                return html.Div("Please complete Gold document first."), "", "Please complete Gold document first."
+            if document_source == 'uploaded' and specific_document:
+                current_document = generate_document(document_type, [specific_document, generated_documents['Gold'], shredded_document])
+            else:
+                current_document = generate_document(document_type, [generated_documents['Gold'], shredded_document])
+        else:
+            current_document = generate_document(document_type, [shredded_document])
 
-            generated_documents[document_type] = current_document
+        generated_documents[document_type] = current_document
 
-            return dcc.Markdown(current_document), f"{document_type} generated", f"{document_type} document generated successfully.", {'display': 'none'}, {'display': 'none'}, {'display': 'none'}
-        except Exception as e:
-            print(f"Error generating document: {str(e)}")
-            return html.Div(f"Error generating document: {str(e)}"), "Error", "An error occurred while generating the document.", {'display': 'none'}, {'display': 'none'}, {'display': 'none'}
+        return dcc.Markdown(current_document), f"{document_type} generated", f"{document_type} document generated successfully."
+    except Exception as e:
+        print(f"Error generating document: {str(e)}")
+        return html.Div(f"Error generating document: {str(e)}"), "Error", "An error occurred while generating the document."
 
-    return dash.no_update, dash.no_update, dash.no_update, dash.no_update, dash.no_update, dash.no_update
-    
 @app.callback(
     Output('specific-document-name', 'children'),
     Input('upload-specific-document', 'contents'),
@@ -374,7 +381,7 @@ def download_document(n_clicks):
     if current_document is None:
         raise dash.exceptions.PreventUpdate
     
-    if document_type in ["LOE", "Pink_Review", "Red_Review", "Gold_Review", "Virtual_Board"]:
+    if document_type in ["LOE", "Pink Review", "Red Review", "Gold Review", "Virtual Board"]:
         # Create a pandas DataFrame for spreadsheet-type documents
         df = pd.read_csv(StringIO(current_document))