Update app.py

app.py

@@ -36,13 +36,6 @@ app.config.suppress_callback_exceptions = True
 log_messages = []
 generated_pptx = None
 
-# Add the missing add_log function
-def add_log(message):
-    global log_messages
-    logger.info(message)
-    log_messages.insert(0, f"{time.strftime('%H:%M:%S')} - {message}")  # Insert at the beginning
-
-
 app.layout = html.Div([
     dbc.Container([
         dbc.Row([
@@ -101,12 +94,146 @@ app.layout = html.Div([
     dcc.Interval(id='interval-component', interval=1*1000, n_intervals=0)  # Updates every 1 second
 ])
 
-# ... (keep all other functions unchanged)
+def add_log(message):
+    global log_messages
+    logger.info(message)
+    log_messages.insert(0, f"{time.strftime('%H:%M:%S')} - {message}")  # Insert at the beginning
+
+def process_document(text):
+    add_log("Starting document processing...")
+    prompt = f"""
+    Create a PowerPoint presentation based on the following document. 
+    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.
+    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:
+    {text}
+    """
+    add_log("Sending request to Gemini model...")
+    response = model.generate_content(prompt)
+    add_log("Received response from Gemini model.")
+    return response.text
+
+def generate_image_prompt(slide_content):
+    add_log(f"Generating image prompt for slide: {slide_content[:30]}...")
+    prompt = f"""
+    Based on the following slide content, create a detailed prompt for generating 
+    an image that represents the main idea of the slide. The image should be 
+    suitable for a professional presentation.
+    Slide content:
+    {slide_content}
+    Image prompt:
+    """
+    response = model.generate_content(prompt)
+    add_log("Image prompt generated.")
+    return response.text
+
+def generate_image(prompt):
+    add_log(f"Generating image for prompt: {prompt[:30]}...")
+    url = "https://api.stability.ai/v2beta/stable-image/generate/sd3"
+    headers = {
+        "Authorization": f"Bearer {STABILITY_API_KEY}",
+        "Accept": "image/*"
+    }
+    data = {
+        "prompt": prompt,
+        "output_format": "png"
+    }
+    files = {"none": ''}
+    
+    try:
+        response = requests.post(url, headers=headers, data=data, files=files)
+        response.raise_for_status()
+        add_log("Image generated successfully.")
+        return response.content
+    except requests.exceptions.RequestException as e:
+        add_log(f"Error generating image: {str(e)}")
+        return None
+
+def markdown_to_pptx(md_text):
+    add_log("Starting PowerPoint generation...")
+    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])  # Blank layout
+            
+            # Add title
+            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
+
+            # Add content (left side)
+            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 image (right side)
+            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)}")
+                    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"
+
+    add_log("PowerPoint generation completed.")
+    output = io.BytesIO()
+    prs.save(output)
+    return output.getvalue()
+
+@app.callback(
+    Output('upload-file-name', 'children'),
+    Input('upload-file', 'filename')
+)
+def update_filename(filename):
+    if filename:
+        return html.Div(f"Selected file: {filename}")
+    return ""
+
 
 @app.callback(
     Output('slides-content', 'value'),
     Output('download-ppt-button', 'disabled'),
-    Output('download-ppt-button', 'color'),
     Input('generate-button', 'n_clicks'),
     State('upload-file', 'contents'),
     State('document-text', 'value')
@@ -120,21 +247,20 @@ def generate_slides(n_clicks, file_contents, document_text):
         decoded = base64.b64decode(content_string)
         document_text = decoded.decode('utf-8')
     elif not document_text:
-        return "Please upload a file or enter text.", True, "secondary"
+        return "Please upload a file or enter text.", True
 
     try:
         add_log("Starting slide generation...")
         slides_markdown = process_document(document_text)
         add_log("Slide generation completed.")
-        return slides_markdown, True, "secondary"  # Keep download button disabled
+        return slides_markdown, False  # Enable download button
     except Exception as e:
         add_log(f"An error occurred during slide generation: {str(e)}")
-        return f"An error occurred: {str(e)}", True, "secondary"
+        return f"An error occurred: {str(e)}", True
 
 @app.callback(
     Output('log-output', 'children'),
     Output('download-ppt-button', 'disabled', allow_duplicate=True),
-    Output('download-ppt-button', 'color', allow_duplicate=True),
     Input('generate-ppt-button', 'n_clicks'),
     State('slides-content', 'value'),
     prevent_initial_call=True
@@ -148,11 +274,11 @@ def generate_powerpoint(n_clicks, slides_content):
         add_log("Starting PowerPoint generation...")
         generated_pptx = markdown_to_pptx(slides_content)
         add_log("PowerPoint generation completed.")
-        return '\n'.join(log_messages[:100]), False, "primary"  # Enable download button and change color
+        return '\n'.join(log_messages[:100]), False  # Enable download button
     except Exception as e:
         error_message = f"An error occurred during PowerPoint generation: {str(e)}"
         add_log(error_message)
-        return '\n'.join(log_messages[:100]), True, "secondary"
+        return '\n'.join(log_messages[:100]), True
 
 @app.callback(
     Output("download-pptx", "data"),