Upload 5 files

feifeilib/feifeifluxapi.py

@@ -6,8 +6,8 @@ from feifeilib.feifeisharpened import feifeisharpened
 client = InferenceClient("black-forest-labs/FLUX.1-dev", token=os.getenv('HF_TOKEN'))
 client.headers["x-use-cache"] = "0"
 
-def feifeifluxapi(seed,prompt,quality_select,sharpened_select, styles_Radio,FooocusExpansion_select,height=1152,width=896,guidance_scale=3.5,num_strength=0.35):
-    prompt,generator = feifeiprompt(seed,prompt,quality_select,styles_Radio,FooocusExpansion_select)
+def feifeifluxapi(randomize_seed,seed,prompt,quality_select,sharpened_select, styles_Radio,FooocusExpansion_select,height=1152,width=896,guidance_scale=3.5,num_strength=0.35):
+    prompt,generator = feifeiprompt(randomize_seed,seed,prompt,quality_select,styles_Radio,FooocusExpansion_select)
     image = client.text_to_image(
         prompt=prompt,
         height=height,

feifeilib/feifeiprompt.py

@@ -13,7 +13,7 @@ MAX_SEED = np.iinfo(np.int32).max
 # 去除每行末尾的换行符
 artists = [artist.strip() for artist in artists]
 
-def feifeiprompt(randomize_seed,prompt,quality_select,styles_Radio,FooocusExpansion_select):
+def feifeiprompt(randomize_seed,seed,prompt,quality_select,styles_Radio,FooocusExpansion_select):
     # 处理随机种子
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)

feifeilib/feifeitexttoimg.py

@@ -26,7 +26,7 @@ def feifeitexttoimg(
         num_strength=0.35,
         progress=gr.Progress(track_tqdm=True),
 ):
-    prompt,generator = feifeiprompt(seed,prompt,quality_select,styles_Radio,FooocusExpansion_select)
+    prompt,generator = feifeiprompt(randomize_seed,seed,prompt,quality_select,styles_Radio,FooocusExpansion_select)
     
     image = pipe(
         prompt="",

feifeiui/feifeiui.py

@@ -227,6 +227,7 @@ def create_ui():
         flux_button.click(
             fn=feifeifluxapi,  # Function to run for this button
             inputs=[
+                randomize_seed,
                 flux_seed,
                 flux_prompt,
                 quality_select,