Update app.py via AI Editor

app.py

@@ -49,12 +49,16 @@ def get_or_create_session_data(session_id):
                 'generated_pptx': None,
                 'uploaded_file': None,
                 'uploaded_filename': None,
-                'lock': threading.Lock()
+                'lock': threading.Lock(),
+                'slide_markdowns': [],  # store generated slide markdowns as list of dicts
+                'slide_markdown_names': [],  # Friendly names for dropdown
+                'current_slide_md': "",
             }
         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'),
     html.Script('''
     (function() {
         function uuidv4() {
@@ -75,7 +79,6 @@ app.layout = html.Div([
         window.dash_clientside.sessionid = sessionid;
     })();
     '''),
-    # Invisible persistent download links for Dash callback tracking
     html.A(id='uploaded-file-download-link', href="", style={'display': 'none'}),
     html.A(id='pptx-download-link', href="", style={'display': 'none'}),
     dbc.Container([
@@ -87,7 +90,7 @@ app.layout = html.Div([
                         dbc.Textarea(
                             id='document-text',
                             placeholder='Enter or paste your document here',
-                            style={'height': '200px'},
+                            style={'height': '80px', 'resize': 'vertical'},
                             className="mb-3"
                         ),
                         dcc.Upload(
@@ -105,12 +108,17 @@ app.layout = html.Div([
                             },
                             className="mb-3"
                         ),
-                        html.Div(id="upload-file-link", className="text-muted"),
-                        html.Div(id="pptx-file-link", className="text-muted"),
-                        dbc.Button("Generate Slides", id='generate-button', color="primary", className="w-100 mt-3", size="lg")
+                        html.Div(id="upload-file-link", className="text-muted mb-2"),
+                        dbc.Button("Generate Slides", id='generate-button', color="primary", className="w-100 mt-2", size="lg"),
+                        html.Div(id="slide-generation-list-container", className="mt-3"),
+                        html.Div(id="pptx-file-link", className="text-muted mb-2"),
+                        html.Div([
+                            html.Div("Status / Log:", className="fw-bold mb-1"),
+                            html.Div(id='log-output', style={'whiteSpace': 'pre-line', 'height': '150px', 'overflowY': 'scroll', 'border': '1px solid #ddd', 'padding': '10px'})
+                        ], className="mt-3"),
                     ])
                 ], className="mb-4 shadow-sm")
-            ], md=6),
+            ], width=4, xs=12, sm=12, md=4, lg=4, xl=4, xxl=4),
             dbc.Col([
                 dbc.Card([
                     dbc.CardHeader("Edit Slide Content", className="bg-light"),
@@ -118,7 +126,7 @@ app.layout = html.Div([
                         dbc.Textarea(
                             id='slides-content',
                             placeholder='Generated or pasted markdown will appear here. You can edit the content.',
-                            style={'height': '400px'},
+                            style={'height': '400px', 'resize': 'vertical'},
                             className="mb-3"
                         ),
                         dcc.Loading(
@@ -129,10 +137,9 @@ app.layout = html.Div([
                             ],
                             style={'width': '100%'}
                         ),
-                        html.Div(id='log-output', style={'whiteSpace': 'pre-line', 'height': '200px', 'overflowY': 'scroll', 'border': '1px solid #ddd', 'padding': '10px', 'marginTop': '10px'})
                     ])
                 ], className="mb-4 shadow-sm")
-            ], md=6)
+            ], width=8, xs=12, sm=12, md=8, lg=8, xl=8, xxl=8)
         ])
     ], fluid=True, className="mt-4"),
     dcc.Download(id="download-dynamic")
@@ -273,16 +280,20 @@ def store_session_id(n_intervals):
     Output('pptx-file-link', 'children'),
     Output('slides-content', 'value'),
     Output('log-output', 'children'),
+    Output('slide-generation-list-container', 'children'),
+    Output('current-slide-index', 'data'),
     Input('upload-file', 'contents'),
     Input('generate-button', 'n_clicks'),
     Input('generate-ppt-button', 'n_clicks'),
+    Input('slide-generation-list', 'value'),
     State('upload-file', 'filename'),
     State('document-text', 'value'),
     State('slides-content', 'value'),
     State('session-id', 'data'),
+    State('current-slide-index', 'data'),
     prevent_initial_call=True
 )
-def unified_callback(upload_contents, gen_btn, gen_ppt_btn, upload_filename, document_text, slides_content, session_id):
+def unified_callback(upload_contents, gen_btn, gen_ppt_btn, selected_slide_idx, upload_filename, document_text, slides_content, session_id, current_slide_idx):
     ctx = callback_context
     if not ctx.triggered or not session_id:
         raise PreventUpdate
@@ -292,6 +303,10 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, upload_filename, doc
     pptx_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
     with session['lock']:
         if session['uploaded_file'] and session['uploaded_filename']:
             upload_file_link = html.A(
@@ -310,6 +325,25 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, upload_filename, doc
                 style={'display': 'block'}
             )
         log_str = '\n'.join(session['log_messages'][:100])
+
+    # Build slide markdown dropdown
+    dropdown_options = [
+        {'label': slide_name, 'value': idx}
+        for idx, slide_name in enumerate(slide_list)
+    ]
+    slide_dropdown = dbc.FormGroup([
+        dbc.Label("Generated Slide Sets", className="fw-bold"),
+        dcc.Dropdown(
+            id='slide-generation-list',
+            options=dropdown_options,
+            value=slide_idx if (slide_idx is not None and slide_idx < len(slide_list)) else (len(slide_list) - 1 if slide_list else None),
+            placeholder="No slides generated yet",
+            clearable=False,
+            style={'marginBottom': '10px'}
+        )
+    ]) if slide_list else html.Div("No slides generated yet.", className="text-muted")
+
+    # Handle triggers
     if button_id == 'upload-file':
         if upload_contents and upload_filename:
             try:
@@ -332,7 +366,7 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, upload_filename, doc
                 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
+        return upload_file_link, pptx_file_link, slides_content, log_str, slide_dropdown, slide_idx
     elif button_id == 'generate-button':
         decoded_file_text = None
         if session.get('uploaded_file'):
@@ -345,7 +379,7 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, upload_filename, doc
                     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
+                    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
         combined_text = ""
         if decoded_file_text and document_text:
             combined_text = decoded_file_text.strip() + "\n\n" + document_text.strip()
@@ -357,20 +391,41 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, upload_filename, doc
             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
+            return upload_file_link, pptx_file_link, "Please upload a file or enter text.", log_str, slide_dropdown, slide_idx
         try:
             add_log("Starting slide generation...", session_id)
             slides_markdown = process_document(combined_text, session_id)
             add_log("Slide generation completed.", session_id)
             with session['lock']:
+                # Add as new slide set to session
+                slide_name = f"Slides {time.strftime('%Y-%m-%d %H:%M:%S')}"
+                session['slide_markdowns'].append(slides_markdown)
+                session['slide_markdown_names'].append(slide_name)
                 slides_md = slides_markdown
+                session['current_slide_md'] = slides_markdown
+                slide_idx = len(session['slide_markdowns']) - 1
                 log_str = '\n'.join(session['log_messages'][:100])
-            return upload_file_link, pptx_file_link, slides_md, log_str
+                dropdown_options = [
+                    {'label': n, 'value': idx}
+                    for idx, n in enumerate(session['slide_markdown_names'])
+                ]
+                slide_dropdown = dbc.FormGroup([
+                    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
+            return upload_file_link, pptx_file_link, f"An error occurred: {str(e)}", log_str, slide_dropdown, slide_idx
     elif button_id == 'generate-ppt-button':
         if not slides_content or not session_id:
             raise PreventUpdate
@@ -388,12 +443,25 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, upload_filename, doc
                 )
                 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
+            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
+            return upload_file_link, pptx_file_link, slides_content, log_str, slide_dropdown, slide_idx
+    elif button_id == 'slide-generation-list':
+        try:
+            if selected_slide_idx is not None and 0 <= selected_slide_idx < len(slide_md_list):
+                slides_md = slide_md_list[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