Update app.py

app.py

@@ -102,12 +102,14 @@ def generate(prompt,
     slider_x = [concept_2, concept_1]
     # check if avg diff for directions need to be re-calculated
     if randomize_seed:
-            seed = random.randint(0, MAX_SEED)
+        seed = random.randint(0, MAX_SEED)
 
     if not sorted(slider_x) == sorted([x_concept_1, x_concept_2]) or recalc_directions:
         gradio_progress(0, desc="Calculating directions...")
         avg_diff = clip_slider.find_latent_direction(slider_x[0], slider_x[1], num_iterations=iterations)
         x_concept_1, x_concept_2 = slider_x[0], slider_x[1]
+    else:
+        avg_diff = avg_diff_x
 
     images = []
     high_scale = scale
@@ -135,13 +137,13 @@ def generate(prompt,
 
     video_path = f"{uuid.uuid4()}.mp4"
     print(video_path)
-    return x_concept_1,x_concept_2, avg_diff_x, export_to_video(images, video_path, fps=5), canvas, images, images[scale_middle], post_generation_slider_update, seed
+    return x_concept_1, x_concept_2, avg_diff_x, export_to_video(images, video_path, fps=5), canvas, images, images[scale_middle], post_generation_slider_update, seed
 
 def update_pre_generated_images(slider_value, total_images):
     number_images = 0
     if total_images:
         number_images = len(total_images)
-    if(number_images > 0):
+    if number_images > 0:
         scale_tuple = convert_to_centered_scale(number_images)
         return total_images[scale_tuple.index(slider_value)][0]
     else:
@@ -150,35 +152,71 @@ def update_pre_generated_images(slider_value, total_images):
 def reset_recalc_directions():
     return True
 
-# Updated examples with English text
+# =======================
+# 개선된 예제 프롬프트
+# =======================
 examples = [
-    ["flower in mountain", "spring", "winter", 1.5],
-    ["a tomato", "super fresh", "rotten", 2.5],
-    ["여자", "아기", "노인", 2.5]
+    [
+        "A serene mountain landscape with a gentle river flowing in the foreground under a bright blue sky.",
+        "Early Spring",
+        "Deep Winter",
+        1.5
+    ],
+    [
+        "A bustling city street at dusk with neon lights reflecting on wet pavement, capturing the contrast between day and night.",
+        "Daytime",
+        "Nighttime",
+        2.5
+    ],
+    [
+        "An abstract digital art composition featuring vibrant colors and dynamic shapes.",
+        "Calm",
+        "Energetic",
+        2.0
+    ],
+    [
+        "여성의 미소와 함께하는 따뜻한 분위기의 인물 사진",
+        "젊음",
+        "노년",
+        2.5
+    ]
 ]
 
-
+# =======================
+# 밝고 세련된 UI CSS
+# =======================
 css = """
+/* 배경 이미지와 밝은 톤 적용 */
+body {
+    background: #ffffff url('https://images.unsplash.com/photo-1506748686214-e9df14d4d9d0?ixlib=rb-1.2.1&auto=format&fit=crop&w=1600&q=80') no-repeat center center fixed;
+    background-size: cover;
+    font-family: "Helvetica Neue", Helvetica, Arial, sans-serif;
+    color: #333;
+}
+
 footer {
     visibility: hidden;
 }
 
 .container {
     max-width: 1200px;
-    margin: auto;
+    margin: 20px auto;
+    padding: 0 10px;
 }
 
 .main-panel {
-    background-color: rgba(255, 255, 255, 0.05);
+    background-color: rgba(255, 255, 255, 0.9);
     border-radius: 12px;
     padding: 20px;
     margin-bottom: 20px;
+    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);
 }
 
 .controls-panel {
-    background-color: rgba(255, 255, 255, 0.02);
+    background-color: rgba(255, 255, 255, 0.85);
     border-radius: 8px;
     padding: 16px;
+    box-shadow: inset 0 2px 4px rgba(0, 0, 0, 0.05);
 }
 
 .image-display {
@@ -194,12 +232,12 @@ footer {
 
 .advanced-panel {
     margin-top: 20px;
-    border-top: 1px solid rgba(255, 255, 255, 0.1);
+    border-top: 1px solid #eaeaea;
     padding-top: 20px;
 }
 """
 
-with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
+with gr.Blocks(css=css) as demo:
     x_concept_1 = gr.State("")
     x_concept_2 = gr.State("")
     total_images = gr.State([])
@@ -317,7 +355,7 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
                             autoplay=True,
                             height=100
                         )
-                with gr.Row(): # Moved this block to be after the video
+                with gr.Row():  # Moved this block to be after the video
                     with gr.Column():
                         post_generation_image = gr.Image(
                             label=english_labels["Generated Images"],
@@ -333,7 +371,6 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
                             label=english_labels["From 1st to 2nd direction"]
                         )
 
-
     # Examples Section
     gr.Examples(
         examples=examples,
@@ -366,4 +403,4 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()