Update app.py

app.py

@@ -9,6 +9,7 @@ from diffusers import (
     AutoencoderTiny,
 )
 from huggingface_hub import hf_hub_download
+from feifeilib.feifeichat import feifeichat
 
 import config
 styles_name = [style["name"] for style in config.style_list]
@@ -141,8 +142,26 @@ with gr.Blocks(css=css) as demo:
                 multiselect=False,
                 value="(None)",
             )
+        feifei_chat_text1 = gr.Textbox(
+            label="输入您的问题1：", show_label=False, container=False, lines=1, value="kpop model")
+        feifei_chat_text2 = gr.Textbox(
+            label="输入您的问题2：", show_label=False, container=False, lines=3, value="bikini at sea")
+        
+        # 定义模型选择下拉框
+        feifei_chat_Dropdown = gr.Dropdown(
+            [
+                "meta/llama-3.3-70b-instruct",
+                "nvidia/llama-3.3-nemotron-super-49b-v1",
+                "mistralai/Mistral-Nemo-Instruct-2411",
+            ],
+            value="mistralai/Mistral-Nemo-Instruct-2411",
+            label="选择模型", show_label=False, container=False
+        )
+        # 定义提交按钮
+        feifei_chat_btn = gr.Button(value="Gen Prompt")
         result = gr.Image(label="Result", show_label=False,height=520)
         Expansion_select = gr.Checkbox(label="FeiFei Expansion") 
+        
         with gr.Accordion("Advanced Settings", open=False):
             
             seed = gr.Slider(
@@ -219,11 +238,15 @@ with gr.Blocks(css=css) as demo:
             cache_examples=False
         )
 
-    gr.on(
-        triggers=[run_button.click, prompt.submit],
+    run_button.click(
         fn = infer,
         inputs = [prompt, styles_Radio, Expansion_select, seed, randomize_seed, width, height, num_inference_steps, guidancescale, num_feifei, nsfw_select, nsfw_slider],
         outputs = [result, seed]
     )
+    feifei_chat_btn.click(
+        fn=feifeichat,
+        inputs=[feifei_chat_text1, feifei_chat_text2, feifei_chat_Dropdown],
+        outputs=prompt
+    )
 
 demo.launch()