From 899abad1ba1029683586d174dd35d79f4b020106 Mon Sep 17 00:00:00 2001
From: ajaysi <ajay.calsoft@gmail.com>
Date: Tue, 18 Jun 2024 09:22:36 +0530
Subject: [PATCH] WIP - UI, firecrawl, long-form - V0.5

---
 lib/ai_writers/keywords_to_blog_streamlit.py  |  7 +++-
 .../main_generate_image_from_prompt.py        | 37 +++++++++++--------
 2 files changed, 27 insertions(+), 17 deletions(-)

diff --git a/lib/ai_writers/keywords_to_blog_streamlit.py b/lib/ai_writers/keywords_to_blog_streamlit.py
index 76bf1018..7816155f 100644
--- a/lib/ai_writers/keywords_to_blog_streamlit.py
+++ b/lib/ai_writers/keywords_to_blog_streamlit.py
@@ -84,8 +84,13 @@ def write_blog_from_keywords(search_keywords, url=None):
         except Exception as err:
             st.error(f"Failed to get blog metadata: {err}")
 
+        generated_image_filepath = None
         try:
-            generated_image_filepath = generate_image(f"{blog_title} + ' ' + {blog_meta_desc}")
+            # FIXME: Temporary fix.
+            text_to_image = f"{blog_title} + ' ' + {blog_meta_desc}"
+            if not text_to_image:
+                text_to_image = blog_markdown_str
+            generated_image_filepath = generate_image(text_to_image)
         except Exception as err:
             st.warning(f"Failed in Image generation: {err}")
 
diff --git a/lib/gpt_providers/text_to_image_generation/main_generate_image_from_prompt.py b/lib/gpt_providers/text_to_image_generation/main_generate_image_from_prompt.py
index 8ec58b44..227b2381 100644
--- a/lib/gpt_providers/text_to_image_generation/main_generate_image_from_prompt.py
+++ b/lib/gpt_providers/text_to_image_generation/main_generate_image_from_prompt.py
@@ -45,20 +45,24 @@ def generate_image(user_prompt):
     """
     # FIXME: Need to remove default value to match sidebar input.
     image_engine = 'Stability-AI'
+    image_stored_at = None
 
-    try:
-        img_prompt = generate_img_prompt(user_prompt)
-        if 'Dalle3' in image_engine:
-            logger.info(f"Calling Dalle3 text-to-image with prompt: {img_prompt}")
-            image_stored_at = generate_dalle3_images(img_prompt)
-        elif 'Stability-AI' in image_engine:
-            logger.info(f"Calling Stable diffusion text-to-image with prompt: \n{img_prompt}")
-            print("\n\n")
-            image_stored_at = generate_stable_diffusion_image(img_prompt)
-    except Exception as err:
-        logger.error(f"Failed to generate Image: {err}")
-        st.warning(f"Failed to generate Image: {err}")
-    return image_stored_at
+    if user_prompt:
+        try:
+            img_prompt = generate_img_prompt(user_prompt)
+            if 'Dalle3' in image_engine:
+                logger.info(f"Calling Dalle3 text-to-image with prompt: {img_prompt}")
+                image_stored_at = generate_dalle3_images(img_prompt)
+            elif 'Stability-AI' in image_engine:
+                logger.info(f"Calling Stable diffusion text-to-image with prompt: \n{img_prompt}")
+                print("\n\n")
+                image_stored_at = generate_stable_diffusion_image(img_prompt)
+            return image_stored_at
+        except Exception as err:
+            logger.error(f"Failed to generate Image: {err}")
+            st.warning(f"Failed to generate Image: {err}")
+    else:
+        logger.error("Skipping Image creation, No prompt provided.")
 
 
 def generate_img_prompt(user_prompt):
@@ -66,15 +70,16 @@ def generate_img_prompt(user_prompt):
     Given prompt, this functions generated a prompt for image generation.
     """
     prompt = f"""
-        As an expert prompt engineer and artist, I will provide you with 'text' for creating image.
-        I want you to act as a prompt generator for AI text to image models(no more than 150 words).
+        As an expert prompt generator for AI text to image models and artist, I will provide you with 'user text' for creating images.
+        Your task is to create a prompt for a highly relevant image from given 'user text'.
         \n
         Choose from various art styles, utilize light & shadow effects etc.
         Make sure to avoid common image generation mistakes.
         Reply with only one answer, no descrition and in plaintext.
         Make sure your prompt is detailed and creative descriptions that will inspire unique and interesting images from the AI. 
         
-        \n\ntext:{user_prompt} """
+        \n\nuser text:  
+        '''{user_prompt}'''"""
 
     response = llm_text_gen(prompt)
     return response