Update app.py

app.py

@@ -77,11 +77,10 @@ class timer:
 if not path.exists(cache_path):
     os.makedirs(cache_path, exist_ok=True)
 
-# 인증된 모델 로드
 pipe = FluxPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
     torch_dtype=torch.bfloat16,
-    use_auth_token=HF_TOKEN
+    use_auth_token=HF_TOKEN  # 경고 메시지가 뜨지만 무시 가능
 )
 
 # Hyper-SD LoRA 로드
@@ -134,7 +133,7 @@ def process_and_save_image(height=1024, width=1024, steps=8, scales=3.5, prompt=
     global pipe
     
     if seed is None:
-        seed = torch.randint(0, 1000000, (1,)).item()
+        seed = get_random_seed()
     
     # 한글 감지 및 번역
     def contains_korean(text):
@@ -201,10 +200,6 @@ with gr.Blocks(
             padding: 20px;
             background: #f5f5f5;
             border-radius: 15px;
-            /* 
-               아래처럼 width:100% & margin:0 auto를 주면 
-               화면 전체를 넓게 사용하면서 중앙 정렬됨 
-            */
             width: 100%;
             margin: 0 auto;
         }
@@ -219,8 +214,8 @@ with gr.Blocks(
     gr.HTML(
         """
         <div style="text-align: center; max-width: 800px; margin: 0 auto; padding: 20px;">
-            <h1 style="font-size: 2.5rem; color: #2196F3;">3D Style Image Generator v2.0</h1>
-            <p style="font-size: 1.2rem; color: #666;">Create amazing 3D-style images with AI. https://discord.gg/openfreeai </p>
+            <h1 style="font-size: 2.5rem; color: #2196F3;">3D Style Image Generator</h1>
+            <p style="font-size: 1.2rem; color: #666;">Create amazing 3D-style images with AI</p>
         </div>
         """
     )
@@ -287,11 +282,10 @@ with gr.Blocks(
                 value="3d.webp"
             )
 
-    # Gallery 섹션 (Row 대신 Box 사용 -> 화면 전체에 확장)
-    with gr.Box(elem_classes="gallery-container"):
+    # gallery-container 부분을 Group으로 감싸 화면 전체에 확장
+    with gr.Group(elem_classes="gallery-container"):
         gr.HTML("<h2 class='gallery-title'>Gallery</h2>")
         
-        # grid-template-columns를 repeat(4, 1fr)로 지정하여 한 줄에 4개씩 표시
         gallery_html = """
         <div style='
             display: grid; 
@@ -302,14 +296,11 @@ with gr.Blocks(
         """
         
         for img_file, prompt_text in SAMPLE_IMAGES.items():
-            img_path = os.path.abspath(img_file)  # 절대 경로로 변환
+            img_path = os.path.abspath(img_file)
             if os.path.exists(img_path):
                 try:
-                    # Base64로 이미지 인코딩
                     with open(img_path, "rb") as img:
                         img_data = base64.b64encode(img.read()).decode()
-                    
-                    # MIME타입(webp)
                     gallery_html += f"""
                     <div style='
                         border: 1px solid #ddd;
@@ -340,7 +331,7 @@ with gr.Blocks(
     # 이벤트 핸들러
     generate_btn.click(
         fn=process_and_save_image,
-        inputs=[height, width, steps, scales, prompt, seed],  # 딕셔너리 대신 리스트 형태
+        inputs=[height, width, steps, scales, prompt, seed],
         outputs=output
     )