Update app.py via AI Editor

app.py

@@ -47,12 +47,12 @@ def get_or_create_session_data(session_id):
             SESSION_DATA[session_id] = {
                 'log_messages': [],
                 'generated_pptx': None,
-                'uploaded_files': [],  # List of dicts: {'name':..., 'content':..., 'type':...}
+                'uploaded_files': [],
                 'lock': threading.Lock(),
                 'slide_markdowns': [],
                 'slide_markdown_names': [],
                 'current_slide_md': "",
-                'files_box': [],  # List of {'name':..., 'content':..., 'type':...}
+                'files_box': [],
             }
         return SESSION_DATA[session_id]
 
@@ -113,7 +113,17 @@ app.layout = html.Div([
                             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(id="slide-generation-list-container", className="mt-3"),
+                        html.Div([
+                            dbc.Label("Generated Slide Sets", className="fw-bold"),
+                            dcc.Dropdown(
+                                id='slide-generation-list',
+                                options=[],
+                                value=None,
+                                placeholder="No slides generated yet",
+                                clearable=False,
+                                style={'marginBottom': '10px'}
+                            )
+                        ], className="mt-3"),
                         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'})
@@ -280,7 +290,8 @@ def store_session_id(n_intervals):
     Output('files-box', 'children'),
     Output('slides-content', 'value'),
     Output('log-output', 'children'),
-    Output('slide-generation-list-container', 'children'),
+    Output('slide-generation-list', 'options'),
+    Output('slide-generation-list', 'value'),
     Output('current-slide-index', 'data'),
     Output("download-dynamic", "data"),
     Input('upload-file', 'contents'),
@@ -312,11 +323,9 @@ def unified_callback(
     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 'files_box' not in session:
             session['files_box'] = []
@@ -350,22 +359,11 @@ def unified_callback(
         {'label': slide_name, 'value': idx}
         for idx, slide_name in enumerate(slide_list)
     ]
-    slide_dropdown = html.Div([
-        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")
+    dropdown_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)
 
     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:
@@ -379,12 +377,10 @@ def unified_callback(
                 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 rebuild_files_box(session), slides_content, log_str, slide_dropdown, slide_idx, None
+        return rebuild_files_box(session), slides_content, log_str, dropdown_options, dropdown_value, slide_idx, None
 
-    # Handle slide generation
     elif button_id == 'generate-button':
         decoded_file_text = None
-        # 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:
@@ -396,7 +392,7 @@ def unified_callback(
                     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 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
+                    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, dropdown_options, dropdown_value, slide_idx, None
         combined_text = ""
         if decoded_file_text and document_text:
             combined_text = decoded_file_text.strip() + "\n\n" + document_text.strip()
@@ -408,7 +404,7 @@ def unified_callback(
             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 rebuild_files_box(session), "Please upload a file or enter text.", log_str, slide_dropdown, slide_idx, None
+            return rebuild_files_box(session), "Please upload a file or enter text.", log_str, dropdown_options, dropdown_value, slide_idx, None
         try:
             add_log("Starting slide generation...", session_id)
             slides_markdown = process_document(combined_text, session_id)
@@ -421,14 +417,18 @@ def unified_callback(
                 session['current_slide_md'] = slides_markdown
                 slide_idx = len(session['slide_markdowns']) - 1
                 log_str = '\n'.join(session['log_messages'][:100])
+            dropdown_options = [
+                {'label': name, 'value': idx}
+                for idx, name in enumerate(session['slide_markdown_names'])
+            ]
+            dropdown_value = 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 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
+            return rebuild_files_box(session), f"An error occurred: {str(e)}", log_str, dropdown_options, dropdown_value, slide_idx, None
+        return rebuild_files_box(session), slides_md, log_str, dropdown_options, dropdown_value, slide_idx, None
 
-    # Handle PowerPoint generation
     elif button_id == 'generate-ppt-button':
         if not slides_content or not session_id:
             raise PreventUpdate
@@ -436,9 +436,7 @@ def unified_callback(
             add_log("Starting PowerPoint generation...", session_id)
             pptx_bytes = markdown_to_pptx(slides_content, session_id)
             with session['lock']:
-                # 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])
@@ -446,27 +444,25 @@ def unified_callback(
             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 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
+            return rebuild_files_box(session), slides_content, log_str, dropdown_options, dropdown_value, slide_idx, None
+        return rebuild_files_box(session), slides_content, log_str, dropdown_options, dropdown_value, 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):
                 slides_md = slide_md_list[selected_slide_idx]
                 slide_idx = selected_slide_idx
+                dropdown_value = slide_idx
             else:
                 slides_md = ""
         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 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
+            return rebuild_files_box(session), slides_content, log_str, dropdown_options, dropdown_value, slide_idx, None
+        return rebuild_files_box(session), slides_md, log_str, dropdown_options, dropdown_value, slide_idx, None
 
-    # 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']
@@ -474,8 +470,7 @@ def unified_callback(
                 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)
+                    return rebuild_files_box(session), slides_content, log_str, dropdown_options, dropdown_value, slide_idx, dcc.send_bytes(file_bytes, file_name)
         elif triggered_id['type'] == 'file-delete':
             idx = triggered_id['index']
             with session['lock']:
@@ -484,7 +479,7 @@ def unified_callback(
                     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
+            return rebuild_files_box(session), slides_content, log_str, dropdown_options, dropdown_value, slide_idx, None
         else:
             raise PreventUpdate