Sinh tuan tu

app.py

@@ -8,7 +8,6 @@ import traceback
 import os
 from pathlib import Path
 
-
 # === Thiết lập môi trường ===
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"Device: {device}")
@@ -119,28 +118,37 @@ def generate_image(prompt, style):
         img_byte_arr.seek(0)
         return blank, img_byte_arr
 
-# === Main processing function ===
-def process(article_text, style_choice):
-    print(f"Processing article: {article_text[:50]}...")
+# === Main processing function with staged outputs ===
+def process_step_by_step(article_text, style_choice, state=None):
+    if state is None:
+        state = {"title": None, "prompt": None, "image": None, "file_path": None}
+    
+    # Bước 1: Tóm tắt và tạo tiêu đề
     title, summary = summarize(article_text)
     print(f"Summary title: {title}")
+    state["title"] = title
+    yield state, title, None, None, None
+    
+    # Bước 2: Tạo prompt
     prompt = generate_prompt(title)
     print(f"Generated prompt: {prompt}")
+    state["prompt"] = prompt
+    yield state, title, prompt, None, None
+    
+    # Bước 3: Tạo ảnh
     image, img_bytes = generate_image(prompt, style_choice)
     print(f"Image generated: {image.size if image else 'None'}")
-
-    # Chuyển BytesIO thành file tạm và trả về đường dẫn
+    
+    # Lưu ảnh tạm thời
     temp_dir = "./temp"
     os.makedirs(temp_dir, exist_ok=True)
     temp_file = os.path.join(temp_dir, f"generated_image_{id(image)}.png")
     image.save(temp_file, format="PNG")
-    with open(temp_file, "rb") as f:
-        img_file = f.read()
-    # Trả về đường dẫn tạm thời cho Gradio
-    file_path = temp_file
-
+    state["image"] = image
+    state["file_path"] = temp_file
+    
     print(f"✅ Process completed")
-    return title, prompt, image, file_path
+    yield state, title, prompt, image, temp_file
 
 # === Gradio UI ===
 def create_app():
@@ -148,6 +156,9 @@ def create_app():
         gr.Markdown("## 📰 Article → 🖼️ Image Generator")
         gr.Markdown("Nhập bài viết → Sinh tiêu đề → Tối ưu prompt → Sinh ảnh minh họa tự động")
 
+        # State để lưu trữ trạng thái tạm thời
+        state = gr.State(value=None)
+
         with gr.Row():
             article_input = gr.Textbox(label="📄 Bài viết", lines=10, placeholder="Dán nội dung bài viết ở đây...")
             style_dropdown = gr.Dropdown(choices=["Art", "Anime", "Watercolor", "Cyberpunk"], label="🎨 Phong cách ảnh", value="Art")
@@ -164,10 +175,11 @@ def create_app():
 
         feedback = gr.Radio(["👍 Hài lòng", "👎 Không hài lòng"], label="📊 Bạn có hài lòng với kết quả không?", value=None)
 
+        # Gắn sự kiện nút submit với hàm process từng bước
         submit_button.click(
-            fn=process,
-            inputs=[article_input, style_dropdown],
-            outputs=[title_output, prompt_output, image_output, download_button]
+            fn=process_step_by_step,
+            inputs=[article_input, style_dropdown, state],
+            outputs=[state, title_output, prompt_output, image_output, download_button]
         )
 
     return demo