using another version of demo

app.py

@@ -6,24 +6,23 @@ from stability_generate_img import generate_image_with_stability
 from finetune_generate_img import generate_finetuned_img
 from midjourney_generate_img import midjourney_generate_img
 
-# Page configuration
+
+# Configuración de la página
 st.set_page_config(layout="wide")
 st.sidebar.title("API Keys")
 st.markdown("<h1 style='text-align: center; color: grey;'>Image Generation App</h1>", unsafe_allow_html=True)
 st.text("Prepared by [email protected] for fomo.ai")
 
-# List of API key names
+# Lista de nombres de las API keys
 api_key_names = ["OPENAI_API_KEY", "MIDJOURNEY_GOAPI_TOKEN", "REPLICATE_API_TOKEN", "STABILITY_API_KEY"]
 
-# Initialize session state if it does not exist
+# Inicializar el estado de la sesión si no existe
 if 'api_keys' not in st.session_state:
     st.session_state['api_keys'] = {key_name: "" for key_name in api_key_names}
 if 'editable_prompt' not in st.session_state:
     st.session_state['editable_prompt'] = ""
-if 'upload_completed' not in st.session_state:
-    st.session_state['upload_completed'] = False
 
-# Define a function to request and update API keys
+# Definir una función para solicitar y actualizar las API keys
 def request_and_update_api_keys():
     all_keys_entered = True
     for key_name in api_key_names:
@@ -35,49 +34,53 @@ def request_and_update_api_keys():
 
 all_keys_entered = request_and_update_api_keys()
 
-# Check if all API keys have been entered
+# Revisar si todas las API keys han sido ingresadas
 if all_keys_entered:
-    # Section to upload the image
-    uploaded_file = st.file_uploader("Upload Image to analyze", type=['jpg', 'jpeg', 'png'], on_change=lambda: setattr(st.session_state, 'upload_completed', True))
+    # Sección para subir la imagen
+    uploaded_file = st.file_uploader("Upload Image to analyze", type=['jpg', 'jpeg', 'png'])
     if uploaded_file is not None:
-        st.session_state['upload_completed'] = True
-        left_co, cent_co, _ = st.columns([1, 2, 1])
+        left_co, cent_co,last_co = st.columns(3)
         with cent_co:
             st.image(uploaded_file, caption="Uploaded Image")
-        _, right_co = st.columns([5, 1])
-        if right_co.button("Generate Prompt"):
+        # Botón para generar el prompt solo si hay una imagen subida
+        if st.button("Generate Prompt"):
             with st.spinner("Generating Prompt..."):
                 with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_file:
                     temp_file.write(uploaded_file.getvalue())
                     temp_path = temp_file.name
                 api_key = st.session_state['api_keys']['OPENAI_API_KEY']
                 prompt = generate_prompt_with_vision(temp_path, api_key=api_key)
-                st.session_state['editable_prompt'] = prompt
-                st.session_state['upload_completed'] = False
+                st.success("Done!")
+                st.session_state['editable_prompt'] = prompt  # Actualizar el prompt en el estado de la sesión
+
+    editable_prompt = st.text_area("Edit the prompt as needed:", placeholder="Enter your prompt here...", height=150, key='editable_prompt', label_visibility='hidden')
+
+    col1, col2, col3, col4 = st.columns(4)
+    
+    if st.session_state['editable_prompt'] and st.button("Generate New Image"):
+        with col1:
+            with st.spinner("Generating DALL·E Image..."):
+                result_path_1 = generate_img_with_dalle(st.session_state['editable_prompt'], api_key=st.session_state['api_keys']['OPENAI_API_KEY'])
+                st.success("Generated DALL·E Image!")
+                st.image(result_path_1, caption="DALL·E Image")
+                
+        with col2:
+            with st.spinner("Generating Stable Diffusion Image..."):
+                result_path_2 = generate_image_with_stability(st.session_state['editable_prompt'], api_key=st.session_state['api_keys']['STABILITY_API_KEY'])
+                st.success("Generated Stable Diffusion Image!")
+                st.image(result_path_2, caption="Stable Diffusion Image")
+        
+        with col3:
+            with st.spinner("Generating Finetuning Image..."):
+                result_path_3 = generate_finetuned_img(st.session_state['editable_prompt'], api_key=st.session_state['api_keys']['REPLICATE_API_TOKEN'])
+                st.success("Generated Image using a finetuned model!")
+                st.image(result_path_3, caption="Finetuned SDXL Image")
+        
+        with col4:
+            with st.spinner("Generating Midjourney Image..."):
+                result_path_4 = midjourney_generate_img(st.session_state['editable_prompt'], api_key=st.session_state['api_keys']['MIDJOURNEY_GOAPI_TOKEN'])
+                st.success("Generated Midjourney Image!")
+                st.image(result_path_4, caption="Midjourney Image")
 
-    if st.session_state['upload_completed']:
-        # Maintain the content of the editable prompt after generating images
-        editable_prompt = st.text_area("Edit the prompt as needed:", value=st.session_state['editable_prompt'], placeholder="Enter your prompt here...", height=150, key='editable_prompt', on_change=lambda: st.session_state.update(editable_prompt=editable_prompt))
-        if st.button("Generate New Image", key='generate_image_btn'):
-            col1, col2, col3, col4 = st.columns(4)
-            with col1:
-                with st.spinner("Generating DALL·E Image..."):
-                    result_path_1 = generate_img_with_dalle(st.session_state['editable_prompt'], api_key=st.session_state['api_keys']['OPENAI_API_KEY'])
-                    st.image(result_path_1, caption="DALL·E Image")
-            with col2:
-                with st.spinner("Generating Stable Diffusion Image..."):
-                    result_path_2 = generate_image_with_stability(st.session_state['editable_prompt'], api_key=st.session_state['api_keys']['STABILITY_API_KEY'])
-                    st.image(result_path_2, caption="Stable Diffusion Image")
-            with col3:
-                with st.spinner("Generating Finetuning Image..."):
-                    result_path_3 = generate_finetuned_img(st.session_state['editable_prompt'], api_key=st.session_state['api_keys']['REPLICATE_API_TOKEN'])
-                    st.image(result_path_3, caption="Finetuned SDXL Image")
-            with col4:
-                with st.spinner("Generating Midjourney Image..."):
-                    result_path_4 = midjourney_generate_img(st.session_state['editable_prompt'], api_key=st.session_state['api_keys']['MIDJOURNEY_GOAPI_TOKEN'])
-                    st.image(result_path_4, caption="Midjourney Image")
-            # Update the prompt in session state to keep the text
-            st.session_state['editable_prompt'] = editable_prompt
-            st.session_state['upload_completed'] = False  # Disable the text area and button after generating images
 else:
-    st.warning('Please enter all required API keys to proceed.', icon="⚠️")
+    st.warning('Please enter all required API keys to proceed.', icon="⚠️")