WIP - Streamlit UI, firecrawl - V0.5

lib/gpt_providers/text_to_image_generation/gen_stabl_diff_img.py

@@ -8,6 +8,7 @@ import os
 import requests
 from PIL import Image
 from io import BytesIO
+import streamlit as st
 
 from .save_image import save_generated_image
 
@@ -18,7 +19,7 @@ def generate_stable_diffusion_image(prompt):
     api_key = os.getenv("STABILITY_API_KEY")
     
     if api_key is None:
-        raise Exception("Missing Stability API key.")
+        st.warning("Missing Stability API key.")
     
     response = requests.post(
         f"{api_host}/v1/generation/{engine_id}/text-to-image",

lib/gpt_providers/text_to_image_generation/main_generate_image_from_prompt.py

@@ -11,6 +11,7 @@
 import os
 import sys
 import datetime
+import streamlit as st
 
 import openai  # OpenAI Python library to make API calls
 from loguru import logger
@@ -56,6 +57,7 @@ def generate_image(user_prompt):
             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