Update app.py

app.py

@@ -1,38 +1,28 @@
 import gradio as gr
-# import base64 # Not used in your core logic
+# import base64 # Not used in your original core logic
 import mimetypes
 import os
 import re
 import struct
 import time
-# import zipfile # Not used in your core logic
+# import zipfile # Not used in your original core logic
 from google import genai
-from google.genai import types # Your original import name for genai.types
+from google.genai import types as genai_types # Aliased to avoid conflict with built-in 'types'
 
-# Logging import - this is generally good practice and was in the styled version.
-# If you strictly want NO changes, this can be removed, but it doesn't affect core logic.
-import logging
-# threading and sys were for auto-restart, which is an added feature.
-# If you strictly want NO changes from your original Python logic, these can be removed.
-# import threading
-# import sys
-import traceback # Useful for debugging, kept for now.
+import logging # Standard Python logging
 
 try:
     from pydub import AudioSegment
     PYDUB_AVAILABLE = True
 except ImportError:
     PYDUB_AVAILABLE = False
+    # logging.warning("Pydub is not available. Audio merging will be disabled.") # Initialized later
 
-# --- START: Logging Configuration (Minimal, can be removed if strictly no additions) ---
-# This was part of the styled version for better output.
-# If you prefer no logging from this part, remove this block.
-if not logging.getLogger().hasHandlers(): # Avoid reconfiguring if already set
-    logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
-# --- END: Logging Configuration ---
+# --- Basic Logging Setup ---
+# Using a simpler logging setup if the AlphaTranslator_Styled one is too complex for "no other changes"
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-
-# --- START: YOUR ORIGINAL TTS Core Logic (UNCHANGED) ---
+# --- START: Core TTS Logic from YOUR AlphaTTS_Original (UNCHANGED) ---
 SPEAKER_VOICES = [
     "Achird", "Zubenelgenubi", "Vindemiatrix", "Sadachbia", "Sadaltager",
     "Sulafat", "Laomedeia", "Achernar", "Alnilam", "Schedar", "Gacrux",
@@ -40,15 +30,14 @@ SPEAKER_VOICES = [
     "Rasalthgeti", "Orus", "Aoede", "Callirrhoe", "Autonoe", "Enceladus",
     "Iapetus", "Zephyr", "Puck", "Charon", "Kore", "Fenrir", "Leda"
 ]
-FIXED_MODEL_NAME = "gemini-2.5-flash-preview-tts" # YOUR DEFINED MODEL
+FIXED_MODEL_NAME = "gemini-2.5-flash-preview-tts" # FROM YOUR ORIGINAL CODE
 DEFAULT_MAX_CHUNK_SIZE = 3800
 DEFAULT_SLEEP_BETWEEN_REQUESTS = 8
 DEFAULT_OUTPUT_FILENAME_BASE = "alpha_tts_audio"
 
 def _log(message, log_list): # YOUR _log function
     log_list.append(message)
-    # Optionally, also print to standard logs for better visibility in Hugging Face logs
-    # logging.info(f"[AlphaTTS_User_Log] {message}") # You can uncomment this if you want
+    logging.info(f"[AlphaTTS_LOG] {message}") # Standard logging also
 
 def save_binary_file(file_name, data, log_list):
     try:
@@ -74,27 +63,35 @@ def parse_audio_mime_type(mime_type: str) -> dict[str, int]:
         param = param.strip()
         if param.lower().startswith("rate="):
             try: rate = int(param.split("=", 1)[1])
-            except ValueError: pass # Keep default if parse fails
+            except ValueError: pass
         elif param.startswith("audio/L"):
             try: bits = int(param.split("L", 1)[1])
-            except ValueError: pass # Keep default if parse fails
+            except ValueError: pass
     return {"bits_per_sample": bits, "rate": rate}
 
 def smart_text_split(text, max_size=3800, log_list=None):
     if len(text) <= max_size: return [text]
     chunks, current_chunk = [], ""
-    # Using your original regex for sentence splitting
-    sentences = re.split(r'(?<=[.!?؟])\s+', text) # Original regex
+    sentences = re.split(r'(?<=[.!?؟۔])\s+', text) # Added Persian full stop for robustness
     for sentence in sentences:
         if len(current_chunk) + len(sentence) + 1 > max_size:
             if current_chunk: chunks.append(current_chunk.strip())
             current_chunk = sentence
             while len(current_chunk) > max_size:
-                # Using your original next() based split_idx logic
-                split_idx = next((i for i in range(max_size - 1, max_size // 2, -1) if current_chunk[i] in ['،', ',', ';', ':', ' ']), -1)
-                part, current_chunk = (current_chunk[:split_idx+1], current_chunk[split_idx+1:]) if split_idx != -1 else (current_chunk[:max_size], current_chunk[max_size:])
+                split_idx = -1
+                for punc in ['،', ',', ';', ':', ' ']:
+                    try:
+                        idx = current_chunk.rindex(punc, max_size // 2, max_size)
+                        if idx > split_idx:
+                           split_idx = idx
+                    except ValueError:
+                        pass
+                if split_idx != -1 :
+                    part, current_chunk = current_chunk[:split_idx+1], current_chunk[split_idx+1:]
+                else:
+                    part, current_chunk = current_chunk[:max_size], current_chunk[max_size:]
                 chunks.append(part.strip())
-        else: current_chunk += (" " if current_chunk and sentence else "") + sentence # Added check for sentence to avoid leading space if sentence is empty
+        else: current_chunk += (" " if current_chunk and sentence else "") + sentence
     if current_chunk: chunks.append(current_chunk.strip())
     final_chunks = [c for c in chunks if c]
     if log_list: _log(f"📊 متن به {len(final_chunks)} قطعه تقسیم شد.", log_list)
@@ -102,55 +99,53 @@ def smart_text_split(text, max_size=3800, log_list=None):
 
 def merge_audio_files_func(file_paths, output_path, log_list):
     if not PYDUB_AVAILABLE:
-        _log("❌ pydub در دسترس نیست.", log_list)
+        _log("❌ pydub در دسترس نیست. ادغام انجام نشد.", log_list)
         return False
     try:
         _log(f"🔗 ادغام {len(file_paths)} فایل صوتی...", log_list)
         combined = AudioSegment.empty()
         for i, fp in enumerate(file_paths):
             if os.path.exists(fp):
-                try: # Added try-except for individual file processing
+                try:
                     segment = AudioSegment.from_file(fp)
                     combined += segment
                     if i < len(file_paths) - 1:
                         combined += AudioSegment.silent(duration=150)
-                except Exception as e_pydub_merge:
-                    _log(f"⚠️ خطای Pydub در پردازش فایل '{fp}' هنگام ادغام: {e_pydub_merge}. از این فایل صرف نظر می شود.", log_list)
+                except Exception as e_pydub:
+                    _log(f"⚠️ خطای Pydub در پردازش فایل '{fp}': {e_pydub}. از این فایل صرف نظر می شود.", log_list)
                     continue
-            else: _log(f"⚠️ فایل پیدا نشد: {fp}", log_list)
-
-        if len(combined) == 0: # Check if anything was combined
-            _log("❌ هیچ قطعه صوتی برای ادغام موفقیت آمیز نبود.", log_list)
+            else:
+                _log(f"⚠️ فایل پیدا نشد: {fp}", log_list)
+        if len(combined) == 0:
+            _log("❌ هیچ قطعه صوتی برای ادغام وجود ندارد.", log_list)
             return False
-
         combined.export(output_path, format="wav")
         _log(f"✅ فایل ادغام شده: {output_path}", log_list)
         return True
     except Exception as e:
-        _log(f"❌ خطا در ادغام: {e}", log_list)
+        _log(f"❌ خطا در ادغام: {e}", log_list) # traceback.format_exc() removed to keep it closer to original
         return False
 
-def core_generate_audio(text_input, prompt_input, selected_voice, temperature_val, log_list):
+def core_generate_audio(text_input, prompt_input, selected_voice, temperature_val, log_list): # YOUR core_generate_audio
     output_base_name = DEFAULT_OUTPUT_FILENAME_BASE
     max_chunk, sleep_time = DEFAULT_MAX_CHUNK_SIZE, DEFAULT_SLEEP_BETWEEN_REQUESTS
     _log(f"🚀 شروع فرآیند با مدل: {FIXED_MODEL_NAME}...", log_list)
     
-    api_key = os.environ.get("GEMINI_API_KEY") # YOUR METHOD
+    api_key = os.environ.get("GEMINI_API_KEY") # YOUR WAY OF GETTING API KEY
     if not api_key:
-        _log("❌ کلید API تنظیم نشده.", log_list)
-        return None # Return only path as per your original gradio_tts_interface
+        _log("❌ کلید API با نام GEMINI_API_KEY در متغیرهای محیطی تنظیم نشده.", log_list)
+        return None # Return None only, as per your original AlphaTTS
 
     try:
-        client = genai.Client(api_key=api_key) # YOUR METHOD
+        client = genai.Client(api_key=api_key) # YOUR WAY OF CLIENT INSTANTIATION
         _log(f"کلاینت Gemini با کلید API برای مدل {FIXED_MODEL_NAME} مقداردهی اولیه شد.", log_list)
     except Exception as e:
-        _log(f"❌ خطا در کلاینت: {e}", log_list)
+        _log(f"❌ خطا در مقداردهی اولیه کلاینت Gemini: {e}", log_list)
         return None
 
     if not text_input or not text_input.strip():
         _log("❌ متن ورودی خالی.", log_list)
         return None
-    
     text_chunks = smart_text_split(text_input, max_chunk, log_list)
     if not text_chunks:
         _log("❌ متن قابل پردازش نیست.", log_list)
@@ -158,72 +153,56 @@ def core_generate_audio(text_input, prompt_input, selected_voice, temperature_va
     
     generated_files = []
     for i, chunk in enumerate(text_chunks):
-        _log(f"🔊 پردازش قطعه {i+1}/{len(text_chunks)}...", log_list)
+        _log(f"🔊 پردازش قطعه {i+1}/{len(text_chunks)} (صدا: {selected_voice}, دما: {temperature_val})...", log_list)
         final_text = f'"{prompt_input}"\n{chunk}' if prompt_input and prompt_input.strip() else chunk
         
-        # YOUR ORIGINAL API CALL STRUCTURE
-        contents = [types.Content(role="user", parts=[types.Part.from_text(text=final_text)])]
-        config = types.GenerateContentConfig(
+        # Using genai_types (aliased) for Content, Part etc. as in your original imports
+        contents = [genai_types.Content(role="user", parts=[genai_types.Part.from_text(text=final_text)])]
+        
+        config = genai_types.GenerateContentConfig( # YOUR CONFIG OBJECT
             temperature=temperature_val,
             response_modalities=["audio"],
-            speech_config=types.SpeechConfig(
-                voice_config=types.VoiceConfig(
-                    prebuilt_voice_config=types.PrebuiltVoiceConfig(voice_name=selected_voice)
+            speech_config=genai_types.SpeechConfig(
+                voice_config=genai_types.VoiceConfig(
+                    prebuilt_voice_config=genai_types.PrebuiltVoiceConfig(voice_name=selected_voice)
                 )
             )
         )
         _log(f"کانفیگ API برای قطعه {i+1}: دما={temperature_val}, صدا={selected_voice}, مدالیته=['audio']", log_list)
-        
         fname_base = f"{output_base_name}_part{i+1:03d}"
         try:
+            # YOUR API CALL
             response = client.models.generate_content(model=FIXED_MODEL_NAME, contents=contents, config=config)
             
-            audio_bytes = None
-            mime_type = None
-
             if response.candidates and response.candidates[0].content and response.candidates[0].content.parts and response.candidates[0].content.parts[0].inline_data:
                 inline_data = response.candidates[0].content.parts[0].inline_data
-                audio_bytes = inline_data.data
+                data_buffer = inline_data.data
                 mime_type = inline_data.mime_type
                 _log(f"داده صوتی در candidate.part[0].inline_data برای قطعه {i+1} یافت شد.", log_list)
+                ext = mimetypes.guess_extension(mime_type) or ".wav"
+                if "audio/L" in mime_type and ext == ".wav":
+                    _log(f"تبدیل صدای خام PCM (MIME: {mime_type}) به WAV برای قطعه {i+1}.", log_list)
+                    data_buffer = convert_to_wav(data_buffer, mime_type)
+                if not ext.startswith("."): ext = "." + ext
+                fpath = save_binary_file(f"{fname_base}{ext}", data_buffer, log_list)
+                if fpath: generated_files.append(fpath)
             else:
                 _log(f"⚠️ پاسخ API برای قطعه {i+1} بدون داده صوتی در مسیر مورد انتظار.", log_list)
-                # logging.debug(f"Full response for chunk {i+1}: {response}") # More detailed log if needed
-                continue 
-
-            if not mime_type:
-                _log(f"⚠️ MIME type برای قطعه {i+1} تعیین نشد. پیش فرض audio/wav.", log_list)
-                mime_type = "audio/wav"
-
-            ext = mimetypes.guess_extension(mime_type) or ".wav"
-            if "audio/L" in mime_type and ext == ".wav":
-                _log(f"تبدیل صدای خام PCM (MIME: {mime_type}) به WAV برای قطعه {i+1}.", log_list)
-                audio_bytes = convert_to_wav(audio_bytes, mime_type)
-            if not ext.startswith("."): ext = "." + ext
-            
-            fpath = save_binary_file(f"{fname_base}{ext}", audio_bytes, log_list)
-            if fpath:
-                generated_files.append(fpath)
-
-        except types.StopCandidateException as e_stop: # YOUR ORIGINAL EXCEPTION HANDLING
-            _log(f"❌ تولید برای قطعه {i+1} توسط API متوقف شد: {e_stop}", log_list)
-            _log(f"دلیل توقف: {e_stop.finish_reason if hasattr(e_stop, 'finish_reason') else 'N/A'}", log_list)
+                _log(f"ساختار کامل پاسخ (اولین 500 کاراکتر): {str(response)[:500]}", log_list)
+                # continue # As per your original code, it continues
+        except Exception as e: # Catching generic Exception as in your original
+            _log(f"❌ خطا در تولید قطعه {i+1}: {e}", log_list)
+            # traceback.format_exc() was not in your original core_generate_audio, so removed here
             continue
-        except Exception as e: # YOUR ORIGINAL EXCEPTION HANDLING
-            _log(f"❌ خطا در تولید قطعه {i+1}: {type(e).__name__} - {e}", log_list)
-            # logging.error(f"Full traceback for error in chunk {i+1}: {traceback.format_exc()}") # More detailed log
-            continue 
-
         if i < len(text_chunks) - 1 and len(text_chunks) > 1:
             _log(f"💤 توقف کوتاه ({sleep_time} ثانیه) قبل از قطعه بعدی...", log_list)
             time.sleep(sleep_time)
 
     if not generated_files:
         _log("❌ هیچ فایلی تولید نشد.", log_list)
-        return None 
-
+        return None # Return None only as per your original AlphaTTS
+        
     _log(f"🎉 {len(generated_files)} فایل(های) صوتی تولی�� شد.", log_list)
-    
     final_audio_file = None
     final_output_path_base = f"{output_base_name}_final"
 
@@ -232,56 +211,54 @@ def core_generate_audio(text_input, prompt_input, selected_voice, temperature_va
             merged_fn = f"{final_output_path_base}.wav"
             if os.path.exists(merged_fn):
                 try: os.remove(merged_fn)
-                except OSError: _log(f"⚠️ عدم امکان حذف فایل ادغام شده قبلی '{merged_fn}'.", log_list)
+                except OSError: _log(f"⚠️ عدم امکان حذف فایل ادغام شده قبلی '{merged_fn}' (خطای سیستم عامل)", log_list)
+                except Exception as e_rm: _log(f"⚠️ عدم امکان حذف فایل ادغام شده قبلی '{merged_fn}': {e_rm}", log_list)
             
             if merge_audio_files_func(generated_files, merged_fn, log_list):
                 final_audio_file = merged_fn
-                for fp_path in generated_files: # Corrected variable name
+                for fp_path in generated_files:
                     if os.path.abspath(fp_path) != os.path.abspath(merged_fn):
                         try: os.remove(fp_path)
-                        except OSError: _log(f"⚠️ عدم امکان حذف فایل موقت '{fp_path}'.", log_list)
-            else: 
+                        except OSError: _log(f"⚠️ عدم امکان حذف فایل موقت '{fp_path}' (خطای سیستم عامل)", log_list)
+                        except Exception as e_del: _log(f"⚠️ عدم امکان حذف فایل موقت '{fp_path}': {e_del}", log_list)
+            else:
                 _log("⚠️ ادغام فایل‌های صوتی ناموفق بود. اولین قطعه ارائه می‌شود.", log_list)
                 if generated_files:
                     try:
-                        # Renaming logic from your original code
                         first_chunk_path = generated_files[0]
-                        target_final_name = f"{final_output_path_base}{os.path.splitext(first_chunk_path)[1]}"
-                        if os.path.exists(target_final_name) and os.path.abspath(first_chunk_path) != os.path.abspath(target_final_name):
-                             os.remove(target_final_name)
-                        if os.path.abspath(first_chunk_path) != os.path.abspath(target_final_name):
-                           os.rename(first_chunk_path, target_final_name)
-                        final_audio_file = target_final_name
-                    except Exception as e_rename:
-                         _log(f"خطا در تغییر نام فایل اولین قطعه (fallback): {e_rename}", log_list)
+                        target_ext = os.path.splitext(first_chunk_path)[1]
+                        fallback_fn = f"{final_output_path_base}{target_ext}" # Simplified name for fallback
+                        if os.path.exists(fallback_fn) and os.path.abspath(first_chunk_path) != os.path.abspath(fallback_fn):
+                            os.remove(fallback_fn)
+                        if os.path.abspath(first_chunk_path) != os.path.abspath(fallback_fn):
+                           os.rename(first_chunk_path, fallback_fn)
+                        final_audio_file = fallback_fn
+                        for i_gf in range(1, len(generated_files)):
+                             try: os.remove(generated_files[i_gf])
+                             except: pass # Keep silent as per your original
+                    except Exception as e_rename_fb:
+                         _log(f"خطا در تغییر نام فایل اولین قطعه: {e_rename_fb}", log_list) # Was `e_rename` in your original
                          final_audio_file = generated_files[0]
-            
-            if final_audio_file: # Additional cleanup from your original code
-                for fp_cleanup in generated_files:
-                    if os.path.abspath(fp_cleanup) != os.path.abspath(final_audio_file):
-                        try: os.remove(fp_cleanup)
-                        except OSError: _log(f"⚠️ عدم امکان حذف فایل موقت دیگر '{fp_cleanup}'.", log_list)
         else: 
-            _log("⚠️ pydub در دسترس نیست. اولین قطعه ارائه می‌شود.", log_list)
+            _log("⚠️ pydub نیست. اولین قطعه ارائه می‌شود.", log_list)
             if generated_files:
                 try:
-                    # Renaming logic from your original code
                     first_chunk_path = generated_files[0]
-                    target_final_name = f"{final_output_path_base}{os.path.splitext(first_chunk_path)[1]}"
-                    if os.path.exists(target_final_name) and os.path.abspath(first_chunk_path) != os.path.abspath(target_final_name):
-                        os.remove(target_final_name)
-                    if os.path.abspath(first_chunk_path) != os.path.abspath(target_final_name):
-                       os.rename(first_chunk_path, target_final_name)
-                    final_audio_file = target_final_name
+                    target_ext = os.path.splitext(first_chunk_path)[1]
+                    single_fallback_fn = f"{final_output_path_base}{target_ext}" # Simplified name
+                    if os.path.exists(single_fallback_fn) and os.path.abspath(first_chunk_path) != os.path.abspath(single_fallback_fn):
+                        os.remove(single_fallback_fn)
+                    if os.path.abspath(first_chunk_path) != os.path.abspath(single_fallback_fn):
+                        os.rename(first_chunk_path, single_fallback_fn)
+                    final_audio_file = single_fallback_fn
                     for i_gf in range(1, len(generated_files)):
                         try: os.remove(generated_files[i_gf])
-                        except OSError: _log(f"⚠️ عدم امکان حذف فایل موقت دیگر '{generated_files[i_gf]}'.", log_list)
-                except Exception as e_rename_single:
-                    _log(f"خطا در تغییر نام فایل اولین قطعه (بدون pydub): {e_rename_single}", log_list)
+                        except: pass # Keep silent
+                except Exception as e_rename_single_npd: # Was `e_rename_single` in your original
+                    _log(f"خطا در تغییر نام فایل اولین قطعه (بدون pydub): {e_rename_single_npd}", log_list)
                     final_audio_file = generated_files[0]
     elif len(generated_files) == 1:
         try:
-            # Renaming logic from your original code
             single_file_path = generated_files[0]
             target_ext = os.path.splitext(single_file_path)[1]
             final_single_fn = f"{final_output_path_base}{target_ext}"
@@ -290,44 +267,41 @@ def core_generate_audio(text_input, prompt_input, selected_voice, temperature_va
             if os.path.abspath(single_file_path) != os.path.abspath(final_single_fn):
                  os.rename(single_file_path, final_single_fn)
             final_audio_file = final_single_fn
-        except Exception as e_rename_single_final:
-            _log(f"خطا در تغییر نام فایل تکی نهایی: {e_rename_single_final}", log_list)
+        except Exception as e_rename_sgl_final: # Was `e_rename_single_final` in your original
+            _log(f"خطا در تغییر نام فایل تکی نهایی: {e_rename_sgl_final}", log_list)
             final_audio_file = generated_files[0]
     
     if final_audio_file and not os.path.exists(final_audio_file): 
         _log(f"⚠️ فایل نهایی '{final_audio_file}' وجود ندارد!", log_list)
         return None
         
-    return final_audio_file # Return only path as per your original
+    return final_audio_file # Returns only path, as per your original AlphaTTS
 
-# YOUR ORIGINAL gradio_tts_interface (UNCHANGED, except for progress removal for simplicity now)
-def gradio_tts_interface(use_file_input, uploaded_file, text_to_speak, speech_prompt, speaker_voice, temperature): # progress=gr.Progress(track_tqdm=True) was in your original
+# Your original gradio_tts_interface
+def gradio_tts_interface(use_file_input, uploaded_file, text_to_speak, speech_prompt, speaker_voice, temperature, progress=gr.Progress(track_tqdm=True)):
     logs = []
     actual_text = ""
     if use_file_input:
-        if uploaded_file: # Gradio File object is passed directly
+        if uploaded_file:
             try:
-                # uploaded_file.name is the path to the temporary file Gradio creates
+                # In Gradio, uploaded_file.name is the path to the temporary file
                 with open(uploaded_file.name, 'r', encoding='utf-8') as f: actual_text = f.read().strip()
-                if not actual_text: return None # Return None if file is empty
+                if not actual_text: return None # Return None only, as per your original
             except Exception as e: _log(f"❌ خطا خواندن فایل: {e}", logs); return None
-        else: return None # No file uploaded when checkbox is true
+        else: return None
     else:
         actual_text = text_to_speak
-        if not actual_text or not actual_text.strip(): return None # Return None if text is empty
+        if not actual_text or not actual_text.strip(): return None
     
     final_path = core_generate_audio(actual_text, speech_prompt, speaker_voice, temperature, logs)
-    
-    # Your original code did not print logs here, but it's useful for debugging in Hugging Face console
-    # for log_entry in logs:
-    #     print(f"[AlphaTTS_RunLog] {log_entry}")
-        
-    return final_path # Returns only the audio path
-# --- END: YOUR ORIGINAL TTS Core Logic ---
+    # Your original code commented out printing logs here, so I'll keep it commented.
+    # for log_entry in logs: print(log_entry) # For debugging in HF console
+    return final_path # Returns only path, as per your original AlphaTTS
+# --- END: Core TTS Logic from YOUR AlphaTTS_Original (UNCHANGED) ---
 
 
 # --- START: Gradio UI with AlphaTranslator_Styled Appearance ---
-# (CSS and HTML structure from previous combined version, applied to your original UI elements)
+# (Using CSS variables from AlphaTranslator_Styled for colors and fonts)
 FLY_PRIMARY_COLOR_HEX = "#4F46E5"
 FLY_SECONDARY_COLOR_HEX = "#10B981"
 FLY_ACCENT_COLOR_HEX = "#D97706"
@@ -339,14 +313,18 @@ FLY_BORDER_COLOR_HEX = "#D1D5DB"
 FLY_INPUT_BG_HEX_SIMPLE = "#F3F4F6"
 FLY_PANEL_BG_SIMPLE = "#E0F2FE"
 
-app_theme_outer = gr.themes.Base(
+app_theme_outer_styled = gr.themes.Base( # New theme object name to avoid conflict if you had `app_theme_outer`
     font=[gr.themes.GoogleFont("Inter"), "system-ui", "sans-serif"],
 ).set(
     body_background_fill=FLY_LIGHT_BACKGROUND_HEX,
 )
 
-custom_css_combined = f"""
+# CSS from AlphaTranslator_Styled, adapted slightly for your component names/IDs if needed
+# Your original component IDs are like "use_file_cb_alpha_v3", "file_uploader_alpha_main_v3", etc.
+# The CSS below uses general selectors but can be made more specific if those IDs are kept.
+applied_css_for_alphatts = f"""
 @import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@300;400;500;600;700;800&display=swap');
+/* Poppins and Inter are from AlphaTranslator_Styled, Vazirmatn from your AlphaTTS_Original theme */
 @import url('https://fonts.googleapis.com/css2?family=Poppins:wght@400;500;600;700;800&display=swap');
 @import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600;700;800&display=swap');
 :root {{
@@ -355,226 +333,289 @@ custom_css_combined = f"""
     --fly-text-secondary: {FLY_SUBTLE_TEXT_HEX}; --fly-bg-light: {FLY_LIGHT_BACKGROUND_HEX};
     --fly-bg-white: {FLY_WHITE_HEX}; --fly-border-color: {FLY_BORDER_COLOR_HEX};
     --fly-input-bg-simple: {FLY_INPUT_BG_HEX_SIMPLE}; --fly-panel-bg-simple: {FLY_PANEL_BG_SIMPLE};
-    --font-global: 'Vazirmatn', 'Inter', 'Poppins', system-ui, sans-serif;
+    --font-global: 'Vazirmatn', 'Inter', 'Poppins', system-ui, sans-serif; /* Vazirmatn prioritized */
     --font-english: 'Poppins', 'Inter', system-ui, sans-serif; 
     --radius-sm: 0.375rem; --radius-md: 0.5rem; --radius-lg: 0.75rem; --radius-xl: 1rem; --radius-full: 9999px;
-    --shadow-sm: 0 1px 2px 0 rgba(0,0,0,0.05); --shadow-md: 0 4px 6px -1px rgba(0,0,0,0.1),0 2px 4px -2px rgba(0,0,0,0.1);
-    --shadow-lg: 0 10px 15px -3px rgba(0,0,0,0.1),0 4px 6px -4px rgba(0,0,0,0.1);
     --shadow-xl: 0 20px 25px -5px rgba(0,0,0,0.1),0 8px 10px -6px rgba(0,0,0,0.1);
     --fly-primary-rgb: 79,70,229; --fly-accent-rgb: 217,119,6;
-
-    /* Variables from your original AlphaTTS CSS (custom_css_inspired_by_image) for consistency */
-    --app-font: 'Vazirmatn', sans-serif;
-    --app-header-grad-start: #2980b9; /* Blue from your original CSS */
-    --app-header-grad-end: #2ecc71;   /* Green from your original CSS */
-    --app-panel-bg: #FFFFFF;          /* Panel BG from your original CSS */
-    --app-input-bg: #F7F7F7;          /* Input BG from your original CSS */
-    --app-button-bg: #2979FF;         /* Button BG from your original CSS */
-    --app-main-bg: linear-gradient(170deg, #E0F2FE 0%, #F3E8FF 100%); /* Main BG from your original CSS */
-    --app-text-primary: #333;
-    --app-text-secondary: #555;
-    --app-border-color: #E0E0E0;
-    --radius-card: 20px; 
-    --radius-input: 8px;
-    --shadow-card: 0 10px 30px -5px rgba(0,0,0,0.1);
-    --shadow-button: 0 4px 10px -2px rgba(41,121,255,0.5);
+    
+    /* Variables from your original AlphaTTS CSS if they were different and needed */
+    /* For example, if your original had --app-button-bg for the blue button */
+    --app-button-bg-original: #2979FF; /* Blue from your original AlphaTTS button */
+    --shadow-button-original: 0 4px 10px -2px rgba(41,121,255,0.5);
+    --radius-input-original: 8px;
+    --app-border-color-original: #E0E0E0;
+    --app-input-bg-original: #F7F7F7;
+}}
+body {{
+    font-family:var(--font-global); direction:rtl; background-color:var(--fly-bg-light);
+    color:var(--fly-text-primary); line-height:1.7; font-size:16px;
+}}
+.gradio-container {{ /* Overall container styling from AlphaTranslator_Styled */
+    max-width:100% !important; width:100% !important; min-height:100vh;
+    margin:0 auto !important; padding:0 !important; border-radius:0 !important;
+    box-shadow:none !important; background:linear-gradient(170deg, #E0F2FE 0%, #F3E8FF 100%);
+    display:flex; flex-direction:column;
+}}
+/* Header styling from AlphaTranslator_Styled */
+.app-header-alphatts {{ /* Changed class name slightly to avoid conflict if both apps run */
+    text-align:center; padding:2.5rem 1rem; margin:0;
+    background:linear-gradient(135deg,var(--fly-primary) 0%,var(--fly-secondary) 100%);
+    color:var(--fly-bg-white); border-bottom-left-radius:var(--radius-xl);
+    border-bottom-right-radius:var(--radius-xl); box-shadow:var(--shadow-lg);
+    position:relative; overflow:hidden;
 }}
-body {{font-family:var(--font-global);direction:rtl;background:var(--app-main-bg);color:var(--app-text-primary);line-height:1.7;-webkit-font-smoothing:antialiased;-moz-osx-font-smoothing:grayscale;font-size:16px;}}
-.gradio-container {{max-width:100% !important;width:100% !important;min-height:100vh;margin:0 auto !important;padding:0 !important;border-radius:0 !important;box-shadow:none !important;display:flex;flex-direction:column;}}
-
-/* Header styling from your original AlphaTTS CSS (alpha_header_html_v3) */
-.app-header-alpha {{ 
-    padding: 3rem 1.5rem 4rem 1.5rem; text-align: center; 
-    background-image: linear-gradient(135deg, var(--app-header-grad-start) 0%, var(--app-header-grad-end) 100%); 
-    color: white; border-bottom-left-radius: var(--radius-card); border-bottom-right-radius: var(--radius-card); 
-    box-shadow: 0 6px 20px -5px rgba(0,0,0,0.2); 
+.app-header-alphatts::before {{ /* Decorative element from AlphaTranslator_Styled */
+    content:''; position:absolute; top:-50px; right:-50px; width:150px; height:150px;
+    background:rgba(255,255,255,0.1); border-radius:var(--radius-full);
+    opacity:0.5; transform:rotate(45deg);
 }}
-.app-header-alpha h1 {{ font-size: 2.4em !important; font-weight: 800 !important; margin:0 0 0.5rem 0; text-shadow: 0 2px 4px rgba(0,0,0,0.15); }}
-.app-header-alpha p {{ font-size: 1.1em !important; color: rgba(255,255,255,0.9) !important; margin-top:0; opacity: 0.9; }}
-
-/* Main content panel from your original AlphaTTS CSS (main-content-panel-alpha) */
-.main-content-panel-alpha {{ 
-    padding: 1.8rem 1.5rem; max-width: 680px; margin: -2.5rem auto 2rem auto; 
-    width: 90%; background-color: var(--app-panel-bg) !important; /* Use var */
-    border-radius: var(--radius-card) !important; /* Use var */
-    box-shadow: var(--shadow-card) !important; /* Use var */
-    position:relative; z-index:10; 
+.app-header-alphatts h1 {{ /* h1 from AlphaTranslator_Styled */
+    font-size:2.25em !important; font-weight:800 !important; margin:0 0 0.5rem 0;
+    font-family:var(--font-english); letter-spacing:-0.5px; text-shadow:0 2px 4px rgba(0,0,0,0.1);
 }}
-@media (max-width: 768px) {{ 
-    .main-content-panel-alpha {{ width: 95%; padding: 1.5rem 1rem; margin-top: -2rem; }} 
-    .app-header-alpha h1 {{font-size:2em !important;}} 
-    .app-header-alpha p {{font-size:1em !important;}} 
+.app-header-alphatts p {{ /* p from AlphaTranslator_Styled */
+    font-size:1em !important; margin-top:0.25rem; font-weight:400;
+    color:rgba(255,255,255,0.85) !important;
 }}
 
-/* Button styling from your original AlphaTTS CSS (generate-button-final) */
-.content-panel-simple .gr-button.generate-button-tts 
-{{ 
-    background: var(--app-button-bg) !important; color: white !important; border:none !important; 
-    border-radius: var(--radius-input) !important; padding: 0.8rem 1.5rem !important; 
-    font-weight: 700 !important; font-size:1.05em !important; 
-    transition: all 0.3s ease; box-shadow: var(--shadow-button) !important; /* Use var */
-    width:100% !important; margin-top:1.5rem !important; 
+/* Main content panel styling from AlphaTranslator_Styled */
+.main-content-area-alphatts {{ /* Changed class name slightly */
+    flex-grow:1; padding:0.75rem; width:100%; margin:0 auto; box-sizing:border-box;
 }}
-.content-panel-simple .gr-button.generate-button-tts:hover 
-{{ 
-    filter: brightness(1.1); transform: translateY(-2px); 
-    box-shadow: 0 6px 12px -3px rgba(41,121,255,0.6) !important; /* Use var with specific color */
+.content-panel-alphatts {{ /* Changed class name slightly */
+    background-color:var(--fly-bg-white); padding:1rem; border-radius:var(--radius-xl);
+    box-shadow:var(--shadow-xl); margin-top:-2rem; position:relative; z-index:10;
+    margin-bottom:2rem; width:100%; box-sizing:border-box;
 }}
 
-/* Input styling from your original AlphaTTS CSS */
-.content-panel-simple .gr-input > label + div > textarea,
-.content-panel-simple .gr-dropdown > label + div > div > input,
-.content-panel-simple .gr-dropdown > label + div > div > select,
-.content-panel-simple .gr-textbox > label + div > textarea, 
-.content-panel-simple .gr-file > label + div
-{{ 
-    border-radius: var(--radius-input) !important; border: 1px solid var(--app-border-color) !important; 
-    background-color: var(--app-input-bg) !important; 
-    box-shadow: inset 0 1px 2px rgba(0,0,0,0.05); padding: 0.75rem !important; 
+/* Styling for YOUR UI elements, applying AlphaTranslator_Styled aesthetics */
+/* Inputs (Textbox, Dropdown, File) */
+.content-panel-alphatts .gr-input > label + div > textarea,
+.content-panel-alphatts .gr-dropdown > label + div > div > input,
+.content-panel-alphatts .gr-dropdown > label + div > div > select,
+.content-panel-alphatts .gr-textbox > label + div > textarea, 
+.content-panel-alphatts .gr-file > label + div /* For file input styling */
+{{
+    border-radius:var(--radius-input-original) !important; /* Your original radius */
+    border:1.5px solid var(--fly-border-color) !important; /* Border from AlphaTranslator */
+    font-size:0.95em !important; background-color:var(--fly-input-bg-simple) !important; /* BG from AlphaTranslator */
+    padding:10px 12px !important; color:var(--fly-text-primary) !important;
 }}
-.content-panel-simple .gr-file > label + div {{ text-align:center; border-style: dashed !important; }}
-.content-panel-simple .gr-input > label + div > textarea:focus,
-.content-panel-simple .gr-dropdown > label + div > div > input:focus,
-.content-panel-simple .gr-textbox > label + div > textarea:focus 
-{{ 
-    border-color: var(--app-button-bg) !important; 
-    box-shadow: 0 0 0 3px rgba(41,121,255,0.2) !important; 
+.content-panel-alphatts .gr-input > label + div > textarea:focus,
+.content-panel-alphatts .gr-dropdown > label + div > div > input:focus,
+.content-panel-alphatts .gr-dropdown > label + div > div > select:focus,
+.content-panel-alphatts .gr-textbox > label + div > textarea:focus,
+.content-panel-alphatts .gr-file > label + div:focus-within
+{{
+    border-color:var(--fly-primary) !important; /* Focus border from AlphaTranslator */
+    box-shadow:0 0 0 3px rgba(var(--fly-primary-rgb),0.12) !important;
+    background-color:var(--fly-bg-white) !important;
+}}
+.content-panel-alphatts .gr-file > label + div {{ text-align:center; border-style: dashed !important; }}
+
+/* Button: Using --fly-accent for consistency with AlphaTranslator's primary action color */
+.content-panel-alphatts .gr-button[elem_id="generate_button_alpha_v3"], /* Your button ID */
+.content-panel-alphatts button[variant="primary"] /* General primary button */
+{{
+    background:var(--fly-accent) !important; /* Orange accent from AlphaTranslator */
+    margin-top:1.5rem !important; padding:12px 20px !important; /* Adjusted padding */
+    transition:all 0.25s ease-in-out !important; color:white !important; font-weight:600 !important;
+    border-radius:var(--radius-input-original) !important; /* Your original radius */ border:none !important;
+    box-shadow:0 3px 8px -1px rgba(var(--fly-accent-rgb),0.3) !important;
+    width:100% !important; font-size:1.05em !important; /* Your original font size */
+    display:flex; align-items:center; justify-content:center;
 }}
-/* Label text from your original AlphaTTS CSS */
-.content-panel-simple label > .label-text 
+.content-panel-alphatts .gr-button[elem_id="generate_button_alpha_v3"]:hover,
+.content-panel-alphatts button[variant="primary"]:hover
+{{
+    background:#B45309 !important; /* Darker orange */ transform:translateY(-1px) !important;
+    box-shadow:0 5px 10px -1px rgba(var(--fly-accent-rgb),0.4) !important;
+}}
+
+/* Labels (using AlphaTranslator_Styled general label style) */
+.content-panel-alphatts label > span.label-text 
+{{
+    font-weight:500 !important; color:#4B5563 !important; 
+    font-size:0.88em !important; margin-bottom:6px !important; display:inline-block;
+}}
+/* Your original specific label styling with icons (if you want to keep them) */
+/* You would need to ensure your Gradio labels have the correct `for` attribute linking to input `elem_id`
+   or use JavaScript to add these pseudo-elements if Gradio doesn't directly support `for` on labels.
+   For simplicity, I'm omitting the ::before icon styles unless you confirm they are essential
+   and your Gradio setup can support them easily. The general label style above will apply.
+*/
+
+/* Temperature description (from your original AlphaTTS CSS) */
+.content-panel-alphatts .temp_description_class_alpha_v3 {{ 
+    font-size: 0.85em; color: #777; margin-top: -0.4rem; margin-bottom: 1rem; 
+}}
+
+/* Audio Player (general styling, can be targeted by ID if set) */
+.content-panel-alphatts .gr-audio audio, /* General audio player */
+.content-panel-alphatts #output_audio_player_alpha_v3 audio /* Your specific ID */
 {{ 
-    font-weight: 700 !important; color: var(--app-text-primary) !important; 
-    font-size: 0.95em !important; margin-bottom: 0.5rem !important; 
+    width: 100%; border-radius: var(--radius-input-original); margin-top:0.8rem; 
 }}
-/* Your original icon styling (if elem_ids match) */
-label[for*="text_input_main_alpha_v3"] > .label-text::before {{ content: '📝'; margin-left: 8px; vertical-align: middle; opacity: 0.7;}}
-label[for*="speech_prompt_alpha_v3"] > .label-text::before {{ content: '🗣️'; margin-left: 8px; vertical-align: middle; opacity: 0.7;}}
-label[for*="speaker_voice_alpha_v3"] > .label-text::before {{ content: '🎤'; margin-left: 8px; vertical-align: middle; opacity: 0.7;}}
-label[for*="temperature_slider_alpha_v3"] > .label-text::before {{ content: '🌡️'; margin-left: 8px; vertical-align: middle; opacity: 0.7;}}
-
-/* Output audio player and temperature description from your original AlphaTTS CSS */
-#output_audio_player_alpha_v3 audio, .output-audio-player-tts audio {{ width: 100%; border-radius: var(--radius-input); margin-top:0.8rem; }}
-.temp_description_class_alpha_v3 {{ font-size: 0.85em; color: #777; margin-top: -0.4rem; margin-bottom: 1rem; }}
-
-/* Footer from your original AlphaTTS CSS */
-.app-footer-final {{
-    text-align:center;font-size:0.9em;color: var(--app-text-secondary);opacity:0.8; 
-    margin-top:3rem;padding:1.5rem 0; border-top:1px solid var(--app-border-color);
+
+/* Examples (using AlphaTranslator_Styled examples button style) */
+.content-panel-alphatts div[label*="نمونه‌های کاربردی"] .gr-button.gr-button-tool, /* Targetting by label */
+.content-panel-alphatts div[label*="نمونه‌های کاربردی"] .gr-sample-button 
+{{
+    background-color:#E0E7FF !important; color:var(--fly-primary) !important;
+    border-radius:6px !important; font-size:0.78em !important; padding:4px 8px !important;
 }}
-/* Hide default Gradio footer */
-footer.svelte-11rogk8, .gradio-footer.svelte-1k000mq {{display:none !important;}} 
-"""
+.content-panel-alphatts .custom-hr {{height:1px;background-color:var(--fly-border-color);margin:1.5rem 0;border:none;}}
+
+/* Footer styling from AlphaTranslator_Styled */
+.app-footer-alphatts {{ /* Changed class name slightly */
+    text-align:center;font-size:0.85em;color:var(--fly-text-secondary);margin-top:2.5rem;
+    padding:1rem 0;background-color:rgba(255,255,255,0.3);backdrop-filter:blur(5px);
+    border-top:1px solid var(--fly-border-color);
+}}
+footer {{display:none !important;}} /* Hides default Gradio footer */
+
 
-# Header HTML from your original AlphaTTS
-alpha_header_html_v3 = """
-<div class='app-header-alpha'>
-  <h1>Alpha TTS</h1>
-  <p>جادوی تبدیل متن به صدا در دستان شما</p>
-</div>
+/* Responsive adjustments from AlphaTranslator_Styled */
+@media (min-width:640px) {{
+    .main-content-area-alphatts {{padding:1.5rem;max-width:700px;}} 
+    .content-panel-alphatts {{padding:1.5rem;}} 
+    .app-header-alphatts h1 {{font-size:2.5em !important;}} 
+    .app-header-alphatts p {{font-size:1.05em !important;}} 
+}}
+@media (min-width:768px) {{
+    .main-content-area-alphatts {{max-width:780px;}} 
+    .content-panel-alphatts {{padding:2rem;}}
+    .content-panel-alphatts .gr-button[elem_id="generate_button_alpha_v3"],
+    .content-panel-alphatts button[variant="primary"]
+    {{
+        width:auto !important; align-self:flex-start;
+    }}
+    .app-header-alphatts h1 {{font-size:2.75em !important;}} 
+    .app-header-alphatts p {{font-size:1.1em !important;}}
+}}
 """
 
-if not PYDUB_AVAILABLE:
-    logging.warning("Pydub (for audio merging) not found. Merging will be disabled.")
-if not os.environ.get("GEMINI_API_KEY"):
-    logging.warning("GEMINI_API_KEY environment variable not set. TTS will likely fail.")
+# Using your original Gradio Blocks structure
+# The theme `gr.themes.Base(font=[gr.themes.GoogleFont("Vazirmatn")])` is from your original.
+# We are applying `app_theme_outer_styled` for the body background and `applied_css_for_alphatts` for specifics.
+with gr.Blocks(theme=app_theme_outer_styled, css=applied_css_for_alphatts, title=f"آلفا TTS ({FIXED_MODEL_NAME.split('-')[1]})") as demo:
+    # Applying the header from AlphaTranslator_Styled structure
+    gr.HTML(f"""
+        <div class='app-header-alphatts'>
+            <h1>🚀 Alpha TTS</h1>
+            <p>جادوی تبدیل متن به صدا در دستان شما (Gemini {FIXED_MODEL_NAME.split('-')[1]})</p>
+        </div>
+    """)
+
+    # Applying the main content panel structure from AlphaTranslator_Styled
+    with gr.Column(elem_classes=["main-content-area-alphatts"]):
+        with gr.Column(elem_classes=["content-panel-alphatts"]): # Your original AlphaTTS used Column, let's keep it simple
+                                                               # AlphaTranslator used Group, but Column is fine.
+
+            # Your original UI layout from AlphaTTS_Original
+            # Note: `elem_id`s are from your original AlphaTTS code.
+            # CSS selectors have been updated to try and match these or use general selectors.
+            
+            # Warning if GEMINI_API_KEY is not set
+            if not os.environ.get("GEMINI_API_KEY"):
+                missing_key_msg = (
+                    "⚠️ هشدار: متغیر محیطی GEMINI_API_KEY تنظیم نشده است. "
+                    "قابلیت تبدیل متن به گفتار احتمالاً کار نخواهد کرد. "
+                    "لطفاً این متغیر را در بخش Secrets این Space تنظیم کنید."
+                )
+                gr.Markdown(f"<div class='api-warning-message'>{missing_key_msg}</div>")
 
+            use_file_input_cb = gr.Checkbox(label="📄 استفاده از فایل متنی (.txt)", value=False, elem_id="use_file_cb_alpha_v3")
+            
+            uploaded_file_input = gr.File(
+                label=" ", 
+                file_types=['.txt'], 
+                visible=False,
+                elem_id="file_uploader_alpha_main_v3"
+            ) 
+            
+            text_to_speak_tb = gr.Textbox(
+                label="متن فارسی برای تبدیل", 
+                placeholder="مثال: سلام، فردا هوا چطور است؟", 
+                lines=5,
+                value="", 
+                visible=True,
+                elem_id="text_input_main_alpha_v3"
+            )
+            
+            # Your original change function for checkbox
+            use_file_input_cb.change(
+                fn=lambda x: (gr.update(visible=x, label=" " if x else "متن فارسی برای تبدیل"), gr.update(visible=not x)),
+                inputs=use_file_input_cb,
+                outputs=[uploaded_file_input, text_to_speak_tb]
+            )
 
-with gr.Blocks(theme=gr.themes.Base(font=[gr.themes.GoogleFont("Vazirmatn")]), css=custom_css_combined, title=f"آلفا TTS ({FIXED_MODEL_NAME.split('-')[1]})") as demo:
-    gr.HTML(alpha_header_html_v3) # YOUR HEADER
+            speech_prompt_tb = gr.Textbox(
+                label="سبک گفتار (اختیاری)",
+                placeholder="مثال: با لحنی شاد و پرانرژی", 
+                value="با لحنی دوستانه و رسا صحبت کن.",
+                lines=2, elem_id="speech_prompt_alpha_v3"
+            )
+            
+            speaker_voice_dd = gr.Dropdown(
+                SPEAKER_VOICES, label="انتخاب گوینده و لهجه", value="Charon", elem_id="speaker_voice_alpha_v3"
+            )
+            
+            temperature_slider = gr.Slider(
+                minimum=0.1, maximum=1.5, step=0.05, value=0.9, label="میزان خلاقیت صدا",
+                elem_id="temperature_slider_alpha_v3"
+            )
+            gr.Markdown("<p class='temp_description_class_alpha_v3'>مقادیر بالاتر = تنوع بیشتر، مقادیر پایین‌تر = یکنواختی بیشتر.</p>")
 
-    # Using 'main-content-panel-alpha' class for the main content area as in your original CSS
-    with gr.Column(elem_classes=["main-content-panel-alpha"]):
-        # YOUR ORIGINAL UI ELEMENTS (with original elem_ids for CSS)
-        use_file_input_cb = gr.Checkbox(label="📄 استفاده از فایل متنی (.txt)", value=False, elem_id="use_file_cb_alpha_v3")
-        
-        uploaded_file_input = gr.File(
-            label=" ", 
-            file_types=['.txt'], 
-            visible=False,
-            elem_id="file_uploader_alpha_main_v3" # YOUR ELEM_ID
-        ) 
-        
-        text_to_speak_tb = gr.Textbox(
-            label="متن فارسی برای تبدیل", 
-            placeholder="مثال: سلام، فردا هوا چطور است؟", 
-            lines=5,
-            value="", 
-            visible=True,
-            elem_id="text_input_main_alpha_v3" # YOUR ELEM_ID
-        )
+            generate_button = gr.Button("🚀 تولید و پخش صدا", elem_id="generate_button_alpha_v3") # Removed variant="primary" to let CSS handle it via elem_id
+            
+            output_audio = gr.Audio(label=" ", type="filepath", elem_id="output_audio_player_alpha_v3")
         
-        # Event handler from your original code
-        use_file_input_cb.change(
-            fn=lambda x: (gr.update(visible=x, label=" " if x else "متن فارسی برای تبدیل"), gr.update(visible=not x)),
-            inputs=use_file_input_cb,
-            outputs=[uploaded_file_input, text_to_speak_tb]
-        )
+            # Your original Examples section
+            # Applying a custom HR from AlphaTranslator_Styled
+            gr.HTML("<hr class='custom-hr'>")
+            gr.Markdown(
+                "<h3 style='text-align:center; font-weight:500; color:var(--fly-text-secondary); margin-top:1.5rem; margin-bottom:1rem;'>نمونه‌های کاربردی</h3>",
+                # elem_id="examples_section_title_v3" # elem_id from your original
+            )
+            gr.Examples(
+                examples=[
+                    [False, None, "سلام بر شما، امیدوارم روز خوبی داشته باشید.", "با لحنی گرم و صمیمی.", "Zephyr", 0.85],
+                    [False, None, "این یک آزمایش برای بررسی کیفیت صدای تولید شده توسط هوش مصنوعی آلفا است.", "با صدایی طبیعی و روان.", "Charon", 0.9],
+                ],
+                inputs=[use_file_input_cb, uploaded_file_input, text_to_speak_tb, speech_prompt_tb, speaker_voice_dd, temperature_slider],
+                outputs=[output_audio], # Original AlphaTTS only output to audio
+                fn=gradio_tts_interface,
+                cache_examples=os.getenv("GRADIO_CACHE_EXAMPLES", "False").lower() == "true" # From AlphaTranslator
+            )
 
-        speech_prompt_tb = gr.Textbox(
-            label="سبک گفتار (اختیاری)",
-            placeholder="مثال: با لحنی شاد و پرانرژی", 
-            value="با لحنی دوستانه و رسا صحبت کن.",
-            lines=2, elem_id="speech_prompt_alpha_v3" # YOUR ELEM_ID
-        )
-        
-        speaker_voice_dd = gr.Dropdown(
-            SPEAKER_VOICES, label="انتخاب گوینده و لهجه", value="Charon", elem_id="speaker_voice_alpha_v3" # YOUR ELEM_ID
-        )
-        
-        temperature_slider = gr.Slider(
-            minimum=0.1, maximum=1.5, step=0.05, value=0.9, label="میزان خلاقیت صدا",
-            elem_id="temperature_slider_alpha_v3" # YOUR ELEM_ID
-        )
-        gr.Markdown("<p class='temp_description_class_alpha_v3'>مقادیر بالاتر = تنوع بیشتر، مقادیر پایین‌تر = یکنواختی بیشتر.</p>")
+    # Footer from AlphaTranslator_Styled
+    gr.Markdown(f"<p class='app-footer-alphatts'>Alpha TTS © 2024 - Model: {FIXED_MODEL_NAME}</p>")
 
-        generate_button = gr.Button(
-            "🚀 تولید و پخش صدا", 
-            elem_classes=["generate-button-tts"], # Applied class for styling from combined CSS
-            elem_id="generate_button_alpha_v3" # YOUR ELEM_ID
-        )
-        
-        output_audio = gr.Audio(
-            label=" ", type="filepath", 
-            elem_id="output_audio_player_alpha_v3", # YOUR ELEM_ID
-            elem_classes=["output-audio-player-tts"] # Added class for CSS consistency
-        )
-    
-        generate_button.click(
-            fn=gradio_tts_interface, # YOUR INTERFACE FUNCTION
-            # Your original inputs, progress removed for now as it was not fully integrated
-            inputs=[use_file_input_cb, uploaded_file_input, text_to_speak_tb, speech_prompt_tb, speaker_voice_dd, temperature_slider],
-            outputs=[output_audio] # Your interface function returns only audio path
-        )
 
-        # Your original Examples section
-        gr.Markdown("<h3 class='section-title-main-alpha' style='margin-top:2.5rem; text-align:center; border-bottom:none;'>نمونه‌های کاربردی</h3>", elem_id="examples_section_title_v3")
-        gr.Examples(
-            examples=[
-                [False, None, "سلام بر شما، امیدوارم روز خوبی داشته باشید.", "با لحنی گرم و صمیمی.", "Zephyr", 0.85],
-                [False, None, "این یک آزمایش برای بررسی کیفیت صدای تولید شده توسط هوش مصنوعی آلفا است.", "با صدایی طبیعی و روان.", "Charon", 0.9],
-            ],
-            inputs=[use_file_input_cb, uploaded_file_input, text_to_speak_tb, speech_prompt_tb, speaker_voice_dd, temperature_slider],
-            outputs=[output_audio], # Outputting only to audio as per your original
+    # --- Event Handlers (from YOUR AlphaTTS_Original) ---
+    if generate_button is not None:
+         generate_button.click(
             fn=gradio_tts_interface,
-            cache_examples=os.getenv("GRADIO_CACHE_EXAMPLES", "False").lower() == "true" # Added this for optional caching
+            inputs=[use_file_input_cb, uploaded_file_input, text_to_speak_tb, speech_prompt_tb, speaker_voice_dd, temperature_slider],
+            outputs=[output_audio] # Original AlphaTTS only output to audio
         )
-        
-    # Footer from your original AlphaTTS (app-footer-final)
-    gr.Markdown(f"<p class='app-footer-final'>Alpha Language Learning © 2024 - Model: {FIXED_MODEL_NAME}</p>")
-
+    else:
+        logging.error("دکمه تولید صدا (generate_button_alpha_v3) به درستی مقداردهی اولیه نشده است.")
+# --- END: Gradio UI ---
 
 if __name__ == "__main__":
-    # Auto-restart logic can be added here if desired, but kept out for strict "no logic change"
-    # if os.getenv("AUTO_RESTART_ENABLED", "true").lower() == "true":
-    #     import threading, sys
-    #     def auto_restart_service(): # Minimal auto-restart
-    #         time.sleep(24*60*60)
-    #         os._exit(1)
-    #     threading.Thread(target=auto_restart_service, daemon=True).start()
+    # Removed auto-restart thread to keep it closer to your original AlphaTTS.
+    # If you need it, you can re-add the auto_restart_service function and thread start.
+
+    # Check if PYDUB is available at launch
+    if not PYDUB_AVAILABLE:
+        logging.warning("Pydub (for audio merging) not found. Please install with `pip install pydub`. Merging will be disabled if multiple audio chunks are generated.")
 
     demo.launch(
-        server_name="0.0.0.0", # Good for Docker/HF Spaces
-        server_port=int(os.getenv("PORT", 7860)), # Standard for HF Spaces
-        debug=os.environ.get("GRADIO_DEBUG", "False").lower() == "true", # Optional debug
+        server_name="0.0.0.0",
+        server_port=int(os.getenv("PORT", 7860)),
+        debug=os.environ.get("GRADIO_DEBUG", "False").lower() == "true",
         show_error=True
     )