Update app.py

app.py

@@ -25,8 +25,6 @@ uploaded_files = {}
 current_document = None
 document_type = None
 shredded_document = None
-pink_review_document = None
-red_review_document = None
 generated_documents = {}
 
 # Document types and their descriptions
@@ -83,14 +81,10 @@ app.layout = dbc.Container([
                 type="dot",
                 children=[html.Div(id="loading-output")]
             ),
-            html.Div(id='document-preview', className="border p-3 mb-3"),
-            dbc.Button("Download Document", id="btn-download", color="success", className="mt-3"),
-            dcc.Download(id="download-document"),
-            html.Hr(),
-            html.Div(id='review-upload', style={'display': 'none'}, children=[
+            html.Div(id='document-upload-area', style={'display': 'none'}, children=[
                 dcc.Upload(
-                    id='upload-review',
-                    children=html.Div(['Drag and Drop or ', html.A('Select Review File')]),
+                    id='upload-specific-document',
+                    children=html.Div(['Drag and Drop or ', html.A('Select Document')]),
                     style={
                         'width': '100%',
                         'height': '60px',
@@ -103,8 +97,23 @@ app.layout = dbc.Container([
                     },
                     multiple=False
                 ),
-                html.Div(id='review-file-name')
+                html.Div(id='specific-document-name')
             ]),
+            html.Div(id='document-choice', style={'display': 'none'}, children=[
+                dcc.RadioItems(
+                    id='document-source',
+                    options=[
+                        {'label': 'Use Generated Document', 'value': 'generated'},
+                        {'label': 'Use Uploaded Document', 'value': 'uploaded'}
+                    ],
+                    value='generated'
+                )
+            ]),
+            dbc.Button("Process Document", id="btn-process-document", color="primary", className="mt-3", style={'display': 'none'}),
+            html.Div(id='document-preview', className="border p-3 mb-3"),
+            dbc.Button("Download Document", id="btn-download", color="success", className="mt-3"),
+            dcc.Download(id="download-document"),
+            html.Hr(),
             dcc.Loading(
                 id="chat-loading",
                 type="dot",
@@ -158,8 +167,8 @@ def update_output(list_of_contents, list_of_names, existing_files):
         if existing_files is None:
             existing_files = []
         shredded_document = None  # Reset shredded document when new files are uploaded
-        return existing_files + new_files, "Document uploaded. Please click 'Shred' to proceed."
-    return existing_files, "Please upload a document and click 'Shred' to begin."
+        return existing_files + new_files, "Document uploaded. Please select a document type to proceed."
+    return existing_files, "Please upload a document to begin."
 
 @app.callback(
     Output('file-list', 'children', allow_duplicate=True),
@@ -176,108 +185,91 @@ def remove_file(n_clicks, existing_files):
     removed_file = ctx.triggered[0]['prop_id'].split(',')[0].split(':')[-1].strip('}')
     uploaded_files.pop(removed_file, None)
     shredded_document = None  # Reset shredded document when a file is removed
-    return [file for file in existing_files if file['props']['children'][1]['props']['children'] != removed_file], "Document removed. Please upload a document and click 'Shred' to begin."
+    return [file for file in existing_files if file['props']['children'][1]['props']['children'] != removed_file], "Document removed. Please upload a document to begin."
 
-def generate_document(document_type, file_contents):
-    prompt = f"""Generate a {document_type} based on the following project artifacts:
-{' '.join(file_contents)}
-Instructions:
-1. Create the {document_type} as a detailed document.
-2. Use proper formatting and structure.
-3. Include all necessary sections and details.
-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
-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 
-approach with innovation for efficiency, 
-productivity, quality and measurable 
-outcomes, the industry standard that 
-methodology is based on if applicable, 
-detail the workflow or steps to accomplish 
-the requirement with labor categories that 
-will do those tasks in that workflow, 
-reference reputable research like gartner, 
-forrester, IDC, Deloitte, Accenture etc 
-with measures of success and substantiation 
-of MicroHealth's approach. Never use soft words
-like maybe, could be, should, possible be definitive in your language and confident.
-6. you must also take into account section L&M of the document which is the evaluation criteria
-to be sure we address them.
-Now, generate the {document_type}:
-"""
-
-    response = model.generate_content(prompt)
-    return response.text
+@app.callback(
+    Output('document-upload-area', 'style'),
+    Output('document-choice', 'style'),
+    Output('btn-process-document', 'style'),
+    [Input(f'btn-{doc_type.lower().replace("_", "-")}', 'n_clicks') for doc_type in document_types.keys()],
+    prevent_initial_call=True
+)
+def show_document_options(*args):
+    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()
+    
+    if document_type == "Shred":
+        return {'display': 'none'}, {'display': 'none'}, {'display': 'block'}
+    else:
+        return {'display': 'block'}, {'display': 'block'}, {'display': 'block'}
 
 @app.callback(
     Output('document-preview', 'children'),
     Output('loading-output', 'children'),
     Output('status-bar', 'children', allow_duplicate=True),
-    Output('review-upload', 'style'),
-    [Input(f'btn-{doc_type.lower().replace("_", "-")}', 'n_clicks') for doc_type in document_types.keys()],
-    State('review-file-name', 'children'),
+    Input('btn-process-document', 'n_clicks'),
+    State('document-source', 'value'),
+    State('specific-document-name', 'children'),
+    [State(f'btn-{doc_type.lower().replace("_", "-")}', 'n_clicks') for doc_type in document_types.keys()],
     prevent_initial_call=True
 )
-def generate_document_preview(*args):
+def generate_document_preview(n_clicks, doc_source, uploaded_doc_name, *args):
     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()
-    review_file = args[-1]
-
+    
+    # Determine which document type button was last clicked
+    for i, arg in enumerate(args):
+        if arg is not None and arg > 0:
+            document_type = list(document_types.keys())[i]
+            break
+    
     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'}
+        return html.Div("Please upload a document first."), "", "Please upload a document first."
 
-    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'}
-        file_contents = list(uploaded_files.values())
-        try:
+    try:
+        if document_type == "Shred":
+            file_contents = list(uploaded_files.values())
             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)}")
-            return html.Div(f"Error generating document: {str(e)}"), "Error", "An error occurred while shredding the document.", {'display': 'none'}
-
-    if shredded_document is None:
-        return html.Div("Please shred a document first."), "", "Please shred a document 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":
-            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])
+            current_document = shredded_document
+        elif doc_source == 'uploaded' and uploaded_doc_name:
+            current_document = process_document(uploaded_doc_name, uploaded_doc_name)
         else:
-            current_document = generate_document(document_type, [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."
+                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."
+                current_document = generate_document(document_type, [generated_documents['Red Review'], 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."
+                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
 
-        return dcc.Markdown(current_document), f"{document_type} generated", f"{document_type} document generated successfully.", {'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.", {'display': 'none'}
+        return html.Div(f"Error generating document: {str(e)}"), "Error", "An error occurred while generating the document."
 
 @app.callback(
-    Output('review-file-name', 'children'),
-    Input('upload-review', 'contents'),
-    State('upload-review', 'filename')
+    Output('specific-document-name', 'children'),
+    Input('upload-specific-document', 'contents'),
+    State('upload-specific-document', 'filename')
 )
-def update_review_filename(contents, filename):
+def update_specific_document_name(contents, filename):
     if contents is not None:
         return filename
     return ""
@@ -319,7 +311,26 @@ 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", "Shred"]:
+        # 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
+
+@app.callback(
+    Output("download-document", "data"),
+    Input("btn-download", "n_clicks"),
+    prevent_initial_call=True
+)
+def download_document(n_clicks):
+    global current_document, document_type
+    if current_document is None:
+        raise dash.exceptions.PreventUpdate
+    
+    if document_type in ["LOE", "Pink Review", "Red Review", "Gold Review", "Virtual Board", "Shred"]:
         # Create a pandas DataFrame for spreadsheet-type documents
         df = pd.read_csv(StringIO(current_document))