Update app.py via AI Editor

app.py

@@ -80,6 +80,7 @@ app.layout = html.Div([
     })();
     '''),
     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([
@@ -272,17 +273,6 @@ def markdown_to_pptx(md_text, session_id):
     prs.save(output)
     return output.getvalue()
 
-def get_pptx_download_button(session):
-    if session.get('generated_pptx'):
-        return dbc.Button(
-            "Download PowerPoint",
-            id='pptx-download-link',
-            color="primary",
-            className="w-100",
-            n_clicks=0
-        )
-    return ""
-
 @app.callback(
     Output('session-id', 'data'),
     Input('session-id', 'data')
@@ -334,7 +324,7 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, selected_slide_idx,
                 id='uploaded-file-download-link-visible',
                 href="#",
                 n_clicks=0,
-                style={'display': 'block'}
+                style={'display': 'block', 'cursor': 'pointer', 'color': '#116F70', 'textDecoration': 'underline'}
             )
         log_str = '\n'.join(session['log_messages'][:100])
 
@@ -354,7 +344,15 @@ 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 = get_pptx_download_button(session)
+    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'}
+        )
 
     if button_id == 'upload-file':
         if upload_contents and upload_filename:
@@ -371,14 +369,13 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, selected_slide_idx,
                         id='uploaded-file-download-link-visible',
                         href="#",
                         n_clicks=0,
-                        style={'display': 'block'}
+                        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])
-        pptx_file_link = get_pptx_download_button(session)
         return upload_file_link, pptx_file_link, slides_content, log_str, slide_dropdown, slide_idx
     elif button_id == 'generate-button':
         decoded_file_text = None
@@ -392,7 +389,6 @@ 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')} - 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])
-                    pptx_file_link = get_pptx_download_button(session)
                     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:
@@ -405,7 +401,6 @@ 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])
-            pptx_file_link = get_pptx_download_button(session)
             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)
@@ -434,13 +429,11 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, selected_slide_idx,
                         style={'marginBottom': '10px'}
                     )
                 ])
-            pptx_file_link = get_pptx_download_button(session)
             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])
-            pptx_file_link = get_pptx_download_button(session)
             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:
@@ -450,7 +443,13 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, selected_slide_idx,
             pptx_bytes = markdown_to_pptx(slides_content, session_id)
             with session['lock']:
                 session['generated_pptx'] = pptx_bytes
-                pptx_file_link = get_pptx_download_button(session)
+                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'}
+                )
                 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
@@ -458,7 +457,6 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, selected_slide_idx,
             add_log(f"An error occurred during PowerPoint generation: {str(e)}", session_id)
             with session['lock']:
                 log_str = '\n'.join(session['log_messages'][:100])
-            pptx_file_link = get_pptx_download_button(session)
             return upload_file_link, pptx_file_link, slides_content, log_str, slide_dropdown, slide_idx
     elif button_id == 'slide-generation-list':
         try:
@@ -467,21 +465,19 @@ def unified_callback(upload_contents, gen_btn, gen_ppt_btn, selected_slide_idx,
                 slide_idx = selected_slide_idx
             else:
                 slides_md = ""
-            pptx_file_link = get_pptx_download_button(session)
             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])
-            pptx_file_link = get_pptx_download_button(session)
             return upload_file_link, pptx_file_link, slides_content, log_str, slide_dropdown, slide_idx
     else:
         raise PreventUpdate
 
 @app.callback(
     Output("download-dynamic", "data"),
-    Input('uploaded-file-download-link', 'n_clicks'),
-    Input('pptx-download-link', 'n_clicks'),
+    Input('uploaded-file-download-link-visible', 'n_clicks'),
+    Input('pptx-file-download-link-visible', 'n_clicks'),
     State('session-id', 'data'),
     prevent_initial_call=True
 )
@@ -492,9 +488,9 @@ def download_dynamic(uploaded_file_nclicks, pptx_nclicks, session_id):
     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' and session.get('uploaded_file') and session.get('uploaded_filename'):
+        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-download-link' and session.get('generated_pptx'):
+        if button_id == 'pptx-file-download-link-visible' and session.get('generated_pptx'):
             return dcc.send_bytes(session['generated_pptx'], "presentation.pptx")
     raise PreventUpdate