Update app.py

app.py

@@ -120,7 +120,7 @@ def evaluate_compliance(document: str, requirements: str) -> str:
     response = model.generate_content(prompt)
     return response.text
 
-def generate_red_document(document: str, compliance_report: str) -> str:
+def generate_red_document(document: str, compliance_report: str, instructions: str) -> str:
     prompt = f"""
     Based on the following document and compliance report:
 
@@ -130,15 +130,18 @@ def generate_red_document(document: str, compliance_report: str) -> str:
     Compliance Report:
     {compliance_report}
 
+    Additional instructions: {instructions}
+
     Generate a revised "Red Team" document that addresses all issues found in the compliance report. 
     Follow these guidelines:
-    1. Use Wikipedia style writing with active voice.  Be firm with the approach, no soft words like could be, may be, should, might.  Use definitve language.
+    1. Use Wikipedia style writing with active voice.  Be firm with the approach, no soft words like could be, may be, should, might.  Use definitive language.
     2. For each requirement, describe in detail how MicroHealth will innovate to address it.
     3. Explain the industry best practices that will be applied and the workflow to accomplish the steps in the best practice to address the requirement.
     4. Provide measurable outcomes for the customer.
     5. Limit the use of bullet points and write predominantly in paragraph format.
     6. Ensure a logical flow of steps taken by MicroHealth for each requirement.
     7. Where applicable, describe the labor category or labor categories that perform the task as part of the process
+    8. {instructions}
     """
     response = model.generate_content(prompt)
     return response.text
@@ -253,6 +256,11 @@ app.layout = dbc.Container([
             dcc.Download(id="download-p-review-doc")
         ]),
         dbc.Tab(label="Red", tab_id="red", children=[
+            dbc.Textarea(
+                id='red-instructions',
+                placeholder="Enter any additional instructions for generating the Red Team document...",
+                style={'height': '100px', 'marginBottom': '10px'}
+            ),
             dcc.Upload(
                 id='upload-red',
                 children=html.Div(['Drag and Drop or ', html.A('Select Files')]),
@@ -442,9 +450,10 @@ def update_loe_output(n_clicks, upload_contents, shred_output):
     Input('generate-red', 'n_clicks'),
     State('upload-red', 'contents'),
     State('upload-red', 'filename'),
-    State('p-review-output', 'children')
+    State('p-review-output', 'children'),
+    State('red-instructions', 'value')
 )
-def update_red_output(n_clicks, contents, filename, p_review_output):
+def update_red_output(n_clicks, contents, filename, p_review_output, instructions):
     if n_clicks is None:
         return "Click 'Generate Red Team Document' to begin."
     
@@ -455,7 +464,7 @@ def update_red_output(n_clicks, contents, filename, p_review_output):
     else:
         return "Please upload a document or complete the P.Review first."
     
-    red_doc = generate_red_document(document, p_review_output)
+    red_doc = generate_red_document(document, p_review_output, instructions or "")
     return dcc.Markdown(red_doc)
 
 @app.callback(