Update app.py via AI Editor

app.py

@@ -1,24 +1,23 @@
 import dash
 from dash import dcc, html, Input, Output, State, no_update
 import dash_bootstrap_components as dbc
+import os
 import base64
 import io
-import re
-import os
-import requests
 import logging
+import uuid
+import threading
+import time
+import re
 from pptx import Presentation
 from pptx.util import Inches, Pt
-from concurrent.futures import ThreadPoolExecutor
-from dash.exceptions import PreventUpdate
-from PIL import Image
-import google.generativeai as genai
-import time
 import docx
 import PyPDF2
+from PIL import Image
+from concurrent.futures import ThreadPoolExecutor
+import requests
 from flask import request as flask_request, after_this_request
-import threading
-import uuid
+import google.generativeai as genai
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
@@ -29,14 +28,15 @@ genai.configure(api_key=GOOGLE_API_KEY)
 model = genai.GenerativeModel('gemini-2.5-pro-preview-03-25')
 
 app = dash.Dash(__name__, external_stylesheets=[
-    dbc.themes.BOOTSTRAP, 
+    dbc.themes.BOOTSTRAP,
     'https://fonts.googleapis.com/css2?family=Roboto:wght@300;400;700&display=swap'
 ])
 app.config.suppress_callback_exceptions = True
 server = app.server
 
-SESSION_LOCK = threading.Lock()
 SESSION_DATA = {}
+SESSION_LOCK = threading.Lock()
+SESSION_THREAD_LOCKS = {}
 
 def get_session_id():
     sid = flask_request.cookies.get('session_id')
@@ -62,6 +62,8 @@ def get_session_dict():
                 "generated_pptx": None,
                 "ppt_ready": False,
             }
+        if sid not in SESSION_THREAD_LOCKS:
+            SESSION_THREAD_LOCKS[sid] = threading.Lock()
         return SESSION_DATA[sid]
 
 def add_log(message):
@@ -121,7 +123,7 @@ def process_document(text):
     Generate slides with titles and content. Format the output as markdown, 
     with each slide starting with '# Slide X:' where X is the slide number. 
     Use '##' for subtitles and '-' for bullet points.  Provide only the 
-    slides in markdown and nothing else, to intro, no acknowlegements, no outro, no summary, just the powerpoint.
+    slides in markdown and nothing else, to intro, no acknowledgements, no outro, no summary, just the powerpoint.
     do not use any other markdown like * or **
     There can only be one slide heading # and one sub heading ## and no more than 5 bullets with "-" per slide
     Document:
@@ -161,7 +163,7 @@ def generate_image(prompt):
     }
     files = {"none": ''}
     try:
-        response = requests.post(url, headers=headers, data=data, files=files)
+        response = requests.post(url, headers=headers, data=data, files=files, timeout=60)
         response.raise_for_status()
         add_log("Image generated successfully.")
         return response.content
@@ -174,55 +176,60 @@ def markdown_to_pptx(md_text):
     prs = Presentation()
     slides = re.split(r'# Slide \d+:', md_text)
     slides = [slide.strip() for slide in slides if slide.strip()]
-    with ThreadPoolExecutor() as executor:
-        futures = []
-        for i, slide_content in enumerate(slides):
-            add_log(f"Processing slide {i+1}/{len(slides)}...")
-            image_prompt = generate_image_prompt(slide_content)
-            futures.append(executor.submit(generate_image, image_prompt))
-        for i, (slide_content, future) in enumerate(zip(slides, futures)):
-            add_log(f"Creating slide {i+1}/{len(slides)}...")
-            lines = slide_content.split('\n')
-            title = lines[0].strip()
-            current_slide = prs.slides.add_slide(prs.slide_layouts[6])
-            title_box = current_slide.shapes.add_textbox(Inches(0.5), Inches(0.5), Inches(9), Inches(1))
-            title_box.text_frame.text = title
-            title_box.text_frame.paragraphs[0].font.size = Pt(32)
-            title_box.text_frame.paragraphs[0].font.bold = True
-            content_box = current_slide.shapes.add_textbox(Inches(0.5), Inches(1.5), Inches(4.5), Inches(6))
-            text_frame = content_box.text_frame
-            text_frame.word_wrap = True
-            for line in lines[1:]:
-                if line.startswith('## '):
-                    p = text_frame.add_paragraph()
-                    p.text = line[3:].strip()
-                    p.font.size = Pt(24)
-                    p.font.bold = True
-                elif line.startswith('- '):
-                    p = text_frame.add_paragraph()
-                    p.text = line[2:].strip()
-                    p.font.size = Pt(18)
-                    p.level = 1
-                    p.bullet = True
-            add_log(f"Adding image to slide {i+1}...")
-            image_data = future.result()
-            if image_data:
-                image_stream = io.BytesIO(image_data)
-                try:
-                    img = Image.open(image_stream)
-                    aspect_ratio = img.width / img.height
-                    img_width = Inches(4)
-                    img_height = img_width / aspect_ratio
-                    current_slide.shapes.add_picture(image_stream, Inches(5.5), Inches(1.5), width=img_width, height=img_height)
-                    add_log(f"Image added successfully to slide {i+1}.")
-                except Exception as e:
-                    add_log(f"Error adding image to slide {i+1}: {str(e)}")
+    session_dict = get_session_dict()
+    sid = get_session_id()
+    thread_lock = SESSION_THREAD_LOCKS[sid]
+    with thread_lock:
+        with ThreadPoolExecutor(max_workers=2) as executor:
+            futures = []
+            for i, slide_content in enumerate(slides):
+                add_log(f"Processing slide {i+1}/{len(slides)}...")
+                image_prompt = generate_image_prompt(slide_content)
+                futures.append(executor.submit(generate_image, image_prompt))
+            for i, (slide_content, future) in enumerate(zip(slides, futures)):
+                add_log(f"Creating slide {i+1}/{len(slides)}...")
+                lines = slide_content.split('\n')
+                title = lines[0].strip() if lines else f"Slide {i+1}"
+                current_slide = prs.slides.add_slide(prs.slide_layouts[6])
+                title_box = current_slide.shapes.add_textbox(Inches(0.5), Inches(0.5), Inches(9), Inches(1))
+                title_box.text_frame.text = title
+                title_box.text_frame.paragraphs[0].font.size = Pt(32)
+                title_box.text_frame.paragraphs[0].font.bold = True
+                content_box = current_slide.shapes.add_textbox(Inches(0.5), Inches(1.5), Inches(4.5), Inches(6))
+                text_frame = content_box.text_frame
+                text_frame.word_wrap = True
+                for line in lines[1:]:
+                    if line.startswith('## '):
+                        p = text_frame.add_paragraph()
+                        p.text = line[3:].strip()
+                        p.font.size = Pt(24)
+                        p.font.bold = True
+                    elif line.startswith('- '):
+                        p = text_frame.add_paragraph()
+                        p.text = line[2:].strip()
+                        p.font.size = Pt(18)
+                        p.level = 1
+                        p.bullet = True
+                add_log(f"Adding image to slide {i+1}...")
+                image_data = future.result()
+                if image_data:
+                    image_stream = io.BytesIO(image_data)
+                    try:
+                        img = Image.open(image_stream)
+                        aspect_ratio = img.width / img.height
+                        img_width = Inches(4)
+                        img_height = img_width / aspect_ratio
+                        image_stream.seek(0)
+                        current_slide.shapes.add_picture(image_stream, Inches(5.5), Inches(1.5), width=img_width, height=img_height)
+                        add_log(f"Image added successfully to slide {i+1}.")
+                    except Exception as e:
+                        add_log(f"Error adding image to slide {i+1}: {str(e)}")
+                        placeholder = current_slide.shapes.add_textbox(Inches(5.5), Inches(1.5), Inches(4), Inches(6))
+                        placeholder.text_frame.text = "Image addition failed"
+                else:
+                    add_log(f"Failed to generate image for slide {i+1}: {title}")
                     placeholder = current_slide.shapes.add_textbox(Inches(5.5), Inches(1.5), Inches(4), Inches(6))
-                    placeholder.text_frame.text = "Image addition failed"
-            else:
-                add_log(f"Failed to generate image for slide {i+1}: {title}")
-                placeholder = current_slide.shapes.add_textbox(Inches(5.5), Inches(1.5), Inches(4), Inches(6))
-                placeholder.text_frame.text = "Image generation failed"
+                    placeholder.text_frame.text = "Image generation failed"
     add_log("PowerPoint generation completed.")
     output = io.BytesIO()
     prs.save(output)
@@ -263,65 +270,84 @@ def get_pptx_download_area(session_dict):
         pptx_link
     ]) if pptx_link else ""
 
-app.layout = html.Div([
-    dcc.Store(id='session-init', storage_type='session'),
-    dcc.Store(id='loading-state', storage_type='session'),
-    html.Div([
-        dbc.Button("Delete", id='delete-file-btn', color="danger", size="sm", n_clicks=0, style={"display": "none"}),
-        dbc.Button("Delete", id='delete-pptx-btn', color="danger", size="sm", n_clicks=0, style={"display": "none"}),
-    ], style={"display": "none"}),
-    dbc.Container([
-        dbc.Row([
-            dbc.Col([
-                dbc.Card([
-                    dbc.CardHeader("Upload or Paste Document", className="bg-light"),
-                    dbc.CardBody([
-                        dbc.Textarea(
-                            id='document-text',
-                            placeholder='Enter or paste your document here',
-                            style={'height': '200px', 'whiteSpace': 'pre-wrap', 'wordBreak': 'break-word'},
-                            className="mb-3"
-                        ),
-                        dcc.Upload(
-                            id='upload-file',
-                            children=html.Div([
-                                html.I(className="fas fa-cloud-upload-alt fa-3x mb-3"),
-                                html.Div("Drag and Drop or Select File")
-                            ], className="text-center"),
-                            style={
-                                'borderWidth': '2px',
-                                'borderStyle': 'dashed',
-                                'borderRadius': '5px',
-                                'padding': '20px',
-                                'transition': 'border .3s ease-in-out'
-                            },
-                            className="mb-3"
-                        ),
-                        html.Div(id="uploaded-file-area", className="mb-2"),
-                        html.Div(id="pptx-download-area", className="mb-2"),
-                        dbc.Button("Generate Slides", id='generate-button', color="primary", className="w-100 mt-3", size="lg"),
-                    ], style={'display': 'flex', 'flexDirection': 'column', 'height': '100%'})
-                ], className="mb-4 shadow-sm", style={'height': '100%'})
-            ], md=4, style={'minHeight': '60vh', 'height': '70vh', 'display': 'flex', 'flexDirection': 'column'}),
-            dbc.Col([
-                dbc.Card([
-                    dbc.CardHeader("Edit Slide Content", className="bg-light"),
-                    dbc.CardBody([
-                        dbc.Textarea(
-                            id='slides-content',
-                            placeholder='Generated or pasted markdown will appear here. You can edit the content.',
-                            style={'height': 'calc(100vh - 120px)', 'whiteSpace': 'pre-wrap', 'wordBreak': 'break-word'},
-                            className="mb-3"
-                        ),
-                        dbc.Button("Generate PowerPoint", id='generate-ppt-button', color="primary", className="w-100 mt-3", size="lg"),
-                    ], style={'display': 'flex', 'flexDirection': 'column', 'height': '100%'})
-                ], className="mb-4 shadow-sm", style={'height': '100%'})
-            ], md=8, style={'minHeight': '70vh', 'height': '80vh', 'display': 'flex', 'flexDirection': 'column'})
-        ], style={'minHeight': '100vh', 'height': '100vh'})
-    ], fluid=True, className="mt-4", style={'minHeight': '100vh', 'height': '100vh'}),
+def get_log_area(session_dict):
+    logs = session_dict.get("log_messages", [])
+    return html.Div(
+        [html.Div(log, style={'fontSize': '0.8em'}) for log in logs[:10]],
+        style={'maxHeight': '180px', 'overflowY': 'auto', "fontFamily": "monospace", "background": "#f8f9fa", "padding": "8px", "borderRadius": "5px"}
+    )
+
+app.layout = dbc.Container([
+    dbc.Row([
+        dbc.Col([
+            dbc.Card([
+                dbc.CardHeader("PowerPoint Generator", className="text-center"),
+            ], className="mb-3"),
+        ], width=12)
+    ], className="mb-2"),
+    dbc.Row([
+        dbc.Col([
+            dbc.Card([
+                dbc.CardHeader("Upload & Controls", className="bg-light"),
+                dbc.CardBody([
+                    dcc.Upload(
+                        id='upload-file',
+                        children=html.Div([
+                            html.I(className="fas fa-cloud-upload-alt fa-2x mb-2"),
+                            html.Div("Drag and Drop or Select File")
+                        ], className="text-center"),
+                        style={
+                            'borderWidth': '2px',
+                            'borderStyle': 'dashed',
+                            'borderRadius': '5px',
+                            'padding': '20px',
+                            'marginBottom': '16px'
+                        }
+                    ),
+                    html.Div(id="uploaded-file-area", className="mb-2"),
+                    dbc.Button("Delete Uploaded File", id='delete-file-btn', color="secondary", size="sm", className="mb-3 w-100"),
+                    html.Hr(),
+                    dbc.Textarea(
+                        id='document-text',
+                        placeholder='Paste your document here',
+                        style={'height': '140px', 'whiteSpace': 'pre-wrap', 'wordBreak': 'break-word'},
+                        className="mb-2"
+                    ),
+                    dbc.Button("Generate Slides", id='generate-button', color="primary", className="w-100 mb-2", size="md"),
+                    html.Hr(),
+                    html.Div(id="pptx-download-area", className="mb-2"),
+                    dbc.Button("Delete Generated PowerPoint", id='delete-pptx-btn', color="secondary", size="sm", className="mb-3 w-100"),
+                    html.Hr(),
+                    html.Div("Recent Logs:", className="mb-1", style={"fontWeight": "bold"}),
+                    html.Div(id="log-area", className="mb-2"),
+                ])
+            ])
+        ], width=4, style={'background': '#f7f7f7', 'height': '100vh', 'position': 'fixed', 'left':0, 'top':0, 'bottom':0, 'paddingTop': '15px', 'zIndex': 2}),
+        dbc.Col([
+            dbc.Card([
+                dbc.CardHeader("Slides Markdown (Editable)", className="bg-light"),
+                dbc.CardBody([
+                    dcc.Loading(
+                        id="loading",
+                        type="default",
+                        fullscreen=False,
+                        children=[
+                            dbc.Textarea(
+                                id='slides-content',
+                                placeholder='Generated slide markdown will appear here. You can edit before generating PowerPoint.',
+                                style={'height': 'calc(90vh - 120px)', 'whiteSpace': 'pre-wrap', 'wordBreak': 'break-word'},
+                                className="mb-3"
+                            ),
+                            dbc.Button("Generate PowerPoint", id='generate-ppt-button', color="primary", className="w-100 mt-1", size="lg"),
+                        ]
+                    ),
+                ], style={'height': '100%'})
+            ], className="mb-4 shadow-sm", style={'height': '93vh'})
+        ], width={"size":8, "offset":4}, style={'paddingLeft': '34%', 'background': '#fff', 'height': '100vh'})
+    ], style={"height": "100vh"}),
     dcc.Download(id="download-pptx"),
     dcc.Interval(id='interval-component', interval=10*1000, n_intervals=0)
-])
+], fluid=True, style={"padding":0, "overflowX": "hidden", "minHeight": "100vh"})
 
 @app.callback(
     Output('uploaded-file-area', 'children'),
@@ -330,7 +356,7 @@ app.layout = html.Div([
     Output('document-text', 'value'),
     Output('slides-content', 'value'),
     Output("download-pptx", "data"),
-    Output('session-init', 'data'),
+    Output('log-area', 'children'),
     Input('upload-file', 'filename'),
     Input('upload-file', 'contents'),
     Input('generate-button', 'n_clicks'),
@@ -346,15 +372,14 @@ app.layout = html.Div([
     State('upload-file', 'contents'),
     prevent_initial_call=True
 )
-def unified_callback(upload_filename, upload_contents, gen_btn, ppt_btn, del_file_btn, del_pptx_btn,
-                    doc_text_val, slides_content_val, interval, 
-                    state_doc_text, state_slides_content, state_upload_filename, state_upload_contents):
+def main_callback(upload_filename, upload_contents, gen_btn, ppt_btn, del_file_btn, del_pptx_btn,
+                  doc_text_val, slides_content_val, interval,
+                  state_doc_text, state_slides_content, state_upload_filename, state_upload_contents):
     ctx = dash.callback_context
     session_dict = get_session_dict()
     trigger = ctx.triggered[0]['prop_id'].split('.')[0] if ctx.triggered else None
     download_data = no_update
 
-    # Restore session state on page intervals/refresh
     if trigger == "interval-component" or trigger is None:
         return (
             get_uploaded_file_area(session_dict),
@@ -363,38 +388,31 @@ def unified_callback(upload_filename, upload_contents, gen_btn, ppt_btn, del_fil
             session_dict.get("document_text", ""),
             session_dict.get("slides_markdown", ""),
             no_update,
-            True
+            get_log_area(session_dict)
         )
 
-    # File upload logic
     if trigger == "upload-file" or (upload_filename and upload_contents):
         session_dict["uploaded_filename"] = upload_filename
         session_dict["uploaded_file"] = upload_contents
         add_log(f"File uploaded: {upload_filename}")
 
-    # File delete logic
     if trigger == "delete-file-btn":
         session_dict["uploaded_filename"] = None
         session_dict["uploaded_file"] = None
         add_log("File deleted by user.")
 
-    # PPTX delete logic
     if trigger == "delete-pptx-btn":
         session_dict["generated_pptx"] = None
         session_dict["ppt_ready"] = False
         add_log("PowerPoint deleted by user.")
 
-    # Document text entry or edit
     if trigger == "document-text":
         session_dict["document_text"] = doc_text_val
 
-    # Slides markdown edit
     if trigger == "slides-content":
         session_dict["slides_markdown"] = slides_content_val
 
-    # Generate Slides
     if trigger == "generate-button":
-        # Merge uploaded file and/or manual text
         file_text = ""
         if session_dict.get("uploaded_file") and session_dict.get("uploaded_filename"):
             try:
@@ -410,7 +428,7 @@ def unified_callback(upload_filename, upload_contents, gen_btn, ppt_btn, del_fil
                     session_dict.get("document_text", ""),
                     msg,
                     no_update,
-                    True
+                    get_log_area(session_dict)
                 )
         user_text = session_dict.get("document_text", "")
         combined = (file_text + "\n\n" + user_text).strip() if file_text and user_text else (file_text or user_text)
@@ -424,7 +442,7 @@ def unified_callback(upload_filename, upload_contents, gen_btn, ppt_btn, del_fil
                 session_dict.get("document_text", ""),
                 msg,
                 no_update,
-                True
+                get_log_area(session_dict)
             )
         try:
             add_log("Sending document to Gemini for slide generation...")
@@ -442,13 +460,11 @@ def unified_callback(upload_filename, upload_contents, gen_btn, ppt_btn, del_fil
             session_dict.get("document_text", ""),
             session_dict.get("slides_markdown", ""),
             no_update,
-            True
+            get_log_area(session_dict)
         )
 
-    # Generate PowerPoint
     if trigger == "generate-ppt-button":
         slides_md = session_dict.get("slides_markdown", "")
-        # Always take the latest textarea if changed
         if slides_content_val is not None and slides_content_val.strip() != "":
             slides_md = slides_content_val
             session_dict["slides_markdown"] = slides_md
@@ -461,7 +477,7 @@ def unified_callback(upload_filename, upload_contents, gen_btn, ppt_btn, del_fil
                 session_dict.get("document_text", ""),
                 slides_md,
                 no_update,
-                True
+                get_log_area(session_dict)
             )
         try:
             add_log("Starting PowerPoint generation from slides markdown...")
@@ -481,10 +497,9 @@ def unified_callback(upload_filename, upload_contents, gen_btn, ppt_btn, del_fil
             session_dict.get("document_text", ""),
             session_dict.get("slides_markdown", ""),
             download_data,
-            True
+            get_log_area(session_dict)
         )
 
-    # If editing/entering text/markdown, just update session
     if trigger in ["document-text", "slides-content"]:
         return (
             get_uploaded_file_area(session_dict),
@@ -493,10 +508,9 @@ def unified_callback(upload_filename, upload_contents, gen_btn, ppt_btn, del_fil
             session_dict.get("document_text", ""),
             session_dict.get("slides_markdown", ""),
             no_update,
-            True
+            get_log_area(session_dict)
         )
 
-    # Default - restore
     return (
         get_uploaded_file_area(session_dict),
         get_pptx_download_area(session_dict),
@@ -504,7 +518,7 @@ def unified_callback(upload_filename, upload_contents, gen_btn, ppt_btn, del_fil
         session_dict.get("document_text", ""),
         session_dict.get("slides_markdown", ""),
         no_update,
-        True
+        get_log_area(session_dict)
     )
 
 @app.server.route("/download-pptx")
@@ -543,5 +557,5 @@ def serve_uploaded_file():
 
 if __name__ == '__main__':
     print("Starting the Dash application...")
-    app.run(debug=True, host='0.0.0.0', port=7860)
+    app.run(debug=True, host='0.0.0.0', port=7860, threaded=True)
     print("Dash application has finished running.")