Update app.py via AI Editor

app.py

@@ -52,8 +52,7 @@ def get_right_col_content(selected_type):
         type="dot",
         children=[html.Div(id="loading-output")]
     ))
-    controls.append(html.Div(id='document-preview', className="border p-3 mb-3"))
-    # Generate and Download buttons in a row, Generate left of Download
+    # Generate and Download buttons at the top
     controls.append(
         dbc.Row([
             dbc.Col(
@@ -68,6 +67,7 @@ def get_right_col_content(selected_type):
         ], className="mb-2")
     )
     controls.append(html.Hr())
+    controls.append(html.Div(id='document-preview', className="border p-3 mb-3"))
     # Upload and radio only for non-shred
     if selected_type != "Shred":
         controls.append(html.Div([
@@ -99,22 +99,57 @@ def get_right_col_content(selected_type):
                 className="mb-2"
             ),
         ], id={'type': 'doc-type-controls', 'index': selected_type}))
-    # Chatbox below HR
-    controls.append(
-        dcc.Loading(
-            id="chat-loading",
-            type="dot",
-            children=[
-                dbc.Input(id="chat-input", type="text", placeholder="Chat with AI to update document...", className="mb-2", style={'whiteSpace':'pre-wrap'}),
-                dbc.Button("Send", id="btn-send-chat", color="primary", className="mb-3"),
-                html.Div(id="chat-output")
-            ]
-        )
-    )
     return dbc.Card(dbc.CardBody(controls))
 
+def get_left_col_content():
+    # Chat card under doc-type buttons
+    chat_card = dbc.Card(
+        dbc.CardBody([
+            html.H5("AI Chat", className="mb-2"),
+            dcc.Loading(
+                id="chat-loading",
+                type="dot",
+                children=[
+                    dbc.Input(id="chat-input", type="text", placeholder="Chat with AI to update document...", className="mb-2", style={'whiteSpace':'pre-wrap'}),
+                    dbc.Button("Send", id="btn-send-chat", color="primary", className="mb-3"),
+                    html.Div(id="chat-output")
+                ]
+            )
+        ]),
+        className="mt-4"
+    )
+    return [
+        html.H4("Proposal Documents", className="mt-3 mb-4"),
+        html.Div([
+            html.Div(className="blinking-dot", style={'margin':'0 auto','width':'16px','height':'16px'}),
+        ], style={'textAlign':'center', 'marginBottom':'10px'}),
+        dcc.Upload(
+            id='upload-document',
+            children=html.Div([
+                'Drag and Drop or ',
+                html.A('Select Files')
+            ]),
+            style={
+                'width': '100%',
+                'height': '60px',
+                'lineHeight': '60px',
+                'borderWidth': '1px',
+                'borderStyle': 'dashed',
+                'borderRadius': '5px',
+                'textAlign': 'center',
+                'margin': '10px 0'
+            },
+            multiple=True
+        ),
+        file_list_component(),
+        html.Hr(),
+        html.Div(
+            id='doc-type-buttons'
+        ),
+        chat_card
+    ]
+
 def file_list_component():
-    # Each file gets a row with a unique id for remove button
     return html.Div(
         id='file-list-container',
         children=[
@@ -127,35 +162,7 @@ app.layout = dbc.Container([
     dbc.Row([
         dbc.Col(
             dbc.Card(
-                dbc.CardBody([
-                    html.H4("Proposal Documents", className="mt-3 mb-4"),
-                    html.Div([
-                        html.Div(className="blinking-dot", style={'margin':'0 auto','width':'16px','height':'16px'}),
-                    ], style={'textAlign':'center', 'marginBottom':'10px'}),
-                    dcc.Upload(
-                        id='upload-document',
-                        children=html.Div([
-                            'Drag and Drop or ',
-                            html.A('Select Files')
-                        ]),
-                        style={
-                            'width': '100%',
-                            'height': '60px',
-                            'lineHeight': '60px',
-                            'borderWidth': '1px',
-                            'borderStyle': 'dashed',
-                            'borderRadius': '5px',
-                            'textAlign': 'center',
-                            'margin': '10px 0'
-                        },
-                        multiple=True
-                    ),
-                    file_list_component(),
-                    html.Hr(),
-                    html.Div(
-                        id='doc-type-buttons'
-                    )
-                ])
+                dbc.CardBody(get_left_col_content())
             ),
             width=3
         ),
@@ -297,7 +304,6 @@ def remove_file(n_clicks, existing_files):
     if existing_files:
         for file in existing_files:
             try:
-                # file is a dbc.Row, children[1] is dbc.Col with html.Span(name)
                 file_id = file['props']['id']
                 if file_id['index'] != removed_file:
                     filtered_files.append(file)
@@ -346,12 +352,10 @@ def generate_any_doc(n_clicks_list, btn_ids, radio_values, upload_contents, uplo
             logging.error(f"Error generating document: {str(e)}")
             return html.Div(f"Error generating document: {str(e)}"), "Error"
 
-    # For other doc types, handle radio/upload logic
     if shredded_document is None:
         logging.info("Shredded document is None. Aborting.")
         return html.Div("Please shred a document first."), ""
 
-    # Defensive: radio_values, upload_contents, upload_filenames may be None or short
     source = None
     if radio_values and len(radio_values) > idx:
         source = radio_values[idx]
@@ -436,14 +440,19 @@ def spreadsheet_preview(text, doc_type=None):
     if doc_type in spreadsheet_types:
         try:
             df = pd.read_csv(StringIO(text))
-            return dbc.Table.from_dataframe(df, striped=True, bordered=True, hover=True, size="sm")
+            table = dbc.Table.from_dataframe(df, striped=True, bordered=True, hover=True, size="sm")
+            return html.Div(table)
         except Exception as e:
             logging.error(f"Error parsing CSV for preview: {str(e)}")
-            return html.Pre("Unable to display spreadsheet preview. Raw output:\n" + text, style={'whiteSpace': 'pre-wrap', 'fontFamily': 'monospace'})
+            return html.Div([
+                html.Div("Unable to display spreadsheet preview. Raw output:", style={'fontWeight': 'bold'}),
+                html.Pre(text, style={'whiteSpace': 'pre-wrap', 'fontFamily': 'monospace'})
+            ])
     else:
         try:
             df = pd.read_csv(StringIO(text))
-            return dbc.Table.from_dataframe(df, striped=True, bordered=True, hover=True, size="sm")
+            table = dbc.Table.from_dataframe(df, striped=True, bordered=True, hover=True, size="sm")
+            return html.Div(table)
         except Exception:
             return html.Pre(text, style={'whiteSpace': 'pre-wrap', 'fontFamily': 'monospace'})