Update image_gen.py

image_gen.py

@@ -103,7 +103,7 @@ def generate_image(prompt_text, style, model="hf"):
             response = requests.get(api_url)
             if response.status_code != 200:
                 logger.error(f"Pollinations API error: {response.status_code}, {response.text}")
-                st.error(f"Error from image generation API: {response.status_code}")
+                print(f"Error from image generation API: {response.status_code}")
                 return None, None
             image_bytes = response.content
             
@@ -115,7 +115,7 @@ def generate_image(prompt_text, style, model="hf"):
                 g_api_key = os.getenv("GEMINI")
                 if not g_api_key:
                     logger.error("GEMINI_API_KEY not found in environment variables")
-                    st.error("Google Gemini API key is missing. Please set the GEMINI_API_KEY environment variable.")
+                    print("Google Gemini API key is missing. Please set the GEMINI_API_KEY environment variable.")
                     return None, None
                     
                 # Initialize Gemini client
@@ -145,7 +145,7 @@ def generate_image(prompt_text, style, model="hf"):
                 
                 # If no image was found in the response
                 logger.error("No image was found in the Gemini API response")
-                st.error("Gemini API didn't return an image")
+                print("Gemini API didn't return an image")
                 return None, None
                 
             except ImportError:
@@ -155,7 +155,7 @@ def generate_image(prompt_text, style, model="hf"):
                 
             except Exception as e:
                 logger.error(f"Gemini API error: {str(e)}")
-                st.error(f"Error from Gemini image generation: {str(e)}")
+                print(f"Error from Gemini image generation: {str(e)}")
                 return None, None
                 
         else:  # Default to Hugging Face model
@@ -167,7 +167,7 @@ def generate_image(prompt_text, style, model="hf"):
             response = requests.post(api_url, headers=headers, json=payload)
             if response.status_code != 200:
                 logger.error(f"Hugging Face API error: {response.status_code}, {response.text}")
-                st.error(f"Error from image generation API: {response.status_code}")
+                print(f"Error from image generation API: {response.status_code}")
                 return None, None
             image_bytes = response.content
 
@@ -180,7 +180,7 @@ def generate_image(prompt_text, style, model="hf"):
             return image, img_str
             
     except Exception as e:
-        st.error(f"Error generating image: {e}")
+        print(f"Error generating image: {e}")
         logger.error(f"Image generation error: {str(e)}")
         
     # Return a placeholder image in case of failure