Update app.py

app.py

@@ -3,25 +3,28 @@ from gradio_client import Client
 import os
 import logging
 
+
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
 
 # API 클라이언트 설정
 api_client = Client("http://211.233.58.202:7960/")
 
-def respond(message):
-    logging.info("Received message: %s", message)
+def respond(message, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
+    logging.info(f"Received message: {message}, seed: {seed}, randomize_seed: {randomize_seed}, "
+                 f"width: {width}, height: {height}, guidance_scale: {guidance_scale}, "
+                 f"num_inference_steps: {num_inference_steps}")
     
     try:
         # 이미지 생성 요청
         result = api_client.predict(
             prompt=message,
-            seed=123,
-            randomize_seed=False,
-            width=1024,
-            height=576,
-            guidance_scale=5,
-            num_inference_steps=28,
+            seed=seed,
+            randomize_seed=randomize_seed,
+            width=width,
+            height=height,
+            guidance_scale=guidance_scale,
+            num_inference_steps=num_inference_steps,
             api_name="/infer_t2i"
         )
         logging.info("API response received: %s", result)
@@ -45,6 +48,7 @@ footer {
 """
 
 
+
 # 이미지 생성을 위한 예제 프롬프트
 examples = [
     ["A glamorous young woman with long, wavy blonde hair and smokey eye makeup, posing in a luxury hotel room. She’s wearing a sparkly gold cocktail dress and holding up a white card with 'openfree.ai' written on it in elegant calligraphy. Soft, warm lighting creates a luxurious atmosphere. ", "q1.webp"],
@@ -78,6 +82,18 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
         input_text = gr.Textbox(label="Enter your prompt for image generation")
         output_image = gr.Image(label="Generated Image")
     
+    with gr.Row():
+        seed = gr.Slider(minimum=0, maximum=1000000, step=1, label="Seed", value=123)
+        randomize_seed = gr.Checkbox(label="Randomize Seed", value=False)
+    
+    with gr.Row():
+        width = gr.Slider(minimum=256, maximum=1024, step=64, label="Width", value=1024)
+        height = gr.Slider(minimum=256, maximum=1024, step=64, label="Height", value=576)
+    
+    with gr.Row():
+        guidance_scale = gr.Slider(minimum=1, maximum=20, step=0.1, label="Guidance Scale", value=5)
+        num_inference_steps = gr.Slider(minimum=1, maximum=100, step=1, label="Number of Inference Steps", value=28)
+    
     with gr.Row():
         for prompt, image_file in examples:
             with gr.Column():
@@ -93,7 +109,11 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                     outputs=input_text
                 )
     
-    input_text.submit(fn=respond, inputs=input_text, outputs=output_image)
+    input_text.submit(
+        fn=respond, 
+        inputs=[input_text, seed, randomize_seed, width, height, guidance_scale, num_inference_steps], 
+        outputs=output_image
+    )
 
 if __name__ == "__main__":
     demo.launch()