Update app.py

app.py

@@ -10,39 +10,20 @@ import time
 import numpy as np
 import re
 import json
-import uuid  # 👈 Session ID ထုတ်ရန်အတွက် Import လုပ်ခြင်း
+import threading
 from google import genai
-from datetime import datetime, date  
 from PIL import Image, ImageDraw, ImageFont
 
 # =====================================================================
-# ⚙️ SECURITY SETTINGS & LOGIN CONFIG
+# ⚙️ SETTINGS & CONFIG (Login & Limits Removed)
 # =====================================================================
 MAX_SUBTITLE_DURATION_SECONDS = 7.0
 
-# 👥 LOGIN ACCOUNTS CONFIG (အကောင့်တစ်ခုတည်းကို လူအများ မျှသုံးနိုင်ပါသည်)
-ACCOUNTS = {
-    "user1": "123123",    # Free User (Session/Browser အလိုက် တစ်ရက် ၂ ပုဒ်)
-    "admin": "admin789"   # Admin User (Unlimited)
-}
-
-EXPIRY_DATE_STR = "2026-8-31"
-
-# 🛑 USER RATE LIMIT TRACKER (Session/Browser ID အလိုက် ရေတွက်မည်)
-USER_LIMIT_TRACKER = {}
-
 SYSTEM_INSTRUCTION = "You are a professional movie recap writer. Translate movie subtitle lines into natural, engaging, and thrilling Burmese movie recap style. Keep it concise."
 MODEL_NAME = "gemini-2.5-flash"
 
 def check_app_expiry():
-    try:
-        expiry_date = datetime.strptime(EXPIRY_DATE_STR, "%Y-%m-%d").date()
-        current_date = datetime.now().date()
-        if current_date > expiry_date:
-            return False, f"❌ ဤ App သည် သက်တမ်းကုန်ဆုံးသွားပါပြီ ({EXPIRY_DATE_STR})။"
-        return True, "Active"
-    except Exception as e:
-        return False, f"လုံခြုံရေး စစ်ဆေးမှု မှားယွင်းနေပါသည်- {str(e)}"
+    return True, "Active"
 
 print("Loading Multilingual Faster-Whisper Base Model...")
 model = WhisperModel("base", device="cpu", compute_type="int8", cpu_threads=4)
@@ -81,12 +62,11 @@ def update_preview_image(video_path, blur_y_percent, blur_strength):
         return None
 
 # =====================================================================
-# ⚡ DYNAMIC USER-PROVIDED API KEY TRANSLATION ENGINE
+# ⚡ TRANSLATION ENGINE
 # =====================================================================
 def google_backup_translate(text, source_lang="en"):
     from urllib.parse import quote
     import urllib.request
-    import json
     try:
         url = f"https://translate.googleapis.com/translate_a/single?client=gtx&sl={source_lang}&tl=my&dt=t&q={quote(text)}"
         req = urllib.request.Request(url, headers={'User-Agent': 'Mozilla/5.0'})
@@ -100,7 +80,7 @@ def translate_segments_batch(segments, user_api_key, source_lang="en"):
     if not segments: return segments
     
     if not user_api_key or not user_api_key.strip():
-        print(f"⚠️ User က Gemini API Key မထည့်ထားပါ။ Google Translate Backup စနစ်ဖြင့် ဘာသာပြန်နေပါသည်။ (Source Lang: {source_lang})")
+        print(f"⚠️ User က Gemini API Key မထည့်ထားပါ။ Google Translate Backup စနစ်ဖြင့် ဘာသာပြန်နေပါသည်။")
         for seg in segments:
             seg['mm_text'] = google_backup_translate(seg['text'], source_lang=source_lang)
         return segments
@@ -138,7 +118,7 @@ def translate_segments_batch(segments, user_api_key, source_lang="en"):
             
         translated_map = json.loads(response_text)
     except Exception as e:
-        print(f"⚠️ User Gemini API Error သို့မဟုတ် Format Error: {e} -> Google Translate သို့ ပြောင်းလဲနေသည်။")
+        print(f"⚠️ Gemini API Error: {e} -> Google Translate သို့ ပြောင်းလဲနေသည်။")
 
     for idx, seg in enumerate(segments):
         key_str = str(idx)
@@ -150,7 +130,7 @@ def translate_segments_batch(segments, user_api_key, source_lang="en"):
     return segments
 
 # =====================================================================
-# 🎬 TEXT WRAPPING & RENDER SYSTEMS
+# 🎬 TEXT WRAPPING & VOICE GENERATION SYSTEMS
 # =====================================================================
 def segment_myanmar_syllables(text):
     return re.findall(r'[a-zA-Z0-9\s\-\.,!\?]+|[\u1000-\u102a\u103f\u1040-\u1049]+[\u102b-\u103e\u1060-\u109f]*|[^\s]', text)
@@ -180,7 +160,34 @@ def draw_line_perfect_rendering(draw, position, text, font_primary, fill_color,
     clean_text = text.replace(" ြ", "ြ").replace("ြ ", "ြ")
     draw.text((x, y), clean_text, font=font_primary, fill=fill_color, stroke_width=stroke_w, stroke_fill=stroke_c)
 
-def generate_voice_sync(text, voice_id, filename, desired_speed, target_duration_sec=None):
+def generate_voice_sync(text, voice_id, filename, desired_speed, target_duration_sec=None, user_api_key=None, voice_engine="Edge-TTS"):
+    # Google AI Studio Audio (Gemini Native Audio) အသုံးပြုလိုပါက
+    if voice_engine == "Google AI Studio Audio" and user_api_key and user_api_key.strip():
+        try:
+            client = genai.Client(api_key=user_api_key.strip())
+            prompt = f"Read the following Burmese movie recap text naturally and clearly with an engaging narrative tone. Generate audio output:\n{text}"
+            response = client.models.generate_content(
+                model=MODEL_NAME,
+                contents=prompt,
+                config={
+                    "response_mime_type": "audio/mp3",
+                }
+            )
+            audio_saved = False
+            for candidate in response.candidates:
+                for part in candidate.content.parts:
+                    if hasattr(part, 'inline_data') and part.inline_data:
+                        with open(filename, "wb") as f:
+                            f.write(part.inline_data.data)
+                        audio_saved = True
+                        break
+                if audio_saved: break
+            if audio_saved and os.path.exists(filename) and os.path.getsize(filename) > 0:
+                return
+        except Exception as e:
+            print(f"⚠️ Google AI Studio Audio Error: {e} -> Edge-TTS သို့ အလိုအလျောက် ပြောင်းလဲနေသည်။")
+
+    # Default: Edge-TTS
     rate_percentage = int((desired_speed - 1.0) * 100)
     rate_str = f"{'+' if rate_percentage >= 0 else ''}{rate_percentage}%"
     
@@ -195,64 +202,27 @@ def generate_voice_sync(text, voice_id, filename, desired_speed, target_duration
         communicate = edge_tts.Communicate(text, voice_id, rate=rate_str)
         await communicate.save(filename)
     
-    try: loop = asyncio.get_event_loop()
-    except RuntimeError:
-        loop = asyncio.new_event_loop()
-        asyncio.set_event_loop(loop)
-        
-    if loop.is_running():
-        import threading
-        t = threading.Thread(target=lambda: asyncio.run(_async_gen()))
+    try:
+        def run_in_thread():
+            asyncio.run(_async_gen())
+        t = threading.Thread(target=run_in_thread)
         t.start()
         t.join()
-    else:
-        loop.run_until_complete(_async_gen())
+    except Exception as e:
+        print(f"Audio Generation Error: {e}")
 
 # =====================================================================
-# 🎬 HIGH-SPEED PRODUCTION AUTOMATION ENGINE
+# 🎬 PRODUCTION AUTOMATION ENGINE (Limits Removed)
 # =====================================================================
 def process_magic_recap_video(
     video_path, user_api_key, ratio_select, background_fill, enable_zoom, zoom_level,
-    logo_file, mirror_flip, filter_color, voice_gender, tone_style,
+    logo_file, mirror_flip, filter_color, voice_gender, voice_engine_select, tone_style,
     text_color, stroke_color, blur_y_percent, blur_strength, sub_pos_percent, desired_speed,
-    session_id, # 👈 Browser Session State
-    request: gr.Request, 
     progress=gr.Progress(track_tqdm=True)
 ):
     is_valid, msg = check_app_expiry()
     if not is_valid: raise gr.Error(msg)
     if not video_path: return None
-        
-    # 🛑 LOGIN ACCOUNT & SESSION ID ယူခြင်း
-    login_username = request.username if (request and hasattr(request, 'username')) else "user1"
-    
-    if login_username == "admin":
-        user_identifier = "admin"
-    else:
-        # Browser Session ID ဖြင့် လူတစ်ဦးစီကို သီးခြား ခွဲခြားမည်
-        user_identifier = f"session_{session_id}"
-
-    current_timestamp = time.time()
-    print(f"👤 အသုံးပြုသူ Account: {login_username} | Identifier: {user_identifier}")
-
-    # Admin မဟုတ်ပါက Session/Browser အလိုက် ၁ ရက် (၂) ပုဒ် Limit စစ်ဆေးမည်
-    if user_identifier != "admin":
-        if user_identifier in USER_LIMIT_TRACKER:
-            user_data = USER_LIMIT_TRACKER[user_identifier]
-            first_time = user_data["first_time"]
-            count = user_data["count"]
-            elapsed_time = current_timestamp - first_time
-
-            if elapsed_time < 86400:  # ၂၄ နာရီ အတွင်း
-                if count >= 2:  # ၂ ပုဒ် ပြည့်ပါက တားဆီးမည်
-                    remaining_seconds = 86400 - elapsed_time
-                    rem_hours = int(remaining_seconds // 3600)
-                    rem_mins = int((remaining_seconds % 3600) // 60)
-                    raise gr.Error(f"❌ ခွင့်ပြုချက်ကျော်လွန်နေပါသည်။ သင့် Browser မှ တစ်ရက်လျှင် ဗီဒီယို (၂) ပုဒ်သာ ထုတ်ယူခွင့်ရှိသည်။ ပြန်လည်စမ်းသပ်ရန် {rem_hours} နာရီ {rem_mins} မိနစ် လိုအပ်ပါသေးသည်။")
-            else:
-                USER_LIMIT_TRACKER[user_identifier] = {"count": 0, "first_time": current_timestamp}
-        else:
-            USER_LIMIT_TRACKER[user_identifier] = {"count": 0, "first_time": current_timestamp}
 
     temp_dir = "temp_space_workspace"
     if os.path.exists(temp_dir): shutil.rmtree(temp_dir)
@@ -293,9 +263,9 @@ def process_magic_recap_video(
 
         detected_lang = info.language
         if user_api_key and user_api_key.strip():
-            progress(0.30, desc=f"⚡ ထည့်သွင်းထားသော Gemini API ({detected_lang}) စနစ်ဖြင့် ဘာသာပြန်နေပါသည်...")
+            progress(0.30, desc=f"⚡ ထည့်သွင်းထားသော Gemini API စနစ်ဖြင့် ဘာသာပြန်နေပါသည်...")
         else:
-            progress(0.30, desc=f"🌐 Google Translate Backup ({detected_lang}) စနစ်ဖြင့် ဘာသာပြန်နေပါသည်...")
+            progress(0.30, desc=f"🌐 Google Translate Backup စနစ်ဖြင့် ဘာသာပြန်နေပါသည်...")
             
         segments = translate_segments_batch(segments, user_api_key, source_lang=detected_lang)
 
@@ -328,7 +298,7 @@ def process_magic_recap_video(
             raw_seg_filename = os.path.join(temp_dir, f"raw_{idx}.mp3")
             fixed_seg_filename = os.path.join(temp_dir, f"fixed_{idx}.mp3")
             
-            generate_voice_sync(mm_text, voice_id, raw_seg_filename, 1.15, orig_dur)
+            generate_voice_sync(mm_text, voice_id, raw_seg_filename, 1.15, orig_dur, user_api_key=user_api_key, voice_engine=voice_engine_select)
             
             if os.path.exists(raw_seg_filename) and os.path.getsize(raw_seg_filename) > 0:
                 subprocess.run([
@@ -356,7 +326,10 @@ def process_magic_recap_video(
         
         font_size = int(target_h * 0.038)
         font_path = "Myanmar font.ttf"
-        font_primary = ImageFont.truetype(font_path, font_size) if os.path.exists(font_path) else ImageFont.load_default()
+        if os.path.exists(font_path):
+            font_primary = ImageFont.truetype(font_path, font_size)
+        else:
+            font_primary = ImageFont.load_default()
 
         t_color, s_color = hex_to_rgb(text_color), hex_to_rgb(stroke_color)
         b_h = int(target_h * 0.12)  
@@ -462,12 +435,6 @@ def process_magic_recap_video(
             subprocess.run(['ffmpeg', '-y', '-i', final_burn_temp, '-c:v', 'libx264', '-preset', 'ultrafast', '-threads', '0', '-pix_fmt', 'yuv420p', output_video_path], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
         
         if os.path.exists(temp_dir): shutil.rmtree(temp_dir)
-        
-        # 📈 ဗီဒီယို အောင်မြင်စွာ ထုတ်ယူပြီးမှသာ ထို Session ၏ Count ကို ၁ တိုးမည်
-        if user_identifier != "admin":
-            USER_LIMIT_TRACKER[user_identifier]["count"] += 1
-            print(f"✅ [{user_identifier}] ဗီဒီယိုထုတ်ယူမှု အောင်မြင်ပါသည်။ ယနေ့ ထုတ်ယူပြီးစီးမှု: {USER_LIMIT_TRACKER[user_identifier]['count']} ကြိမ်")
-
         return output_video_path
         
     except Exception as e:
@@ -478,18 +445,9 @@ def process_magic_recap_video(
 # 🎨 GRADIO INTERFACE
 # =====================================================================
 def create_main_app_interface():
-    is_valid, msg = check_app_expiry()
-    
     with gr.Blocks() as main_app:
-        # 🔑 အသုံးပြုသူတိုင်းအတွက် သီးသန့် Session ID ထုတ်ပေးခြင်း
-        session_id_state = gr.State(lambda: str(uuid.uuid4()))
-        
-        gr.Markdown("<h1 style='text-align: center; color: #5B50F3;'>✨ Video Auto Recap (Multi-Account) ✨</h1>")
-        if not is_valid:
-            gr.Markdown(f"### {msg}")
-            return main_app
-            
-        gr.Markdown("<div style='background-color: #333; padding: 10px; border-radius: 8px; text-align: center; color: #fff;'>📢 <b>Free App Rules:</b> တစ်ရက်လျှင် Video (၂) ပုဒ် အခမဲ့ ထုတ်ယူခွင့် ရှိပါသည်</div>")
+        gr.Markdown("<h1 style='text-align: center; color: #5B50F3;'>✨ Video Auto Recap (Unlimited & AI Audio) ✨</h1>")
+        gr.Markdown("<div style='background-color: #222; padding: 10px; border-radius: 8px; text-align: center; color: #fff;'>📢 <b>Notice:</b> Login စနစ်နှင့် ကန့်သတ်ချက်များ (Limits) အားလုံး ဖြုတ်ထားပြီးဖြစ်ပါသည်။ Google AI Studio Audio (သို့) Edge-TTS ကို ရွေးချယ်အသုံးပြုနိုင်ပါသည်။</div>")
         
         with gr.Row():
             with gr.Column(scale=2):
@@ -497,13 +455,14 @@ def create_main_app_interface():
                 preview_image = gr.Image(label="🖼️ Interactive Blur Preview", interactive=False)
             
             with gr.Column(scale=1):
-                user_api_key = gr.Textbox(label="🔑 သင်၏ Gemini API Key ထည့်ရန် (မထည့်ပါက Google Translate ဖြင့် လုပ်ဆောင်မည်)", placeholder="AIZAcy... (Optional)", type="password")
+                user_api_key = gr.Textbox(label="🔑 သင်၏ Gemini API Key ထည့်ရန် (ဘာသာပြန်နှင့် Google AI Studio အသံအတွက် လိုအပ်သည်)", placeholder="AIZAcy...", type="password")
                 ratio_select = gr.Dropdown(choices=["9:16 (Tiktok/Reels)", "16:9 (Landscape)"], value="9:16 (Tiktok/Reels)", label="ဗီဒီယိုအမျိုးအစား")
                 background_fill = gr.Radio(choices=["Blur Background (အနောက်ခံ ဝါးမည်)", "Black Background (အမည်းရောင်ထားမည်)"], value="Blur Background (အနောက်ခံ ဝါးမည်)", label="နောက်ခံ ဖြည့်စွက်မှု")
 
         with gr.Column(variant="panel"):
             gr.Markdown("### 🗣️ အသံနှင့် စာတန်းထိုး ဆက်တင်များ")
-            voice_select = gr.Radio(choices=["🧕 မိန်းကလေး (Female)", "👨 ယောကျာ်လေး (Male)"], value="🧕 မိန်းကလေး (Female)", label="AI အသံ အမျိုးအစား")
+            voice_engine_select = gr.Radio(choices=["Edge-TTS", "Google AI Studio Audio"], value="Google AI Studio Audio", label="🎙️ အသံထုတ်လုပ်သည့်စနစ် (Voice Engine)")
+            voice_select = gr.Radio(choices=["🧕 မိန်းကလေး (Female)", "👨 ယောကျာ်လေး (Male)"], value="🧕 မိန်းကလေး (Female)", label="AI အသံ ပုံစံ (Edge-TTS အတွက်သာ)")
             tone_style = gr.Dropdown(choices=["Thriller", "Comedy", "Dramatic", "Action/Epic"], value="Thriller", label="🎬 Narrative Tone")
             text_color_input = gr.ColorPicker(label="စာလုံးအရောင်", value="#FFFF00")
             stroke_color_input = gr.ColorPicker(label="အနားသတ်အရောင်", value="#000000")
@@ -532,21 +491,15 @@ def create_main_app_interface():
             fn=process_magic_recap_video,
             inputs=[
                 video_input, user_api_key, ratio_select, background_fill, enable_zoom, zoom_level, 
-                logo_file, mirror_flip, filter_color, voice_select, tone_style, text_color_input, 
-                stroke_color_input, blur_y_percent, blur_strength, sub_pos_percent, desired_speed,
-                session_id_state # 👈 session_id ကို Function ထဲ ပို့ပေးခြင်း
+                logo_file, mirror_flip, filter_color, voice_select, voice_engine_select, tone_style, text_color_input, 
+                stroke_color_input, blur_y_percent, blur_strength, sub_pos_percent, desired_speed
             ],
             outputs=[output_video]
         )
         
     return main_app
 
-def verify_member(username, password):
-    if username in ACCOUNTS and ACCOUNTS[username] == password:
-        return True
-    return False
-
 demo_app = create_main_app_interface()
 
 if __name__ == "__main__":
-    demo_app.launch(auth=verify_member, theme=gr.themes.Default())
+    demo_app.launch(theme=gr.themes.Default())