Update app.py via AI Editor

app.py

@@ -47,18 +47,19 @@ def get_or_create_session_data(session_id):
             SESSION_DATA[session_id] = {
                 'log_messages': [],
                 'generated_pptx': None,
-                'uploaded_file': None,
-                'uploaded_filename': None,
+                'uploaded_files': [],  # List of dicts: {'name':..., 'content':..., 'type':...}
                 'lock': threading.Lock(),
                 'slide_markdowns': [],
                 'slide_markdown_names': [],
                 'current_slide_md': "",
+                'files_box': [],  # List of {'name':..., 'content':..., 'type':...}
             }
         return SESSION_DATA[session_id]
 
 app.layout = html.Div([
     dcc.Store(id='session-id', storage_type='local'),
     dcc.Store(id='current-slide-index', storage_type='session'),
+    dcc.Store(id='files-box-store', storage_type='session'),
     html.Script('''
     (function() {
         function uuidv4() {
@@ -79,9 +80,6 @@ app.layout = html.Div([
         window.dash_clientside.sessionid = sessionid;
     })();
     '''),
-    html.A(id='uploaded-file-download-link', href="", style={'display': 'none'}),
-    html.A(id='pptx-file-download-link', href="", style={'display': 'none'}),
-    dcc.Dropdown(id='slide-generation-list', style={'display': 'none'}),
     dcc.Download(id="download-dynamic"),
     dbc.Container([
         dbc.Row([
@@ -110,18 +108,11 @@ app.layout = html.Div([
                             },
                             className="mb-3"
                         ),
-                        html.Div(id="upload-file-link", className="text-muted mb-2"),
+                        dbc.Card([
+                            dbc.CardHeader("All Uploaded/Generated Files", className="bg-light"),
+                            dbc.CardBody(id="files-box", className="mb-2"),
+                        ], className="mb-3 shadow-sm"),
                         dbc.Button("Generate Slides", id='generate-button', color="primary", className="w-100 mt-2", size="lg"),
-                        html.Div(
-                            dbc.Card([
-                                dbc.CardHeader("Generated PowerPoint", className="bg-light"),
-                                dbc.CardBody(
-                                    id="pptx-file-link",
-                                    className="text-muted mb-2"
-                                )
-                            ], className="mt-3 shadow-sm"),
-                            style={'width': '100%'}
-                        ),
                         html.Div(id="slide-generation-list-container", className="mt-3"),
                         html.Div([
                             html.Div("Status / Log:", className="fw-bold mb-1"),
@@ -286,47 +277,74 @@ def store_session_id(n_intervals):
     return session_id
 
 @app.callback(
-    Output('upload-file-link', 'children'),
-    Output('pptx-file-link', 'children'),
+    Output('files-box', 'children'),
     Output('slides-content', 'value'),
     Output('log-output', 'children'),
     Output('slide-generation-list-container', 'children'),
     Output('current-slide-index', 'data'),
+    Output("download-dynamic", "data"),
     Input('upload-file', 'contents'),
     Input('generate-button', 'n_clicks'),
     Input('generate-ppt-button', 'n_clicks'),
     Input('slide-generation-list', 'value'),
+    Input({'type': 'file-download', 'index': dash.ALL}, 'n_clicks'),
+    Input({'type': 'file-delete', 'index': dash.ALL}, 'n_clicks'),
     State('upload-file', 'filename'),
     State('document-text', 'value'),
     State('slides-content', 'value'),
     State('session-id', 'data'),
     State('current-slide-index', 'data'),
+    State({'type': 'file-download', 'index': dash.ALL}, 'id'),
+    State({'type': 'file-delete', 'index': dash.ALL}, 'id'),
     prevent_initial_call=True
 )
-def unified_callback(upload_contents, gen_btn, gen_ppt_btn, selected_slide_idx, upload_filename, document_text, slides_content, session_id, current_slide_idx):
+def unified_callback(
+    upload_contents, gen_btn, gen_ppt_btn, selected_slide_idx, file_downloads, file_deletes,
+    upload_filename, document_text, slides_content, session_id, current_slide_idx,
+    download_ids, delete_ids
+):
     ctx = callback_context
     if not ctx.triggered or not session_id:
         raise PreventUpdate
     button_id = ctx.triggered[0]['prop_id'].split('.')[0]
     session = get_or_create_session_data(session_id)
-    upload_file_link = ""
     slides_md = slides_content
     log_str = ""
     slide_list = session.get('slide_markdown_names', [])
     slide_md_list = session.get('slide_markdowns', [])
     slide_dropdown = None
     slide_idx = current_slide_idx if current_slide_idx is not None else None
+    download_data = None
 
+    # Ensure files_box is initialized
     with session['lock']:
-        if session['uploaded_file'] and session['uploaded_filename']:
-            upload_file_link = html.A(
-                session['uploaded_filename'],
-                id='uploaded-file-download-link-visible',
-                href="#",
-                n_clicks=0,
-                style={'display': 'block', 'cursor': 'pointer', 'color': '#116F70', 'textDecoration': 'underline'}
+        if 'files_box' not in session:
+            session['files_box'] = []
+
+    def rebuild_files_box(session):
+        files_box = []
+        for idx, f in enumerate(session['files_box']):
+            files_box.append(
+                html.Div([
+                    html.A(
+                        f['name'],
+                        href="#",
+                        id={'type': 'file-download', 'index': idx},
+                        n_clicks=0,
+                        style={'marginRight': '16px', 'color': '#116F70', 'textDecoration': 'underline'}
+                    ),
+                    dbc.Button(
+                        "Delete",
+                        id={'type': 'file-delete', 'index': idx},
+                        size="sm",
+                        color="secondary",
+                        style={'marginLeft': '4px', 'marginBottom': '2px'}
+                    )
+                ], style={'display': 'flex', 'alignItems': 'center', 'marginBottom': '8px'}, id={'type': 'file-row', 'index': idx})
             )
-        log_str = '\n'.join(session['log_messages'][:100])
+        if not files_box:
+            return html.Div("No files uploaded or generated yet.", className="text-muted")
+        return files_box
 
     dropdown_options = [
         {'label': slide_name, 'value': idx}
@@ -344,52 +362,41 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, selected_slide_idx,
         )
     ]) if slide_list else html.Div("No slides generated yet.", className="text-muted")
 
-    pptx_file_link = ""
-    if session.get('generated_pptx'):
-        pptx_file_link = html.A(
-            "presentation.pptx",
-            id='pptx-file-download-link-visible',
-            href="#",
-            n_clicks=0,
-            style={'display': 'block', 'cursor': 'pointer', 'color': '#116F70', 'textDecoration': 'underline'}
-        )
+    with session['lock']:
+        log_str = '\n'.join(session['log_messages'][:100])
 
+    # Handle file upload
     if button_id == 'upload-file':
         if upload_contents and upload_filename:
             try:
                 header, content_string = upload_contents.split(',')
                 file_bytes = base64.b64decode(content_string)
-                session = get_or_create_session_data(session_id)
                 with session['lock']:
-                    session['uploaded_file'] = file_bytes
-                    session['uploaded_filename'] = upload_filename
+                    session['files_box'].append({'name': upload_filename, 'content': file_bytes, 'type': 'uploaded'})
                     session['log_messages'].insert(0, f"{time.strftime('%H:%M:%S')} - Uploaded file '{upload_filename}' saved for session.")
-                    upload_file_link = html.A(
-                        upload_filename,
-                        id='uploaded-file-download-link-visible',
-                        href="#",
-                        n_clicks=0,
-                        style={'display': 'block', 'cursor': 'pointer', 'color': '#116F70', 'textDecoration': 'underline'}
-                    )
                     log_str = '\n'.join(session['log_messages'][:100])
             except Exception as e:
                 with session['lock']:
                     session['log_messages'].insert(0, f"{time.strftime('%H:%M:%S')} - Error uploading file: {str(e)}")
                     log_str = '\n'.join(session['log_messages'][:100])
-        return upload_file_link, pptx_file_link, slides_content, log_str, slide_dropdown, slide_idx
+        return rebuild_files_box(session), slides_content, log_str, slide_dropdown, slide_idx, None
+
+    # Handle slide generation
     elif button_id == 'generate-button':
         decoded_file_text = None
-        if session.get('uploaded_file'):
+        # Use first uploaded file for text extraction if present
+        uploaded_files = [f for f in session['files_box'] if f['type'] == 'uploaded']
+        if uploaded_files:
             try:
-                decoded_file_text = session['uploaded_file'].decode('utf-8')
+                decoded_file_text = uploaded_files[0]['content'].decode('utf-8')
             except UnicodeDecodeError:
                 try:
-                    decoded_file_text = session['uploaded_file'].decode('latin-1')
+                    decoded_file_text = uploaded_files[0]['content'].decode('latin-1')
                 except Exception:
                     with session['lock']:
                         session['log_messages'].insert(0, f"{time.strftime('%H:%M:%S')} - Could not decode uploaded file as text. Please upload a UTF-8 or Latin-1 encoded text file.")
                         log_str = '\n'.join(session['log_messages'][:100])
-                    return upload_file_link, pptx_file_link, "Could not decode uploaded file as text. Please upload a UTF-8 or Latin-1 encoded text file.", log_str, slide_dropdown, slide_idx
+                    return rebuild_files_box(session), "Could not decode uploaded file as text. Please upload a UTF-8 or Latin-1 encoded text file.", log_str, slide_dropdown, slide_idx, None
         combined_text = ""
         if decoded_file_text and document_text:
             combined_text = decoded_file_text.strip() + "\n\n" + document_text.strip()
@@ -401,7 +408,7 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, selected_slide_idx,
             with session['lock']:
                 session['log_messages'].insert(0, f"{time.strftime('%H:%M:%S')} - Please upload a file or enter text.")
                 log_str = '\n'.join(session['log_messages'][:100])
-            return upload_file_link, pptx_file_link, "Please upload a file or enter text.", log_str, slide_dropdown, slide_idx
+            return rebuild_files_box(session), "Please upload a file or enter text.", log_str, slide_dropdown, slide_idx, None
         try:
             add_log("Starting slide generation...", session_id)
             slides_markdown = process_document(combined_text, session_id)
@@ -414,27 +421,14 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, selected_slide_idx,
                 session['current_slide_md'] = slides_markdown
                 slide_idx = len(session['slide_markdowns']) - 1
                 log_str = '\n'.join(session['log_messages'][:100])
-                dropdown_options = [
-                    {'label': n, 'value': idx}
-                    for idx, n in enumerate(session['slide_markdown_names'])
-                ]
-                slide_dropdown = html.Div([
-                    dbc.Label("Generated Slide Sets", className="fw-bold"),
-                    dcc.Dropdown(
-                        id='slide-generation-list',
-                        options=dropdown_options,
-                        value=slide_idx,
-                        placeholder="No slides generated yet",
-                        clearable=False,
-                        style={'marginBottom': '10px'}
-                    )
-                ])
-            return upload_file_link, pptx_file_link, slides_md, log_str, slide_dropdown, slide_idx
         except Exception as e:
             add_log(f"An error occurred during slide generation: {str(e)}", session_id)
             with session['lock']:
                 log_str = '\n'.join(session['log_messages'][:100])
-            return upload_file_link, pptx_file_link, f"An error occurred: {str(e)}", log_str, slide_dropdown, slide_idx
+            return rebuild_files_box(session), f"An error occurred: {str(e)}", log_str, slide_dropdown, slide_idx, None
+        return rebuild_files_box(session), slides_md, log_str, slide_dropdown, slide_idx, None
+
+    # Handle PowerPoint generation
     elif button_id == 'generate-ppt-button':
         if not slides_content or not session_id:
             raise PreventUpdate
@@ -442,22 +436,20 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, selected_slide_idx,
             add_log("Starting PowerPoint generation...", session_id)
             pptx_bytes = markdown_to_pptx(slides_content, session_id)
             with session['lock']:
-                session['generated_pptx'] = pptx_bytes
-                pptx_file_link = html.A(
-                    "presentation.pptx",
-                    id='pptx-file-download-link-visible',
-                    href="#",
-                    n_clicks=0,
-                    style={'display': 'block', 'cursor': 'pointer', 'color': '#116F70', 'textDecoration': 'underline'}
-                )
+                # Remove previous generated pptx files for this session
+                session['files_box'] = [f for f in session['files_box'] if f['type'] != 'generated_pptx']
+                # Add new pptx file
+                session['files_box'].append({'name': 'presentation.pptx', 'content': pptx_bytes, 'type': 'generated_pptx'})
                 add_log("PowerPoint generation completed.", session_id)
                 log_str = '\n'.join(session['log_messages'][:100])
-            return upload_file_link, pptx_file_link, slides_content, log_str, slide_dropdown, slide_idx
         except Exception as e:
             add_log(f"An error occurred during PowerPoint generation: {str(e)}", session_id)
             with session['lock']:
                 log_str = '\n'.join(session['log_messages'][:100])
-            return upload_file_link, pptx_file_link, slides_content, log_str, slide_dropdown, slide_idx
+            return rebuild_files_box(session), slides_content, log_str, slide_dropdown, slide_idx, None
+        return rebuild_files_box(session), slides_content, log_str, slide_dropdown, slide_idx, None
+
+    # Handle slide set dropdown selection
     elif button_id == 'slide-generation-list':
         try:
             if selected_slide_idx is not None and 0 <= selected_slide_idx < len(slide_md_list):
@@ -465,34 +457,39 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, selected_slide_idx,
                 slide_idx = selected_slide_idx
             else:
                 slides_md = ""
-            return upload_file_link, pptx_file_link, slides_md, log_str, slide_dropdown, slide_idx
         except Exception as e:
             add_log(f"An error occurred while loading slide set: {str(e)}", session_id)
             with session['lock']:
                 log_str = '\n'.join(session['log_messages'][:100])
-            return upload_file_link, pptx_file_link, slides_content, log_str, slide_dropdown, slide_idx
-    else:
-        raise PreventUpdate
+            return rebuild_files_box(session), slides_content, log_str, slide_dropdown, slide_idx, None
+        return rebuild_files_box(session), slides_md, log_str, slide_dropdown, slide_idx, None
 
-@app.callback(
-    Output("download-dynamic", "data"),
-    Input('uploaded-file-download-link-visible', 'n_clicks'),
-    Input('pptx-file-download-link-visible', 'n_clicks'),
-    State('session-id', 'data'),
-    prevent_initial_call=True
-)
-def download_dynamic(uploaded_file_nclicks, pptx_nclicks, session_id):
-    ctx = callback_context
-    if not ctx.triggered or not session_id:
+    # Handle file download and delete
+    elif button_id.startswith("{"):
+        # Determine which pattern (download or delete) triggered
+        triggered_id = eval(button_id)
+        if triggered_id['type'] == 'file-download':
+            idx = triggered_id['index']
+            with session['lock']:
+                if 0 <= idx < len(session['files_box']):
+                    file_bytes = session['files_box'][idx]['content']
+                    file_name = session['files_box'][idx]['name']
+                    # No other output changes, just return download
+                    return rebuild_files_box(session), slides_content, log_str, slide_dropdown, slide_idx, dcc.send_bytes(file_bytes, file_name)
+        elif triggered_id['type'] == 'file-delete':
+            idx = triggered_id['index']
+            with session['lock']:
+                if 0 <= idx < len(session['files_box']):
+                    deleted_name = session['files_box'][idx]['name']
+                    del session['files_box'][idx]
+                    session['log_messages'].insert(0, f"{time.strftime('%H:%M:%S')} - File '{deleted_name}' deleted.")
+                    log_str = '\n'.join(session['log_messages'][:100])
+            return rebuild_files_box(session), slides_content, log_str, slide_dropdown, slide_idx, None
+        else:
+            raise PreventUpdate
+
+    else:
         raise PreventUpdate
-    button_id = ctx.triggered[0]['prop_id'].split('.')[0]
-    session = get_or_create_session_data(session_id)
-    with session['lock']:
-        if button_id == 'uploaded-file-download-link-visible' and session.get('uploaded_file') and session.get('uploaded_filename'):
-            return dcc.send_bytes(session['uploaded_file'], session['uploaded_filename'])
-        if button_id == 'pptx-file-download-link-visible' and session.get('generated_pptx'):
-            return dcc.send_bytes(session['generated_pptx'], "presentation.pptx")
-    raise PreventUpdate
 
 if __name__ == '__main__':
     print("Starting the Dash application...")