Add application file

app.py

@@ -83,9 +83,9 @@ button.primary:hover {
 }
 """
 
-def load_synthesizer():
-    # Status for loading
-    status_block.update("در حال بارگذاری مدل... لطفاً منتظر بمانید")
+def load_synthesizer(status_output):
+    # Update status message
+    status = "در حال بارگذاری مدل... لطفاً منتظر بمانید"
     
     try:
         # Download model files from Hugging Face Hub
@@ -107,50 +107,42 @@ def load_synthesizer():
             use_cuda=False  # Usually no GPU in free Spaces
         )
         
-        status_block.update("مدل با موفقیت بارگذاری شد! اکنون می‌توانید از سیستم استفاده کنید.")
-        return synthesizer
+        status = "مدل با موفقیت بارگذاری شد! اکنون می‌توانید از سیستم استفاده کنید."
+        return synthesizer, status
     
     except Exception as e:
         error_msg = f"خطا در بارگذاری مدل: {str(e)}"
-        status_block.update(f"❌ {error_msg}")
+        status = f"❌ {error_msg}"
         raise RuntimeError(error_msg)
 
-def tts(text, speed=1.0):
+def tts(text, speed, status_output):
     if not text.strip():
-        return None, "لطفاً متنی وارد کنید."
+        return None, "لطفاً متنی وارد کنید.", status_output.value
     
     try:
-        status_block.update("در حال تبدیل متن به گفتار...")
-        
-        # Show processing animation
-        for i in range(3):
-            time.sleep(0.3)
-            status_block.update(f"در حال پردازش{'.' * (i+1)}")
+        status = "در حال تبدیل متن به گفتار..."
         
         # Generate speech
         wav = synthesizer.tts(text, speed=speed)
         output_path = "output.wav"
         synthesizer.save_wav(wav, output_path)
         
-        status_block.update("✅ صدا با موفقیت تولید شد!")
-        return output_path, "تبدیل با موفقیت انجام شد."
+        status = "✅ صدا با موفقیت تولید شد!"
+        return output_path, "تبدیل با موفقیت انجام شد.", status
     
     except Exception as e:
         error_msg = f"خطا در تولید صدا: {str(e)}"
-        status_block.update(f"❌ {error_msg}")
-        return None, error_msg
-
-# Create a status block for feedback
-status_block = gr.Markdown("در حال آماده‌سازی سیستم...")
+        status = f"❌ {error_msg}"
+        return None, error_msg, status
 
-# First create the interface without the synthesizer
+# Create the interface
 with gr.Blocks(css=custom_css) as demo:
     with gr.Column(elem_classes="container"):
         gr.Markdown("# سامانه تبدیل متن فارسی به گفتار", elem_classes="main-header")
         
         # Status area
         with gr.Column(elem_classes="status-panel"):
-            status_output = gr.Markdown("", elem_id="status")
+            status_output = gr.Markdown("در حال آماده‌سازی سیستم...", elem_id="status")
         
         # Input panel
         with gr.Column(elem_classes="input-panel"):
@@ -200,25 +192,28 @@ with gr.Blocks(css=custom_css) as demo:
         )
         
         gr.Markdown(
-            "توسعه داده شده با استفاده از مدل VITS فارسی | [WaeliFatima/vits-fa-voice](https://huggingface.co/WaeliFatima/vits-fa-voice)",
+            "توسعه داده شده با استفاده از مدل VITS فارسی | [QomSSLab/vits-fa-voice](https://huggingface.co/QomSSLab/vits-fa-voice)",
             elem_classes="footer"
         )
 
-# Initialize the synthesizer
-try:
-    synthesizer = load_synthesizer()
+    # Initialize the synthesizer on app startup
+    synthesizer = None
+    
+    @demo.load
+    def init_synthesizer():
+        global synthesizer
+        try:
+            synthesizer, status = load_synthesizer(status_output)
+            return status_output.update(status)
+        except Exception as e:
+            return status_output.update(f"❌ خطا در بارگذاری مدل: {str(e)}")
+    
     # Connect the function to the button
     submit_btn.click(
         fn=tts,
-        inputs=[text_input, speed_slider],
-        outputs=[output_audio, result_text]
+        inputs=[text_input, speed_slider, status_output],
+        outputs=[output_audio, result_text, status_output]
     )
-    # Update the status block
-    status_block.update("سیستم آماده استفاده است!")
-    
-except Exception as e:
-    print(f"Error: {str(e)}")
-    status_block.update(f"❌ خطا در بارگذاری مدل: {str(e)}")
 
 # Launch the interface
 demo.launch()