Update app.py

app.py

@@ -1,64 +1,73 @@
 import gradio as gr
 import os
-import zipfile
+import base64
 from PIL import Image
+from io import BytesIO
 
-def process_and_preview(input_files):
-    if not input_files:
-        return None, None
+def image_to_base64(img):
+    buffered = BytesIO()
+    # Downscale for mobile brochure feel (max width 600px)
+    img.thumbnail((600, 600))
+    img.save(buffered, format="PNG")
+    return base64.b64encode(buffered.getvalue()).decode()
+
+def generate_brochure(image_files, info_file):
+    if not image_files or not info_file:
+        return None
     
-    output_folder = "processed_images"
-    os.makedirs(output_folder, exist_ok=True)
+    # 1. Read and split the info text
+    with open(info_file.name, 'r') as f:
+        text_content = f.read().split('\n\n') # Assumes double new-line separates info
     
-    processed_paths = []
+    md_content = "# Project Brochure\n\n"
     
-    # Process images
-    for i, file_path in enumerate(input_files, start=1):
-        img = Image.open(file_path)
-        new_name = f"image{i}.png"
-        save_path = os.path.join(output_folder, new_name)
-        img.save(save_path)
-        processed_paths.append(save_path)
-
-    # Create ZIP
-    zip_path = "processed_batch.zip"
-    with zipfile.ZipFile(zip_path, 'w') as zipf:
-        for file in processed_paths:
-            zipf.write(file, arcname=os.path.basename(file))
+    # 2. Logic to group 3 images and 3 text blocks
+    # We loop through images in steps of 3
+    for i in range(0, len(image_files), 3):
+        md_content += '<div style="max-width: 400px; margin: auto; border: 1px solid #ccc; padding: 10px;">\n\n'
+        md_content += f"## Page { (i//3) + 1 }\n\n"
+        
+        # Get the next 3 images and next 3 text blocks
+        current_images = image_files[i:i+3]
+        current_texts = text_content[i:i+3]
+        
+        for img_path, txt in zip(current_images, current_texts):
+            img = Image.open(img_path)
+            b64 = image_to_base64(img)
             
-    # Return both the zip file and the list of paths for the Gallery preview
-    return zip_path, processed_paths
+            # Add to Markdown with HTML for mobile-style sizing
+            md_content += f'<img src="data:image/png;base64,{b64}" style="width:100%; border-radius:8px;">\n\n'
+            md_content += f"{txt}\n\n---\n\n"
+        
+        md_content += '</div>\n<div style="page-break-after: always;"></div>\n\n'
+
+    # 3. Save the result
+    output_path = "brochure.md"
+    with open(output_path, "w", encoding="utf-8") as f:
+        f.write(md_content)
+        
+    return output_path
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🖼️ Image Batch Processor (Gradio 6.9)")
-    gr.Markdown("Upload images, rename them, and download the ZIP. [GitHub](https://github.com)")
+# --- Gradio UI ---
+with gr.Blocks() as demo:
+    gr.Markdown("# 📱 Mobile Brochure Generator")
+    gr.Markdown("Upload images and an `info.txt` file to generate a self-contained Markdown brochure.")
     
     with gr.Row():
-        with gr.Column(scale=1):
-            file_input = gr.File(
-                label="Step 1: Upload Images", 
-                file_count="multiple", 
-                file_types=["image"]
-            )
-            process_btn = gr.Button("Step 2: Process & Zip", variant="primary")
+        with gr.Column():
+            img_input = gr.File(label="Upload Images", file_count="multiple", file_types=["image"])
+            txt_input = gr.File(label="Upload info.txt", file_types=[".txt"])
+            gen_btn = gr.Button("Generate Brochure", variant="primary")
             
-        with gr.Column(scale=2):
-            # The preview gallery
-            preview_gallery = gr.Gallery(
-                label="Image Preview", 
-                columns=4, 
-                height="auto",
-                object_fit="contain"
-            )
-            # The final download button/file
-            zip_output = gr.File(label="Step 3: Download ZIP")
+        with gr.Column():
+            out_file = gr.File(label="Download .md Brochure")
 
-    # Connect the button to the function
-    process_btn.click(
-        fn=process_and_preview, 
-        inputs=file_input, 
-        outputs=[zip_output, preview_gallery]
+    gen_btn.click(
+        fn=generate_brochure,
+        inputs=[img_input, txt_input],
+        outputs=out_file
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    # Note: Using the new Gradio 6.0+ launch style
+    demo.launch(theme=gr.themes.Soft())