Update app.py

app.py

@@ -3,40 +3,78 @@ from safetensors.torch import save_file, load_file
 import gradio as gr
 import os
 
-def convert_embedding(uploaded_file):
-    output_path = "embedding.safetensors"
-    file_extension = os.path.splitext(uploaded_file.name)[1]
-
-    #The sample files are probably structured in these ways because the pt files were probably all created with automatic1111, and the safetensors files were probably created with kohya_ss
-    #If we learn of other programs that structure the embedding file differently, we'll have to adjust the logic.
-    if file_extension == '.pt':
-        sd15_embedding = torch.load(uploaded_file.name, map_location=torch.device('cpu'), weights_only=True)
-        sd15_tensor = sd15_embedding['string_to_param']['*']
-    elif file_extension == '.safetensors':
-        loaded_tensors = load_file(uploaded_file.name)
-        sd15_tensor = loaded_tensors['emb_params']
-    else:
-        raise ValueError("Unsupported file format")
+def convert_embedding(uploaded_files):
+    output_files = []
+    
+    for uploaded_file in uploaded_files:
+        file_name, file_extension = os.path.splitext(uploaded_file.name)
+        output_path = f"{file_name}_XL.safetensors"
+        
+        if file_extension == '.pt':
+            sd15_embedding = torch.load(uploaded_file.name, map_location=torch.device('cpu'))
+            sd15_tensor = sd15_embedding.get('string_to_param', {}).get('*')
+        elif file_extension == '.safetensors':
+            loaded_tensors = load_file(uploaded_file.name)
+            sd15_tensor = loaded_tensors.get('emb_params')
+        else:
+            raise ValueError(f"Unsupported file format: {file_extension}")
+        
+        if sd15_tensor is None:
+            raise ValueError(f"Invalid embedding structure in file: {uploaded_file.name}")
+        
+        num_vectors = sd15_tensor.shape[0]
+        clip_g_shape = (num_vectors, 1280)
+        clip_l_shape = (num_vectors, 768)
+        clip_g = torch.zeros(clip_g_shape, dtype=torch.float16)
+        clip_l = torch.zeros(clip_l_shape, dtype=torch.float16)
+        clip_l[:sd15_tensor.shape[0], :sd15_tensor.shape[1]] = sd15_tensor.to(dtype=torch.float16)
+        
+        save_file({"clip_g": clip_g, "clip_l": clip_l}, output_path)
+        output_files.append(output_path)
     
-    num_vectors = sd15_tensor.shape[0]
-    clip_g_shape = (num_vectors, 1280)
-    clip_l_shape = (num_vectors, 768)
-    clip_g = torch.zeros(clip_g_shape, dtype=torch.float16)
-    clip_l = torch.zeros(clip_l_shape, dtype=torch.float16)
-    clip_l[:sd15_tensor.shape[0], :sd15_tensor.shape[1]] = sd15_tensor.to(dtype=torch.float16)
-    save_file({"clip_g": clip_g, "clip_l": clip_l}, output_path)
+    return output_files
 
-    # Return the path to the converted file for download
-    return output_path
+custom_css = """
+    body {
+        background-color: #1e1e2e;
+        color: #ffffff;
+        font-family: Arial, sans-serif;
+    }
+    .gradio-container {
+        max-width: 800px;
+        margin: auto;
+        padding: 20px;
+        border-radius: 10px;
+        background: #2a2a3a;
+        box-shadow: 0 0 10px rgba(0, 0, 0, 0.3);
+    }
+    .gradio-container h1 {
+        text-align: center;
+        font-size: 24px;
+    }
+    .gradio-container button {
+        background-color: #4caf50;
+        color: white;
+        padding: 10px 15px;
+        border: none;
+        border-radius: 5px;
+        cursor: pointer;
+        font-size: 16px;
+    }
+    .gradio-container button:hover {
+        background-color: #45a049;
+    }
+"""
 
 iface = gr.Interface(
     fn=convert_embedding,
-    inputs=gr.File(label="Upload SD-1.5 embedding"),
-    outputs=gr.File(label="Download converted SDXL safetensors embedding"),
-    title="SD-1.5 to SDXL Embedding Converter | Now running on Gradio 5 ⚡",
-    description="Upload an SD-1.5 embedding file to convert it to SDXL."
+    inputs=gr.File(label="Upload SD-1.5 embeddings", type="file", multiple=True),
+    outputs=gr.File(label="Download converted SDXL safetensors embeddings", type="file", multiple=True),
+    title="✨ SD-1.5 to SDXL Embedding Converter | Now supports multiple files ⚡",
+    description="Upload one or more SD-1.5 embedding files to convert them to SDXL. Stylish and efficient!",
+    theme="default",
+    css=custom_css
 )
 
 if __name__ == "__main__":
     iface.launch()
-