Update app.py

app.py

@@ -26,6 +26,8 @@ current_document = None
 document_type = None
 shredded_document = None
 pink_review_document = None
+red_review_document = None
+generated_documents = {}
 
 # Document types and their descriptions
 document_types = {
@@ -34,7 +36,7 @@ document_types = {
     "Pink Review": "Evaluate compliance of the Pink Team document against the requirements and output a spreadsheet of non compliant findings by pws number, the goal of that pws section, what made it non compliant and your recommendations for recovery",
     "Red": "Produce a Red Team document based on the Pink Review by pws sections. Your goal is to be compliant and compelling by recovering all the findings in Pink Review",
     "Red Review": "Evaluate compliance of the Red Team document against the requirements and output a spreadsheet of non compliant findings by pws number, the goal of that pws section, what made it non compliant and your recommendations for recovery",
-    "Gold": "Create a Pink Team document based on the PWS response by pws sections. Your goal is to be compliant and compelling by recovering all the findings in Red Review",
+    "Gold": "Create a Gold Team document based on the PWS response by pws sections. Your goal is to be compliant and compelling by recovering all the findings in Red Review",
     "Gold Review": "Perform a final compliance review against the requirements and output a spreadsheet of non compliant findings by pws number, the goal of that pws section, what made it non compliant and your recommendations for recovery",
     "Virtual Board": "Based on the requirements and in particular the evaulation criteria, you will evaluate the proposal as if you were a contracting office and provide section by section evaluation as unsatisfactory, satisfactory, good, very good, excellent and why in a spreadsheet",
     "LOE": "Generate a Level of Effort (LOE) breakdown as a spreadsheet"
@@ -85,10 +87,10 @@ app.layout = dbc.Container([
             dbc.Button("Download Document", id="btn-download", color="success", className="mt-3"),
             dcc.Download(id="download-document"),
             html.Hr(),
-            html.Div(id='pink-review-upload', style={'display': 'none'}, children=[
+            html.Div(id='review-upload', style={'display': 'none'}, children=[
                 dcc.Upload(
-                    id='upload-pink-review',
-                    children=html.Div(['Drag and Drop or ', html.A('Select Pink Review File')]),
+                    id='upload-review',
+                    children=html.Div(['Drag and Drop or ', html.A('Select Review File')]),
                     style={
                         'width': '100%',
                         'height': '60px',
@@ -101,7 +103,7 @@ app.layout = dbc.Container([
                     },
                     multiple=False
                 ),
-                html.Div(id='pink-review-file-name')
+                html.Div(id='review-file-name')
             ]),
             dcc.Loading(
                 id="chat-loading",
@@ -186,7 +188,7 @@ Instructions:
 4. Start the output immediately with the document content.
 5. IMPORTANT: If the document type is Pink, Red, Gold and not review type, loe or board
 then your goal is to be compliant and compelling based on the
-requrements, write in paragraph in active voice as
+requirements, write in paragraph in active voice as
 MicroHealth, limit bullets, answer the
 requirement with what MicroHealth will do 
 to satisfy the requirement, the technical 
@@ -214,19 +216,19 @@ Now, generate the {document_type}:
     Output('document-preview', 'children'),
     Output('loading-output', 'children'),
     Output('status-bar', 'children', allow_duplicate=True),
-    Output('pink-review-upload', 'style'),
+    Output('review-upload', 'style'),
     [Input(f'btn-{doc_type.lower().replace("_", "-")}', 'n_clicks') for doc_type in document_types.keys()],
-    State('pink-review-file-name', 'children'),
+    State('review-file-name', 'children'),
     prevent_initial_call=True
 )
 def generate_document_preview(*args):
-    global current_document, document_type, shredded_document, pink_review_document
+    global current_document, document_type, shredded_document, generated_documents
     ctx = dash.callback_context
     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()
-    pink_review_file = args[-1]
+    review_file = args[-1]
 
     if not uploaded_files and document_type != "Shred":
         return html.Div("Please upload and shred a document first."), "", "Please upload and shred a document first.", {'display': 'none'}
@@ -237,6 +239,7 @@ def generate_document_preview(*args):
         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. You can now proceed with other operations.", {'display': 'none'}
         except Exception as e:
             print(f"Error generating document: {str(e)}")
@@ -245,22 +248,24 @@ def generate_document_preview(*args):
     if shredded_document is None:
         return html.Div("Please shred a document first."), "", "Please shred a document first.", {'display': 'none'}
 
-    if document_type == "Pink Review":
-        return html.Div("Please upload a Pink Team document or use the generated one."), "", "Please upload a Pink Team document or use the generated one.", {'display': 'block'}
-
-    if document_type in ["Red", "Red Review"] and pink_review_document is None:
-        return html.Div("Please complete Pink Review first."), "", "Please complete Pink Review first.", {'display': 'none'}
+    if document_type in ["Pink Review", "Red Review", "Gold Review"]:
+        return html.Div(f"Please upload a {document_type.split()[0]} Team document or use the generated one."), "", f"Please upload a {document_type.split()[0]} Team document or use the generated one.", {'display': 'block'}
 
     try:
-        if document_type == "Pink Review" and pink_review_file:
-            current_document = generate_document(document_type, [pink_review_file, shredded_document])
-        elif document_type in ["Red", "Red Review"]:
-            current_document = generate_document(document_type, [pink_review_document, shredded_document])
+        if document_type == "Pink":
+            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'}
+            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'}
+            current_document = generate_document(document_type, [generated_documents['Red Review'], shredded_document])
         else:
             current_document = generate_document(document_type, [shredded_document])
 
-        if document_type == "Pink Review":
-            pink_review_document = 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'}
     except Exception as e:
@@ -268,11 +273,11 @@ def generate_document_preview(*args):
         return html.Div(f"Error generating document: {str(e)}"), "Error", "An error occurred while generating the document.", {'display': 'none'}
 
 @app.callback(
-    Output('pink-review-file-name', 'children'),
-    Input('upload-pink-review', 'contents'),
-    State('upload-pink-review', 'filename')
+    Output('review-file-name', 'children'),
+    Input('upload-review', 'contents'),
+    State('upload-review', 'filename')
 )
-def update_pink_review_filename(contents, filename):
+def update_review_filename(contents, filename):
     if contents is not None:
         return filename
     return ""
@@ -314,14 +319,14 @@ def download_document(n_clicks):
     if current_document is None:
         raise dash.exceptions.PreventUpdate
     
-    if document_type == "LOE":
-        # Create a pandas DataFrame for LOE
+    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))
         
         # Save the DataFrame to an Excel file
         output = BytesIO()
         with pd.ExcelWriter(output, engine='xlsxwriter') as writer:
-            df.to_excel(writer, sheet_name='LOE', index=False)
+            df.to_excel(writer, sheet_name=document_type, index=False)
         
         return dcc.send_bytes(output.getvalue(), f"{document_type}.xlsx")
     else: