Update app.py via AI Editor

app.py

@@ -53,14 +53,23 @@ def get_right_col_content(selected_type):
         children=[html.Div(id="loading-output")]
     ))
     controls.append(html.Div(id='document-preview', className="border p-3 mb-3"))
-    controls.append(dbc.Button("Download Document", id="btn-download", color="success", className="mt-3"))
-    controls.append(dcc.Download(id="download-document"))
+    # Generate and Download buttons in a row, Generate left of Download
+    controls.append(
+        dbc.Row([
+            dbc.Col(
+                dbc.Button("Generate Document", id={'type': 'btn-generate-doc', 'index': selected_type}, color="primary", className="mb-3 w-100"),
+                width=6
+            ),
+            dbc.Col(
+                dbc.Button("Download Document", id="btn-download", color="success", className="mb-3 w-100"),
+                width=6
+            ),
+            dcc.Download(id="download-document")
+        ], className="mb-2")
+    )
     controls.append(html.Hr())
-    if selected_type == "Shred":
-        controls.append(
-            dbc.Button("Generate Document", id={'type': 'btn-generate-doc', 'index': selected_type}, color="primary", className="mb-3")
-        )
-    elif selected_type != "Shred":
+    # Upload and radio only for non-shred
+    if selected_type != "Shred":
         controls.append(html.Div([
             html.Label(f"Upload {selected_type} Document"),
             dcc.Upload(
@@ -89,18 +98,29 @@ def get_right_col_content(selected_type):
                 inline=True,
                 className="mb-2"
             ),
-            dbc.Button("Generate Document", id={'type': 'btn-generate-doc', 'index': selected_type}, color="primary", className="mb-3"),
         ], id={'type': 'doc-type-controls', 'index': selected_type}))
-    controls.append(dcc.Loading(
-        id="chat-loading",
-        type="dot",
+    # 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 file_list_component():
+    # Each file gets a row with a unique id for remove button
+    return html.Div(
+        id='file-list-container',
         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")
+            html.Div(id='file-list')
         ]
-    ))
-    return dbc.Card(dbc.CardBody(controls))
+    )
 
 app.layout = dbc.Container([
     dcc.Store(id='selected-doc-type', data="Shred"),
@@ -130,7 +150,7 @@ app.layout = dbc.Container([
                         },
                         multiple=True
                     ),
-                    html.Div(id='file-list'),
+                    file_list_component(),
                     html.Hr(),
                     html.Div(
                         id='doc-type-buttons'
@@ -228,10 +248,18 @@ def update_output(list_of_contents, list_of_names, existing_files):
         for i, (content, name) in enumerate(zip(list_of_contents, list_of_names)):
             file_content = process_document(content, name)
             uploaded_files[name] = file_content
-            new_files.append(html.Div([
-                html.Button('×', id={'type': 'remove-file', 'index': name}, style={'marginRight': '5px', 'fontSize': '10px'}),
-                html.Span(name)
-            ]))
+            new_files.append(
+                dbc.Row([
+                    dbc.Col(
+                        html.Button('×', id={'type': 'remove-file', 'index': name}, style={'marginRight': '5px', 'fontSize': '10px'}),
+                        width=1
+                    ),
+                    dbc.Col(
+                        html.Span(name, style={'wordBreak': 'break-all'}),
+                        width=11
+                    )
+                ], id={'type': 'file-row', 'index': name}, align="center", className="mb-1")
+            )
         if existing_files is None:
             existing_files = []
         elif not isinstance(existing_files, list):
@@ -269,8 +297,9 @@ def remove_file(n_clicks, existing_files):
     if existing_files:
         for file in existing_files:
             try:
-                file_name = file['props']['children'][1]['props']['children']
-                if file_name != removed_file:
+                # 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)
             except Exception as e:
                 filtered_files.append(file)