Update app.py

app.py

@@ -16,48 +16,54 @@ from fastapi.responses import JSONResponse
 
 warnings.filterwarnings("ignore")
 
-# ذخیره‌سازی نتایج
+# ذخیره‌سازی نتایج (متن همیشه ذخیره می‌شود)
 results_cache = {}
 cache_lock = threading.Lock()
 
 # FastAPI instance برای API endpoints
 app = FastAPI()
 
-# تابع برای پاکسازی cache قدیمی
+# تابع برای پاکسازی cache قدیمی (بهبود: فقط event_idهای کامل‌شده و قدیمی را پاک کن)
 def cleanup_old_cache():
     while True:
         time.sleep(3600)  # هر ساعت
         with cache_lock:
             current_time = datetime.now()
             keys_to_remove = []
-            for key, value in results_cache.items():
-                if current_time - value['timestamp'] > timedelta(hours=24):
-                    keys_to_remove.append(key)
-            for key in keys_to_remove:
-                del results_cache[key]
+            for event_id, value in results_cache.items():
+                if value['completed'] and (current_time - value['timestamp'] > timedelta(hours=24)):
+                    keys_to_remove.append(event_id)
+                    # پاک کردن فایل MP3 اگر وجود دارد
+                    if value['mp3_path'] and os.path.exists(value['mp3_path']):
+                        try:
+                            os.remove(value['mp3_path'])
+                        except:
+                            pass
+            for event_id in keys_to_remove:
+                del results_cache[event_id]
             if keys_to_remove:
-                print(f"[CACHE] {len(keys_to_remove)} نتیجه قدیمی پاک شد.")
+                print(f"[CACHE] {len(keys_to_remove)} event_id قدیمی پاک شد.")
 
 cleanup_thread = threading.Thread(target=cleanup_old_cache, daemon=True)
 cleanup_thread.start()
 
-def get_cache_key(video_url, language):
-    """ایجاد کلید یکتا برای cache"""
+def get_event_id(video_url, language):
+    """ایجاد event_id یکتا (قبلاً cache_key)"""
     return hashlib.md5(f"{video_url}_{language}".encode()).hexdigest()
 
-def save_result_to_cache(cache_key, video_url, language, mp3_path, text, status_msg, processing=False):
-    """ذخیره نتیجه در cache"""
-    # کپی فایل MP3 به مکان دائمی
+def save_result_to_cache(event_id, video_url, language, mp3_path, text, status_msg, processing=False):
+    """ذخیره نتیجه در cache (متن همیشه ذخیره می‌شود)"""
+    # کپی فایل MP3 به مکان دائمی اگر موجود باشد
     cache_mp3_path = None
     if mp3_path and os.path.exists(mp3_path):
-        cache_mp3_path = f"cache_{cache_key}.mp3"
+        cache_mp3_path = f"cache_{event_id}.mp3"
         shutil.copy2(mp3_path, cache_mp3_path)
     
     with cache_lock:
-        results_cache[cache_key] = {
+        results_cache[event_id] = {
             'timestamp': datetime.now(),
             'mp3_path': cache_mp3_path,
-            'text': text,
+            'text': text,  # متن همیشه ذخیره می‌شود (حتی اگر None باشد)
             'status_msg': status_msg,
             'video_url': video_url,
             'language': language,
@@ -65,53 +71,45 @@ def save_result_to_cache(cache_key, video_url, language, mp3_path, text, status_
             'completed': not processing and (text is not None or status_msg.startswith("خطا"))
         }
     
-    print(f"[CACHE] نتیجه ذخیره شد: {cache_key} (processing: {processing})")
+    print(f"[CACHE] نتیجه ذخیره شد: {event_id} (processing: {processing}, text_length: {len(text) if text else 0})")
 
-def get_result_from_cache(cache_key):
-    """دریافت نتیجه از cache"""
+def get_result_from_cache(event_id):
+    """دریافت نتیجه از cache با event_id"""
     with cache_lock:
-        if cache_key in results_cache:
-            result = results_cache[cache_key]
-            print(f"[CACHE] نتیجه یافت شد: {cache_key}")
+        if event_id in results_cache:
+            result = results_cache[event_id]
+            print(f"[CACHE] نتیجه یافت شد: {event_id}")
             return result
     return None
 
 def convert_to_mp3_and_transcribe_wrapper(video_url, language):
-    """Wrapper function که cache_key را در ابتدا برمی‌گرداند"""
+    """Wrapper function که event_id را در ابتدا برمی‌گرداند"""
     if not video_url:
         return None, None, "لینک ویدیو را وارد کنید.", None
     
-    # ایجاد cache key
-    cache_key = get_cache_key(video_url, language)
+    # ایجاد event_id
+    event_id = get_event_id(video_url, language)
     
     # چک کردن cache
-    cached_result = get_result_from_cache(cache_key)
+    cached_result = get_result_from_cache(event_id)
     if cached_result and cached_result['completed']:
         return (cached_result['mp3_path'], 
                 cached_result['text'], 
                 f"[از حافظ��] {cached_result['status_msg']}", 
-                cache_key)
+                event_id)
     
     # اگر در حال پردازش است
     if cached_result and cached_result['processing']:
-        return None, None, "در حال پردازش... لطفاً صبر کنید.", cache_key
+        return None, None, "در حال پردازش... لطفاً صبر کنید.", event_id
     
     # ثبت شروع پردازش
-    save_result_to_cache(cache_key, video_url, language, None, None, "در حال پردازش...", processing=True)
+    save_result_to_cache(event_id, video_url, language, None, None, "در حال پردازش...", processing=True)
     
-    # پردازش در background
-    def process_async():
-        try:
-            mp3_path, text, status_msg = convert_to_mp3_and_transcribe(video_url, language)
-            save_result_to_cache(cache_key, video_url, language, mp3_path, text, status_msg, processing=False)
-        except Exception as e:
-            save_result_to_cache(cache_key, video_url, language, None, None, f"خطا: {str(e)}", processing=False)
-    
-    # شروع پردازش (برای سادگی، همینجا انجام می‌دهیم - در production باید async باشد)
+    # پردازش در background (برای سادگی، همینجا انجام می‌دهیم - در production async کنید)
     mp3_path, text, status_msg = convert_to_mp3_and_transcribe(video_url, language)
-    save_result_to_cache(cache_key, video_url, language, mp3_path, text, status_msg, processing=False)
+    save_result_to_cache(event_id, video_url, language, mp3_path, text, status_msg, processing=False)
     
-    return mp3_path, text, status_msg, cache_key
+    return mp3_path, text, status_msg, event_id
 
 def convert_to_mp3_and_transcribe(video_url, language):
     """تابع اصلی پردازش (بدون cache)"""
@@ -276,29 +274,44 @@ def transcribe_audio(mp3_path, progress, language, chunk_length_ms=60000, overla
     
     return final_text, ""
 
-# API endpoint
-@app.get("/api/check_result/{cache_key}")
-async def check_result_api(cache_key: str):
-    """API endpoint برای بررسی وضعیت نتیجه"""
-    result = get_result_from_cache(cache_key)
+# API endpoints (بهبود: event_id به جای cache_key، و endpoint جدید برای لیست)
+@app.get("/api/check_result/{event_id}")
+async def check_result_api(event_id: str):
+    """API endpoint برای بررسی وضعیت نتیجه با event_id (متن اگر کامل شده برگردانده می‌شود)"""
+    result = get_result_from_cache(event_id)
     
     if not result:
-        return JSONResponse({"status": 0, "message": "Not found"})
+        return JSONResponse({"status": 0, "message": "Event ID not found"})
     
     if result['processing']:
-        return JSONResponse({"status": 0, "message": "Processing"})
+        return JSONResponse({"status": 0, "message": "Processing..."})
     
     if result['completed']:
         return JSONResponse({
             "status": 1,
-            "text": result['text'],
-            "mp3_available": result['mp3_path'] is not None,
-            "status_message": result['status_msg']
+            "event_id": event_id,
+            "text": result['text'],  # متن استخراج‌شده همیشه برگردانده می‌شود
+            "mp3_available": result['mp3_path'] is not None and os.path.exists(result['mp3_path']),
+            "status_message": result['status_msg'],
+            "timestamp": result['timestamp'].isoformat()
         })
     
     return JSONResponse({"status": 0, "message": "Unknown status"})
 
-# رابط Gradio
+@app.get("/api/list_events")
+async def list_events_api(limit: int = 10):
+    """API endpoint جدید: لیست event_idهای اخیر (برای مدیریت)"""
+    with cache_lock:
+        recent_events = sorted(
+            [(eid, data['timestamp']) for eid, data in results_cache.items() if data['completed']],
+            key=lambda x: x[1],
+            reverse=True
+        )[:limit]
+        return JSONResponse({
+            "events": [{"event_id": eid, "timestamp": ts.isoformat()} for eid, ts in recent_events]
+        })
+
+# رابط Gradio (بهبود: event_id برجسته‌تر)
 iface = gr.Interface(
     fn=convert_to_mp3_and_transcribe_wrapper,
     inputs=[
@@ -316,9 +329,9 @@ iface = gr.Interface(
         gr.File(label="دانلود MP3"),
         gr.Textbox(label="متن استخراج‌شده", lines=10),
         gr.Textbox(label="وضعیت"),
-        gr.Textbox(label="Cache Key", visible=True)
+        gr.Textbox(label="Event ID (برای استعلام API)", visible=True, info="این ID را برای چک کردن نتیجه از API استفاده کنید: /api/check_result/{event_id}")
     ],
-    title="تبدیل ویدیو به MP3 و متن",
+    title="تبدیل ویدیو به MP3 و متن (با ذخیره event_id)",
     examples=[
         ["https://www.youtube.com/watch?v=5qap5aO4i9A", "fa-IR"],
         ["https://www.youtube.com/watch?v=dQw4w9WgXcQ", "en-US"]