Update app.py

app.py

@@ -9,8 +9,8 @@ import torch
 from diffusers import DiffusionPipeline
 from PIL import Image
 
-# Create permanent storage directory
-SAVE_DIR = "saved_images"  # Gradio will handle the persistence
+# ---------- 초기 설정 및 모델 로드 ----------
+SAVE_DIR = "saved_images"  # Gradio가 저장소 관리를 수행
 if not os.path.exists(SAVE_DIR):
     os.makedirs(SAVE_DIR, exist_ok=True)
 
@@ -26,16 +26,15 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
 def save_generated_image(image, prompt):
-    # Generate unique filename with timestamp
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     unique_id = str(uuid.uuid4())[:8]
     filename = f"{timestamp}_{unique_id}.png"
     filepath = os.path.join(SAVE_DIR, filename)
     
-    # Save the image
+    # 이미지 저장
     image.save(filepath)
     
-    # Save metadata
+    # 메타데이터 저장
     metadata_file = os.path.join(SAVE_DIR, "metadata.txt")
     with open(metadata_file, "a", encoding="utf-8") as f:
         f.write(f"{filename}|{prompt}|{timestamp}\n")
@@ -46,22 +45,28 @@ def load_generated_images():
     if not os.path.exists(SAVE_DIR):
         return []
     
-    # Load all images from the directory
+    # 디렉토리 내 이미지 파일 로드
     image_files = [
         os.path.join(SAVE_DIR, f)
         for f in os.listdir(SAVE_DIR)
         if f.endswith(('.png', '.jpg', '.jpeg', '.webp'))
     ]
-    # Sort by creation time (newest first)
+    # 생성 시각 기준 정렬 (최신 파일 우선)
     image_files.sort(key=lambda x: os.path.getctime(x), reverse=True)
     return image_files
 
 def load_predefined_images():
-    # Return empty list since we're not using predefined images
+    # 별도 사전 이미지 없음
     return []
 
-# 간단한 CSS: 타이틀 크기, 중앙 정렬, footer 숨김 등
 css = """
+/* 배경 그라디언트를 주거나, 폰트/타이틀 크기 등을 원하는 대로 꾸밀 수 있습니다. */
+body {
+    font-family: 'Open Sans', sans-serif;
+    background: linear-gradient(135deg, #f5f7fa, #c3cfe2);
+    margin: 0;  /* 기본 여백 제거 */
+    padding: 0;
+}
 .title {
     font-size: 1.8em;
     font-weight: bold;
@@ -99,30 +104,25 @@ def inference(
         joint_attention_kwargs={"scale": lora_scale},
     ).images[0]
     
-    # Save the generated image
     filepath = save_generated_image(image, prompt)
-    
-    # Return the image, seed, and updated gallery
     return image, seed, load_generated_images()
 
+
+# ---------- 예시 프롬프트 ----------
 examples = [
     "Pepe the frog playing fetch with a golden retriever in a sunny park. He wears casual weekend clothes and tosses a bright red frisbee with a goofy grin. The dog leaps gracefully through the air, tail wagging with excitement. The warm afternoon sunlight filters through the trees, creating a humorous meme-like atmosphere. [pepe]",
-
     "Pepe the frog dressed in full military gear, standing at attention with a standard-issue rifle. His crisp uniform is adorned with cartoonish medals. Other frog soldiers march in formation behind him during a grand meme parade, conveying discipline mixed with comical charm. [pepe]",
-
     "A medieval Pepe knight in gleaming armor, proudly holding an ornate sword and shield. He stands in front of a majestic castle with a swirling moat. His shield features a cartoon frog crest, and sunlight gleams off his polished armor, adding a humorous yet epic feel. [pepe]",
-
     "A charismatic Pepe the frog addressing a crowd from a podium. He wears a well-fitted suit and gestures with exaggerated cartoon expressions while speaking. The audience is filled with fellow frog characters holding supportive banners. Cameras capture this grand meme moment. [pepe]",
-
     "Pepe the frog enjoying a peaceful morning at home, reading a newspaper at his kitchen table. He wears comfy pajamas and sips coffee from a novelty frog mug. Sunlight streams through the window, illuminating a quaint plant on the countertop in this cozy, meme-inspired scene. [pepe]",
-
     "Businessman Pepe walking confidently through a sleek office lobby, briefcase in hand. He wears a tailored navy suit, and his wide frog eyes convey determination. Floor-to-ceiling windows reveal a bustling cityscape behind him, blending corporate professionalism with meme humor. [pepe]"
 ]
 
 
-# Blocks에서 기존 theme 제거 후 css만 적용
-with gr.Blocks(css=css, analytics_enabled=False) as demo:
-    gr.HTML('<div class="title"> PEPE Meme Generator </div>')
+# ---------- UI ----------
+# 원하는 그라디오 테마를 선택해 적용합니다. 아래는 Soft 테마에 primary_hue="emerald"를 지정한 예시입니다.
+with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="emerald"), analytics_enabled=False) as demo:
+    gr.HTML('<div class="title">PEPE Meme Generator</div>')
 
     gr.HTML("""
         <a href="https://visitorbadge.io/status?path=https%3A%2F%2Fopenfree-pepe.hf.space">
@@ -153,7 +153,10 @@ with gr.Blocks(css=css, analytics_enabled=False) as demo:
                         step=1,
                         value=42,
                     )
-                    randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+                    randomize_seed = gr.Checkbox(
+                        label="Randomize seed",
+                        value=True
+                    )
 
                     with gr.Row():
                         width = gr.Slider(
@@ -201,7 +204,7 @@ with gr.Blocks(css=css, analytics_enabled=False) as demo:
                 )
 
         with gr.Tab("Gallery"):
-            gallery_header = gr.Markdown("### Generated Images Gallery")
+            gr.Markdown("### Generated Images Gallery")
             generated_gallery = gr.Gallery(
                 label="Generated Images",
                 columns=6,
@@ -212,8 +215,7 @@ with gr.Blocks(css=css, analytics_enabled=False) as demo:
             )
             refresh_btn = gr.Button("🔄 Refresh Gallery")
 
-
-    # Event handlers
+    # Gallery 새로고침 핸들러
     def refresh_gallery():
         return load_generated_images()
 
@@ -223,6 +225,7 @@ with gr.Blocks(css=css, analytics_enabled=False) as demo:
         outputs=generated_gallery,
     )
 
+    # Run 버튼 & 프롬프트 입력 이벤트 처리
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=inference,