Update app.py

app.py

@@ -9,7 +9,7 @@ import random
 
 import gradio as gr
 
-def generate(logo=None, Vtubername=""):
+def generate(logo=None, Vtubername="", sdkey=""):
   if(logo==None):
         gr.Warning('Please Select Your Photo📸')
 
@@ -17,6 +17,9 @@ def generate(logo=None, Vtubername=""):
         Vtubername = "unkown"
         gr.Warning('Please Select Your Name😱')
 
+  if(sdkey==""):
+        gr.Warning('Please Set valid Stability AI API Key🔑')
+
   def extract_dominant_colors(img, num_colors=3, ignore_edges=True):
     if img.mode == 'RGBA':
           image = img.convert('RGB')
@@ -86,7 +89,6 @@ def generate(logo=None, Vtubername=""):
 
   import requests
   import os
-  sdkey = os.environ["sdkey"]
   from huggingface_hub import InferenceClient
 
   client = InferenceClient(model="mistralai/Mixtral-8x7B-Instruct-v0.1")
@@ -227,10 +229,10 @@ iface = gr.Interface(
     theme=gr.themes.Default(primary_hue="pink",font=[gr.themes.GoogleFont("Mochiy Pop One")]),
     css=css,
     fn=generate,
-    inputs=[gr.Image(label="VTuber Photo", type="pil"), gr.Textbox(label="VTuber Name（*alphabet only!*）")],
+    inputs=[gr.Image(label="VTuber Photo", type="pil"), gr.Textbox(label="VTuber Name（*alphabet only!*）"), gr.Textbox(label="Stability AI API Key")],
     outputs=gr.Image(label="Generated Logo"),
     title="VTuber Logo Generator",
-    description="①Upload photo you wanna VTuberize❤️ <br>② Input the name（*alphabet only!*）⭐️ <br>③Press Submit🧙",
+    description="①Upload photo you wanna VTuberize❤️ <br>② Input the name（*alphabet only!*）⭐️ <br>③ Set your SD3 key🔑(https://platform.stability.ai/account/keys) <br>④Press Submit🧙",
     examples=[["image.jpeg", "gojiteji"]],
     
     allow_flagging=False