Update app.py via AI Editor

app.py

@@ -286,85 +286,83 @@ def update_filename(filename):
     Output('download-ppt-button', 'disabled'),
     Output('log-output', 'children'),
     Input('generate-button', 'n_clicks'),
+    Input('generate-ppt-button', 'n_clicks'),
     State('upload-file', 'contents'),
     State('document-text', 'value'),
-    State('session-id', 'data')
-)
-def generate_slides(n_clicks, file_contents, document_text, session_id):
-    triggered = callback_context.triggered[0]['prop_id'].split('.')[0]
-    if n_clicks is None:
-        raise PreventUpdate
-    if not session_id:
-        raise PreventUpdate
-    decoded_file_text = None
-    if file_contents:
-        content_type, content_string = file_contents.split(',')
-        decoded = base64.b64decode(content_string)
-        try:
-            decoded_file_text = decoded.decode('utf-8')
-        except UnicodeDecodeError:
-            try:
-                decoded_file_text = decoded.decode('latin-1')
-            except Exception:
-                session = get_or_create_session_data(session_id)
-                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 "Could not decode uploaded file as text. Please upload a UTF-8 or Latin-1 encoded text file.", True, log_str
-    combined_text = ""
-    if decoded_file_text and document_text:
-        combined_text = decoded_file_text.strip() + "\n\n" + document_text.strip()
-    elif decoded_file_text:
-        combined_text = decoded_file_text.strip()
-    elif document_text:
-        combined_text = document_text.strip()
-    else:
-        session = get_or_create_session_data(session_id)
-        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 "Please upload a file or enter text.", True, log_str
-
-    try:
-        add_log("Starting slide generation...", session_id)
-        slides_markdown = process_document(combined_text, session_id)
-        add_log("Slide generation completed.", session_id)
-        session = get_or_create_session_data(session_id)
-        with session['lock']:
-            log_str = '\n'.join(session['log_messages'][:100])
-        return slides_markdown, False, log_str
-    except Exception as e:
-        add_log(f"An error occurred during slide generation: {str(e)}", session_id)
-        session = get_or_create_session_data(session_id)
-        with session['lock']:
-            log_str = '\n'.join(session['log_messages'][:100])
-        return f"An error occurred: {str(e)}", True, log_str
-
-@app.callback(
-    Output('log-output', 'children'),
-    Output('download-ppt-button', 'disabled', allow_duplicate=True),
-    Input('generate-ppt-button', 'n_clicks'),
     State('slides-content', 'value'),
     State('session-id', 'data'),
     prevent_initial_call=True
 )
-def generate_powerpoint(n_clicks, slides_content, session_id):
-    if n_clicks is None or not slides_content or not session_id:
+def unified_callback(gen_btn, gen_ppt_btn, file_contents, document_text, slides_content, session_id):
+    ctx = callback_context
+    if not ctx.triggered or not session_id:
         raise PreventUpdate
-    session = get_or_create_session_data(session_id)
-    with session['lock']:
+    button_id = ctx.triggered[0]['prop_id'].split('.')[0]
+    # Slide generation logic
+    if button_id == 'generate-button':
+        decoded_file_text = None
+        if file_contents:
+            content_type, content_string = file_contents.split(',')
+            decoded = base64.b64decode(content_string)
+            try:
+                decoded_file_text = decoded.decode('utf-8')
+            except UnicodeDecodeError:
+                try:
+                    decoded_file_text = decoded.decode('latin-1')
+                except Exception:
+                    session = get_or_create_session_data(session_id)
+                    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 "Could not decode uploaded file as text. Please upload a UTF-8 or Latin-1 encoded text file.", True, log_str
+        combined_text = ""
+        if decoded_file_text and document_text:
+            combined_text = decoded_file_text.strip() + "\n\n" + document_text.strip()
+        elif decoded_file_text:
+            combined_text = decoded_file_text.strip()
+        elif document_text:
+            combined_text = document_text.strip()
+        else:
+            session = get_or_create_session_data(session_id)
+            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 "Please upload a file or enter text.", True, log_str
+
         try:
-            add_log("Starting PowerPoint generation...", session_id)
-            pptx_bytes = markdown_to_pptx(slides_content, session_id)
-            session['generated_pptx'] = pptx_bytes
-            add_log("PowerPoint generation completed.", session_id)
-            log_str = '\n'.join(session['log_messages'][:100])
-            return log_str, False
+            add_log("Starting slide generation...", session_id)
+            slides_markdown = process_document(combined_text, session_id)
+            add_log("Slide generation completed.", session_id)
+            session = get_or_create_session_data(session_id)
+            with session['lock']:
+                log_str = '\n'.join(session['log_messages'][:100])
+            return slides_markdown, False, log_str
         except Exception as e:
-            error_message = f"An error occurred during PowerPoint generation: {str(e)}"
-            add_log(error_message, session_id)
-            log_str = '\n'.join(session['log_messages'][:100])
-            return log_str, True
+            add_log(f"An error occurred during slide generation: {str(e)}", session_id)
+            session = get_or_create_session_data(session_id)
+            with session['lock']:
+                log_str = '\n'.join(session['log_messages'][:100])
+            return f"An error occurred: {str(e)}", True, log_str
+    # PowerPoint generation logic
+    elif button_id == 'generate-ppt-button':
+        if gen_ppt_btn is None or not slides_content or not session_id:
+            raise PreventUpdate
+        session = get_or_create_session_data(session_id)
+        with session['lock']:
+            try:
+                add_log("Starting PowerPoint generation...", session_id)
+                pptx_bytes = markdown_to_pptx(slides_content, session_id)
+                session['generated_pptx'] = pptx_bytes
+                add_log("PowerPoint generation completed.", session_id)
+                log_str = '\n'.join(session['log_messages'][:100])
+                return slides_content, False, log_str
+            except Exception as e:
+                error_message = f"An error occurred during PowerPoint generation: {str(e)}"
+                add_log(error_message, session_id)
+                log_str = '\n'.join(session['log_messages'][:100])
+                return slides_content, True, log_str
+    else:
+        raise PreventUpdate
 
 @app.callback(
     Output("download-pptx", "data"),