Update app.py

app.py

@@ -85,6 +85,30 @@ def save_metadata_to_file(image_input, prompt, neg_prompt, model, loras, seed, s
     return txt_file, gr.Info(f"Đang tải file metadata.txt...", duration=2)
 
 
+    
+def read_image_metadata(image_path):
+    try:
+        if not image_path:
+            raise ValueError("Không có ảnh được tải lên.")
+        
+        with Image.open(image_path) as img:
+            prompt_info = parser_manager.parse(img)
+            
+            if not prompt_info:
+                raise ValueError("Không thể đọc thông tin từ ảnh.")
+
+            # Kiểm tra loại generator
+            if prompt_info.generator == "AUTOMATIC1111":
+                return handle_automatic1111(prompt_info)
+            elif prompt_info.generator == "ComfyUI":
+                return handle_comfyui(prompt_info)
+            else:
+                raise ValueError("Loại generator không được hỗ trợ.")
+                
+    except Exception as e:
+        print(f"Lỗi khi xử lý ảnh: {str(e)}")  
+        return "Không tìm thấy thông tin!", "", "", "", "", "", "", ""
+
 def handle_automatic1111(prompt_info):
     prompt = prompt_info.full_prompt
     negative_prompt = prompt_info.full_negative_prompt if prompt_info.full_negative_prompt else "N/A"