Update app.py

app.py

@@ -14,7 +14,6 @@ from PIL import Image
 from transformers import pipeline
 import base64
 
-
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
 
 # Hugging Face 토큰 설정
@@ -124,7 +123,7 @@ def process_and_save_image(height=1024, width=1024, steps=8, scales=3.5, prompt=
         translated = translator(prompt)[0]['translation_text']
         prompt = translated
     
-    formatted_prompt = f"wbgmsst, 3D, {prompt} ,white background"
+    formatted_prompt = f"wbgmsst, 3D, {prompt}, white background"
     
     with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16), timer("inference"):
         try:
@@ -147,19 +146,9 @@ def process_and_save_image(height=1024, width=1024, steps=8, scales=3.5, prompt=
             print(f"Error in image generation: {str(e)}")
             return None
 
-def generate_image(h, w, s, sc, p, sd):
-    """Gradio에서 넘겨받은 파라미터 순서에 맞춰 이미지 생성 함수 호출"""
-    return process_and_save_image(
-        height=h,
-        width=w,
-        steps=s,
-        scales=sc,
-        prompt=p,
-        seed=sd
-    )
-
 def update_random_seed():
-    return gr.Number.update(value=get_random_seed())
+    """버튼으로 눌렀을 때 새로운 시드를 업데이트"""
+    return gr.update(value=get_random_seed())
 
 # Gradio 인터페이스
 with gr.Blocks(
@@ -270,12 +259,12 @@ with gr.Blocks(
                 value="3d.webp"
             )
 
-# Gallery 섹션
+    # Gallery 섹션
     with gr.Row(elem_classes="gallery-container"):
         gr.HTML("<h2 class='gallery-title'>Gallery</h2>")
         gallery_html = "<div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>"
         
-        for img_file, prompt in SAMPLE_IMAGES.items():
+        for img_file, prompt_text in SAMPLE_IMAGES.items():
             img_path = os.path.abspath(img_file)  # 절대 경로로 변환
             if os.path.exists(img_path):
                 try:
@@ -283,6 +272,7 @@ with gr.Blocks(
                     with open(img_path, "rb") as img:
                         img_data = base64.b64encode(img.read()).decode()
                     
+                    # 3d1.webp 파일이므로 MIME타입은 webp
                     gallery_html += f"""
                     <div style='
                         border: 1px solid #ddd;
@@ -301,7 +291,7 @@ with gr.Blocks(
                             font-weight: bold;
                             color: #333;
                             padding: 10px;
-                        '>Prompt: {prompt}</p>
+                        '>Prompt: {prompt_text}</p>
                     </div>
                     """
                 except Exception as e:
@@ -310,31 +300,10 @@ with gr.Blocks(
         gallery_html += "</div>"
         gr.HTML(gallery_html)
     
-# 이벤트 핸들러 부분만 수정
-    def generate_image(h, w, s, sc, p, sd):
-        return process_and_save_image(
-            height=h,
-            width=w,
-            steps=s,
-            scales=sc,
-            prompt=p,
-            seed=sd
-        )
-
-    def update_random_seed():
-        return gr.update(value=get_random_seed())
-
-    # 이벤트 바인딩
+    # 이벤트 핸들러
     generate_btn.click(
-        fn=process_and_save_image,  # 직접 process_and_save_image 함수 사용
-        inputs={
-            "height": height,
-            "width": width,
-            "steps": steps,
-            "scales": scales,
-            "prompt": prompt,
-            "seed": seed
-        },
+        fn=process_and_save_image,
+        inputs=[height, width, steps, scales, prompt, seed],  # 딕셔너리 대신 리스트 형태
         outputs=output
     )
 
@@ -345,4 +314,4 @@ with gr.Blocks(
     )
 
 if __name__ == "__main__":
-    demo.launch(allowed_paths=[PERSISTENT_DIR])
+    demo.launch(allowed_paths=[PERSISTENT_DIR])