Upload web-ui.py

web-ui.py

@@ -37,17 +37,25 @@ static_model_names = [
 	"stablediffusionapi/realcartoon3d",
 	"stablediffusionapi/disney-pixar-cartoon",
 	"stablediffusionapi/pastel-mix-stylized-anime",
-	"stablediffusionapi/anything-v5",
+	"stablediffusionapi/anything-v5",	
     "SG161222/Realistic_Vision_V2.0",
     "SG161222/Realistic_Vision_V4.0_noVAE",
     "SG161222/Realistic_Vision_V5.1_noVAE",
-	r"C:\Users\King\Downloads\New folder\3D Animation Diffusion"
+	#r"G:\model\model_diffusers"
 ]
 
 # Cache for loaded models
 model_cache = {}
 max_cache_size = args.cache_limit
 
+def convert_model(checkpoint_path, output_path):
+    try:
+        pipe = StableDiffusionPipeline.from_single_file(checkpoint_path)
+        pipe.save_pretrained(output_path)
+        return f"Model converted and saved to {output_path}"
+    except Exception as e:
+        return f"Error: {str(e)}"
+
 # Function to load and cache model
 def load_model(model_name):
     if model_name in model_cache:
@@ -167,6 +175,12 @@ with gr.Blocks() as demo:
         output_gallery = gr.Gallery(label="Generated Images")
         output_text = gr.Textbox(label="Output Info")
         display_seed = gr.Textbox(label="Used Seed", interactive=False)
+		
+    with gr.Row():
+        checkpoint_path_input = gr.Textbox(label="Enter Checkpoint File Path .e.g G:\model\model.safetensors", )
+        output_path_input = gr.Textbox(label="Enter Output Folder Path, e.g. G:\model\model_diffusers")
+        convert_btn = gr.Button("Convert Model")
+
 
     generate_btn.click(
         generate_image,
@@ -174,17 +188,26 @@ with gr.Blocks() as demo:
         outputs=[output_gallery, output_text, display_seed],
     )
 	
+    convert_btn.click(
+        convert_model,
+        inputs=[checkpoint_path_input, output_path_input],
+        outputs=[gr.Text(label="Conversion Status")],
+    )	
+	
+#sadly doesnt work
 def start_ngrok():
+    print("1")
     time.sleep(10)  # Delay for 10 seconds to ensure Gradio starts first
+    print("2")
     ngrok.set_auth_token(args.ngrok_token)
     public_url = ngrok.connect(port=7860)  # Adjust to your Gradio app's port
     print(f"ngrok tunnel started at {public_url}")
 
 if __name__ == "__main__":
-    if args.ngrok_token:
+    #if args.ngrok_token:
         # Start ngrok in a daemon thread with a delay
-        ngrok_thread = threading.Thread(target=start_ngrok, daemon=True)
-        ngrok_thread.start()
+    #    ngrok_thread = threading.Thread(target=start_ngrok, daemon=False)
+    #    ngrok_thread.start()
 
     # Launch the Gradio app
     demo.launch(share=args.share, inbrowser=True)