Update app.py via AI Editor

app.py

@@ -124,6 +124,7 @@ def get_right_col_content(selected_type):
         )
 
 app.layout = dbc.Container([
+    dcc.Store(id='selected-doc-type', data="Shred"),
     dbc.Row([
         dbc.Col(
             dbc.Card(
@@ -152,15 +153,9 @@ app.layout = dbc.Container([
                     ),
                     html.Div(id='file-list'),
                     html.Hr(),
-                    html.Div([
-                        dbc.Button(
-                            doc_type,
-                            id={'type': 'btn-doc-type', 'index': doc_type},
-                            color="link",
-                            className="mb-2 w-100 text-left custom-button",
-                            style={'overflow': 'hidden', 'text-overflow': 'ellipsis', 'white-space': 'nowrap'}
-                        ) for doc_type in document_types.keys()
-                    ])
+                    html.Div(
+                        id='doc-type-buttons'
+                    )
                 ])
             ),
             width=3
@@ -172,6 +167,52 @@ app.layout = dbc.Container([
     ])
 ], fluid=True)
 
+@app.callback(
+    Output('doc-type-buttons', 'children'),
+    Input('selected-doc-type', 'data')
+)
+def render_doc_type_buttons(selected_type):
+    buttons = []
+    for doc_type in document_types.keys():
+        btn_style = {'overflow': 'hidden', 'text-overflow': 'ellipsis', 'white-space': 'nowrap'}
+        btn_class = "mb-2 w-100 text-left custom-button"
+        if doc_type == selected_type:
+            btn_class += " active-doc-type"
+        buttons.append(
+            dbc.Button(
+                doc_type,
+                id={'type': 'btn-doc-type', 'index': doc_type},
+                color="link",
+                className=btn_class,
+                style=btn_style
+            )
+        )
+    return buttons
+
+@app.callback(
+    Output('selected-doc-type', 'data'),
+    [Input({'type': 'btn-doc-type', 'index': ALL}, 'n_clicks')],
+    [State({'type': 'btn-doc-type', 'index': ALL}, 'id')],
+    prevent_initial_call=True
+)
+def update_selected_doc_type(n_clicks_list, btn_ids):
+    triggered = callback_context.triggered
+    if not triggered or all(x is None for x in n_clicks_list):
+        raise dash.exceptions.PreventUpdate
+    idx = [i for i, x in enumerate(n_clicks_list) if x]
+    if idx:
+        selected_type = btn_ids[idx[-1]]['index']
+    else:
+        selected_type = "Shred"
+    return selected_type
+
+@app.callback(
+    Output('right-col-content', 'children'),
+    Input('selected-doc-type', 'data')
+)
+def update_right_col(selected_type):
+    return get_right_col_content(selected_type)
+
 def process_document(contents, filename):
     content_type, content_string = contents.split(',')
     decoded = base64.b64decode(content_string)
@@ -235,73 +276,6 @@ def remove_file(n_clicks, existing_files):
     logging.info(f"Removed file: {removed_file}")
     return [file for file in existing_files if file['props']['children'][1]['props']['children'] != removed_file]
 
-@app.callback(
-    Output('right-col-content', 'children'),
-    [Input({'type': 'btn-doc-type', 'index': ALL}, 'n_clicks')],
-    [State({'type': 'btn-doc-type', 'index': ALL}, 'id')]
-)
-def update_right_col(n_clicks_list, btn_ids):
-    triggered = callback_context.triggered
-    if not triggered or all(x is None for x in n_clicks_list):
-        selected_type = "Shred"
-    else:
-        idx = [i for i, x in enumerate(n_clicks_list) if x]
-        if idx:
-            selected_type = btn_ids[idx[-1]]['index']
-        else:
-            selected_type = "Shred"
-    return get_right_col_content(selected_type)
-
-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 
-Pink Review, Red Review, Gold Review, or spreadsheet, loe or virtual board
-then your goal is to be compliant and compelling based on the
-requrements, 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}:
-"""
-
-    logging.info(f"Generating document for type: {document_type}")
-    try:
-        response = openai.ChatCompletion.create(
-            model="gpt-4-1106-preview",
-            messages=[
-                {"role": "system", "content": "You are a helpful, expert government proposal writer."},
-                {"role": "user", "content": prompt}
-            ],
-            max_tokens=4096,
-            temperature=0.25,
-        )
-        generated_text = response['choices'][0]['message']['content']
-        logging.info("Document generated successfully.")
-        return generated_text
-    except Exception as e:
-        logging.error(f"Error generating document: {str(e)}")
-        raise
-
 @app.callback(
     Output('document-preview', 'children'),
     Output('loading-output', 'children'),
@@ -405,6 +379,56 @@ def generate_other_doc(n_clicks_list, btn_ids, radio_values, upload_contents, up
         logging.error(f"Error generating document: {str(e)}")
         return html.Div(f"Error generating document: {str(e)}"), "Error"
 
+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 
+Pink Review, Red Review, Gold Review, or spreadsheet, loe or virtual board
+then your goal is to be compliant and compelling based on the
+requrements, 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}:
+"""
+
+    logging.info(f"Generating document for type: {document_type}")
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-4-1106-preview",
+            messages=[
+                {"role": "system", "content": "You are a helpful, expert government proposal writer."},
+                {"role": "user", "content": prompt}
+            ],
+            max_tokens=4096,
+            temperature=0.25,
+        )
+        generated_text = response['choices'][0]['message']['content']
+        logging.info("Document generated successfully.")
+        return generated_text
+    except Exception as e:
+        logging.error(f"Error generating document: {str(e)}")
+        raise
+
 @app.callback(
     Output('chat-output', 'children'),
     Output('document-preview', 'children', allow_duplicate=True),