Update gui.py

gui.py

@@ -355,6 +355,9 @@ def roformer_separator(audio, model_key, seg_size, override_seg_size, overlap, p
     if not audio:
         raise ValueError("No audio file provided.")
     
+    # override_seg_size string olarak geliyor, boolean'a çevir
+    override_seg_size = override_seg_size == "True"
+    
     # output klasörünü temizle
     if os.path.exists(output_dir):
         shutil.rmtree(output_dir)
@@ -400,6 +403,9 @@ def auto_ensemble_process(audio, model_keys, seg_size, overlap, out_format, use_
     if not audio or not model_keys:
         raise ValueError("Audio or models missing.")
     
+    # use_tta string olarak geliyor, boolean'a çevir
+    use_tta = use_tta == "True"
+    
     # output klasörünü temizle
     if os.path.exists(output_dir):
         shutil.rmtree(output_dir)
@@ -505,7 +511,12 @@ def create_interface():
                         roformer_overlap = gr.Slider(2, 10, value=8, step=1, label="🔄 Overlap", interactive=True)
                     with gr.Row():
                         roformer_pitch_shift = gr.Slider(-12, 12, value=0, step=1, label="🎵 Pitch Shift", interactive=True)
-                        roformer_override_seg_size = gr.Checkbox(value=False, label="🔧 Override Segment Size", interactive=True)
+                        roformer_override_seg_size = gr.Dropdown(
+                            choices=["True", "False"],
+                            value="False",
+                            label="🔧 Override Segment Size",
+                            interactive=True
+                        )
                     roformer_button = gr.Button("✂️ Separate Now!", variant="primary")
                     with gr.Row():
                         roformer_stem1 = gr.Audio(label="🎸 Stem 1", type="filepath", interactive=False)
@@ -526,7 +537,12 @@ def create_interface():
                     with gr.Row():
                         ensemble_seg_size = gr.Slider(32, 4000, value=256, step=32, label="📏 Segment Size", interactive=True)
                         ensemble_overlap = gr.Slider(2, 10, value=8, step=1, label="🔄 Overlap", interactive=True)
-                        ensemble_use_tta = gr.Checkbox(value=False, label="🔍 Use TTA", interactive=True)
+                        ensemble_use_tta = gr.Dropdown(
+                            choices=["True", "False"],
+                            value="False",
+                            label="🔍 Use TTA",
+                            interactive=True
+                        )
                     ensemble_method = gr.Dropdown(label="⚙️ Ensemble Method", choices=['avg_wave', 'median_wave', 'max_wave', 'min_wave', 'avg_fft', 'median_fft', 'max_fft', 'min_fft'], value='avg_wave', interactive=True)
                     ensemble_weights = gr.Textbox(label="⚖️ Weights", placeholder="e.g., 1.0, 1.0 (comma-separated)", interactive=True)
                     ensemble_button = gr.Button("🎛️ Run Ensemble!", variant="primary")
@@ -535,7 +551,7 @@ def create_interface():
 
         gr.HTML("<div class='footer'>Powered by Audio-Separator 🌟🎶 | Made with ❤️</div>")
 
-        # Event Handlers (Aynı kalıyor)
+        # Event Handlers
         roformer_category.change(update_roformer_models, inputs=[roformer_category], outputs=[roformer_model])
         download_roformer.click(fn=download_audio, inputs=[url_ro], outputs=[roformer_audio])
         roformer_button.click(
@@ -562,7 +578,11 @@ if __name__ == "__main__":
     parser.add_argument("--ngrok-token", type=str, default=None, help="Ngrok token for tunneling")
     args = parser.parse_args()
 
+    from pyngrok import ngrok
     app = create_interface()
-    app.launch(server_name="0.0.0.0", server_port=args.port, share=False)
-       
+    if args.ngrok_token:
+        ngrok.set_auth_token(args.ngrok_token)
+        public_url = ngrok.connect(args.port, bind_tls=True).public_url
+        print(f"Running on public URL: {public_url}")
+    app.launch(server_name="0.0.0.0", server_port=args.port)
     app.close()