Upload app.py with huggingface_hub

app.py

@@ -62,6 +62,27 @@ class FluxSpaceApp:
                     # LoRA Management
                     gr.Markdown("### LoRA Management")
                     
+                    # Pre-loaded LoRAs
+                    gr.Markdown("#### Pre-loaded LoRAs")
+                    preloaded_lora_selector = gr.Dropdown(
+                        choices=["T11-Ultra-Portrait-E04"],
+                        value=None,
+                        label="Select Pre-loaded LoRA",
+                        info="Load LoRAs directly from Hugging Face"
+                    )
+                    
+                    preloaded_lora_strength = gr.Slider(
+                        minimum=0.0,
+                        maximum=2.0,
+                        value=1.0,
+                        step=0.1,
+                        label="Pre-loaded LoRA Strength"
+                    )
+                    
+                    load_preloaded_lora_btn = gr.Button("Load Pre-loaded LoRA", variant="secondary")
+                    
+                    # Custom LoRA Upload
+                    gr.Markdown("#### Custom LoRA Upload")
                     lora_upload = gr.File(
                         label="Upload LoRA (.safetensors)",
                         file_types=[".safetensors"],
@@ -78,12 +99,11 @@ class FluxSpaceApp:
                         maximum=2.0,
                         value=1.0,
                         step=0.1,
-                        label="LoRA Strength",
-                        info="How strongly to apply the LoRA"
+                        label="Custom LoRA Strength"
                     )
                     
                     with gr.Row():
-                        load_lora_btn = gr.Button("Load LoRA", variant="secondary")
+                        load_lora_btn = gr.Button("Load Custom LoRA", variant="secondary")
                         unload_lora_btn = gr.Button("Unload LoRA", variant="stop")
                     
                     # LoRA Status
@@ -211,11 +231,39 @@ class FluxSpaceApp:
                 except Exception as e:
                     return f"Error: {str(e)}", "Error: Model loading failed"
             
+            def load_preloaded_lora_handler(lora_name, strength):
+                """Handle pre-loaded LoRA loading"""
+                try:
+                    if not lora_name:
+                        return "Error: No LoRA selected", "LoRAs: None loaded", []
+                    
+                    # Load pre-loaded LoRA
+                    success = self.model_manager.load_preloaded_lora(lora_name, strength)
+                    
+                    if success:
+                        # Get trigger words
+                        lora_info = self.model_manager.get_preloaded_loras().get(lora_name, {})
+                        trigger_words = lora_info.get('trigger_words', '')
+                        
+                        status_text = f"Pre-loaded LoRA Loaded: {lora_name}"
+                        if trigger_words:
+                            status_text += f" (Trigger: {trigger_words})"
+                        
+                        lora_status_text = f"LoRAs: {lora_name}"
+                        lora_list = [lora_name]
+                        
+                        return status_text, lora_status_text, lora_list
+                    else:
+                        return f"Error: Failed to load pre-loaded LoRA", "LoRAs: None loaded", []
+                        
+                except Exception as e:
+                    return f"Error: {str(e)}", "LoRAs: None loaded", []
+            
             def load_lora_handler(file, name, strength):
-                """Handle LoRA loading"""
+                """Handle custom LoRA loading"""
                 try:
                     if file is None:
-                        return "Error: No file uploaded", "LoRAs: None loaded"
+                        return "Error: No file uploaded", "LoRAs: None loaded", []
                     
                     file_path = file.name
                     lora_name = name if name else os.path.splitext(os.path.basename(file_path))[0]
@@ -235,7 +283,7 @@ class FluxSpaceApp:
                         # Update LoRA list
                         lora_list = list(self.lora_manager.loaded_loras.keys())
                         
-                        status_text = f"LoRA Loaded: {lora_name}"
+                        status_text = f"Custom LoRA Loaded: {lora_name}"
                         lora_status_text = f"LoRAs: {', '.join(lora_list)}"
                         
                         return status_text, lora_status_text, lora_list
@@ -298,6 +346,12 @@ LoRAs: {', '.join(gen_info['loras']) if gen_info['loras'] else 'None'}
                 outputs=[model_status, model_info]
             )
             
+            load_preloaded_lora_btn.click(
+                fn=load_preloaded_lora_handler,
+                inputs=[preloaded_lora_selector, preloaded_lora_strength],
+                outputs=[lora_status, lora_status, lora_list]
+            )
+            
             load_lora_btn.click(
                 fn=load_lora_handler,
                 inputs=[lora_upload, lora_name, lora_strength],