Update src/streamlit_app.py

src/streamlit_app.py

@@ -272,84 +272,15 @@ def bulk_download_button(urls: List[str], filename: str = "images_bundle.zip"):
     st.download_button("Download All Images", data=zip_buffer, file_name=filename, mime="application/zip", use_container_width=True)
 
 def main_app():
-    st.set_page_config(page_title="Image Generator + Creative Library", layout="wide")
-    st.title("Multi Model Image Generator")
+    st.set_page_config(page_title="File-to-Image • Creative Library", layout="wide")
+    st.title("File-to-Image Generator")
     with st.sidebar:
-        page = st.radio(" ", ["Generate Bulk Images", "Generate from JSON", "Creative Library"], index=0)
-    if page == "Generate Bulk Images":
-        render_generate_page()
-    elif page == "Generate from JSON":
+        page = st.radio(" ", ["Generate from JSON", "Creative Library"], index=0)
+    if page == "Generate from JSON":
         render_json_page()
     elif page == "Creative Library":
         render_library_page()
 
-def render_generate_page():
-    colA, colB = st.columns([1, 1])
-    with colA:
-        model_key = st.selectbox("Model", list(MODEL_REGISTRY.keys()), index=0)
-        aspect_options = MODEL_REGISTRY[model_key]["aspect_ratios"]
-        aspect_ratio = st.selectbox("Aspect Ratio", aspect_options, index=0)
-        num_images = st.slider("Number of images", min_value=1, max_value=50, value=1, step=1)
-    with colB:
-        prompt = st.text_area("Prompt", placeholder="Describe the image you want to generate...", height=160)
-        debug_mode = st.checkbox("Debug Mode")
-    if st.button("Generate Images", type="primary", use_container_width=True):
-        handle_image_generation_optimized(model_key, aspect_ratio, prompt, num_images, debug_mode)
-
-def handle_image_generation_optimized(model_key: str, aspect_ratio: str, prompt: str, num_images: int, debug_mode: bool = False):
-    if not REPLICATE_API_TOKEN:
-        st.error("Missing REPLICATE_API_TOKEN. Set it as an environment variable.")
-        return
-    if not prompt.strip():
-        st.warning("Please enter a prompt.")
-        return
-    progress = st.progress(0, text="Starting generation...")
-    status_container = st.empty()
-    start_time = time.time()
-    try:
-        with status_container.container():
-            st.info(f"Generating {num_images} image(s) in parallel...")
-        progress.progress(0.1, text="Initializing parallel generation...")
-        all_r2_urls, all_source_urls, generation_errors = generate_images_parallel(model_key, aspect_ratio, prompt.strip(), num_images)
-        progress.progress(0.8, text="Saving results...")
-        rec_id = None
-        if all_r2_urls:
-            rec_id = save_creative_record_optimized(model_key, aspect_ratio, prompt.strip(), all_r2_urls)
-        progress.progress(1.0, text="Complete!")
-        generation_time = time.time() - start_time
-        if all_r2_urls:
-            with status_container.container():
-                st.success(f"Generated {len(all_r2_urls)} image(s) in {generation_time:.1f}s. Saved to DB: {rec_id or 'N/A'}")
-            display_image_gallery_optimized(all_r2_urls)
-            bulk_download_button(all_r2_urls, filename="generated_images.zip")
-        elif all_source_urls:
-            with status_container.container():
-                st.warning("Images generated but R2 upload failed. Showing originals:")
-            display_image_gallery_optimized(all_source_urls)
-            bulk_download_button(all_source_urls, filename="generated_images.zip")
-        else:
-            with status_container.container():
-                st.error("No images were generated.")
-            if generation_errors and debug_mode:
-                with st.expander("Generation Errors", expanded=True):
-                    for error in generation_errors:
-                        st.error(f"{error}")
-    except Exception as e:
-        with status_container.container():
-            st.error(f"Generation failed: {str(e)}")
-
-def display_image_gallery_optimized(urls: List[str]):
-    if not urls:
-        return
-    num_cols = min(4, len(urls)) if len(urls) > 1 else 1
-    cols = st.columns(num_cols)
-    for idx, url in enumerate(urls):
-        with cols[idx % num_cols]:
-            try:
-                display_image_with_download_optimized(url)
-            except Exception as e:
-                st.error(f"Failed to display image: {e}")
-
 def render_library_page():
     st.subheader("Creative Library")
     if "library_page" not in st.session_state:
@@ -567,7 +498,7 @@ def handle_bulk_json_generation(prompts: List[Dict[str, Any]], default_model: st
                 st.error(e)
 
 def main():
-    st.set_page_config(page_title="Bulk Creative Generation", layout="wide")
+    st.set_page_config(page_title="File-to-Image • Creative Library", layout="wide")
     if "authenticated" not in st.session_state:
         st.session_state["authenticated"] = False
     if not st.session_state["authenticated"]: