Upload folder using huggingface_hub

app.py

@@ -202,8 +202,9 @@ def main():
                     st.rerun()
         
         with col_settings3:
-            # Pause/Resume controls (only when auto mode is enabled and optimization started)
-            if st.session_state.auto_mode and st.session_state.optimization_started:
+            # Pause/Resume controls (only when auto mode is enabled, optimization started, and not completed)
+            optimization_completed = st.session_state.current_results is not None and st.session_state.current_results[0]
+            if st.session_state.auto_mode and st.session_state.optimization_started and not optimization_completed:
                 if st.session_state.auto_paused:
                     if st.button("▶️ Resume", key="resume_btn"):
                         st.session_state.auto_paused = False
@@ -221,13 +222,24 @@ def main():
             if st.session_state.current_results is not None:
                 is_completed, prompt, generated_image = st.session_state.current_results
                 
-                # Always show the actual current results (most up-to-date)
+                # Get current model info
+                current_model_name = st.session_state.optimizer.get_current_model_name()
+                current_history = st.session_state.optimizer.history if hasattr(st.session_state.optimizer, 'history') else []
+                
+                # If we have current model history, show its latest result instead of current_results
+                if current_history:
+                    latest_step = current_history[-1]
+                    display_image = latest_step['image']
+                    display_prompt = latest_step['prompt']
+                else:
+                    # Fallback to current_results
+                    display_image = generated_image
+                    display_prompt = prompt
+                
+                # Always show the actual current model's latest result
                 with col2:
                     st.subheader("Current Generated Image")
-                    st.image(generated_image, width='stretch')
-                    # Show current step info
-                    current_model_name = st.session_state.optimizer.get_current_model_name()
-                    current_history = st.session_state.optimizer.history if hasattr(st.session_state.optimizer, 'history') else []
+                    st.image(display_image, width='stretch')
                     
                     if current_history:
                         # Show info about the current state
@@ -237,21 +249,23 @@ def main():
                             st.caption(f"🎯 {current_model_name}")
                     else:
                         st.caption(f"🚀 {current_model_name} - Initializing...")
-                
-                # Display current prompt
-                st.text_area("Current Prompt", prompt, height=100)
+                    
+                    # Display current prompt under the image
+                    st.text_area("Current Prompt", display_prompt, height=100)
             else:
                 # Show loading state
                 with col2:
                     st.subheader("Generated Image")
                     st.info("Initializing optimization...")
-                st.text_area("Current Prompt", "Generating initial prompt...", height=100)
+                    # Display loading prompt under the placeholder
+                    st.text_area("Current Prompt", "Generating initial prompt...", height=100)
                 is_completed = False
                 prompt = ""
             
             # Multi-model progress info
             progress_info = st.session_state.optimizer.get_progress_info()
-            current_model = progress_info['current_model_name']
+            # Ensure we get the actual current model name (not cached)
+            current_model = st.session_state.optimizer.get_current_model_name()
             
             # Progress metrics
             col1, col2, col3, col4 = st.columns(4)