v 1.0

app.py

@@ -7,9 +7,11 @@ import nltk
 #nltk.download('punkt')
 
 def generate_image(api_key, prompt):
-    if not api_key:
-        raise ValueError("API Key is required.")
-
+    try:
+        if not api_key:
+            raise ValueError("API Key is required.")
+    except ValueError as e:
+        return e
     openai.api_key = api_key
     response = openai.Image.create(
         prompt=prompt,