231228: update diffusion

app.py

@@ -184,6 +184,7 @@ with st.expander("Press Text Generation Zone"):
 
         # Add a text input for the HF key
         hf_key_mistral = st.secrets["hf_key"] 
+        
        
 
         #defaults for Mistral
@@ -336,7 +337,7 @@ with st.expander("Image Generation Territory"):
     if st.button("Generate Image from New Caption of Artwork"):
         if st.session_state['new_caption']:
             prompt_caption = f"contemporary art of {st.session_state['new_caption']}"
-            st.session_state['image_from_caption'] = client.text_to_image(prompt_caption)
+            st.session_state['image_from_caption'] = client.text_to_image(prompt_caption, model="prompthero/openjourney-v4")
 
     # Display the image generated from new caption
     if st.session_state['image_from_caption'] is not None:
@@ -346,7 +347,7 @@ with st.expander("Image Generation Territory"):
     if st.button("Generate Image from Simplified Text"):
         if st.session_state['simplified_text']:
             prompt_summary = f"contemporary art of {st.session_state['simplified_text']}"
-            st.session_state['image_from_simplified_text'] = client.text_to_image(prompt_summary)
+            st.session_state['image_from_simplified_text'] = client.text_to_image(prompt_summary, model="prompthero/openjourney-v4")
 
     # Display the image generated from simplified text
     if st.session_state['image_from_simplified_text'] is not None:
@@ -354,9 +355,20 @@ with st.expander("Image Generation Territory"):
 
     # Button to generate image from press text from simplified text
     if st.button("Generate Image from new Press Text from Simplified Text"):
-        if st.session_state['message_content_from_simplified_text']:
-            prompt_press_text = f"contemporary art of {st.session_state['message_content_from_simplified_text']}"
-            st.session_state['image_from_press_text'] = client.text_to_image(prompt_press_text)
+        text_to_use = None
+
+        # Check which variable is available and set it to text_to_use
+        if 'mistral_from_simplified' in st.session_state and st.session_state['mistral_from_simplified']:
+            text_to_use = st.session_state['mistral_from_simplified']
+        elif 'message_content_from_simplified_text' in st.session_state and st.session_state['message_content_from_simplified_text']:
+            text_to_use = st.session_state['message_content_from_simplified_text']
+
+        # Use the available text to generate the image
+        if text_to_use:
+            prompt_press_text_simple = f"contemporary art of {text_to_use}"
+            st.session_state['image_from_press_text'] = client.text_to_image(prompt_press_text_simple, model="prompthero/openjourney-v4")
+        else:
+            st.error("First generate a caption.")
 
     # Display the image generated from press text from simplified text
     if st.session_state['image_from_press_text'] is not None:
@@ -364,14 +376,30 @@ with st.expander("Image Generation Territory"):
                 caption="Image from Press Text from simplified Text", 
                 use_column_width=True)
 
+
     # Button to generate image from press text from caption
     if st.button("Generate Image from new Press Text from new Caption"):
-        if st.session_state['message_content_from_caption']:
-            prompt_press_text_caption = f"contemporary art of {st.session_state['message_content_from_caption']}"
-            st.session_state['image_from_press_text_from_caption'] = client.text_to_image(prompt_press_text_caption)
+        # Initialize the variable
+        text_to_use_cap = None
+
+        # Check which variable is available and set it to text_to_use
+        if 'mistral_from_caption' in st.session_state and st.session_state['mistral_from_caption']:
+            text_to_use_cap = st.session_state['mistral_from_caption']
+        elif 'message_content_from_caption' in st.session_state and st.session_state['message_content_from_caption']:
+            text_to_use_cap = st.session_state['message_content_from_caption']
+
+        # Use the available text to generate the image
+        if text_to_use_cap:
+            prompt_press_text_caption = f"contemporary art of {text_to_use_cap}"
+            st.session_state['image_from_press_text_from_caption'] = client.text_to_image(prompt_press_text_caption, model="prompthero/openjourney-v4")
+        else:
+            st.error("First generate a caption.")
 
     # Display the image generated from press text from caption
     if st.session_state['image_from_press_text_from_caption'] is not None:
         st.image(st.session_state['image_from_press_text_from_caption'], 
                 caption="Image from Press Text from new Caption", 
                 use_column_width=True)
+        
+st.markdown("---")
+