Upload app.py

app.py

@@ -18,6 +18,8 @@ import json
 from urllib.parse import urlparse
 import tempfile
 import shutil
+import time
+from datetime import datetime, timedelta
 
 app = FastAPI(
     title="Edge TTS API",
@@ -38,6 +40,11 @@ app.add_middleware(
 OUTPUT_DIR = tempfile.mkdtemp(prefix="edge_tts_")
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
+# 檔案清理設定
+FILE_CLEANUP_INTERVAL = 300  # 5分鐘清理一次
+FILE_MAX_AGE = 1800  # 檔案最大存活時間：30分鐘
+MAX_FILES = 100  # 最大檔案數量
+
 # 掛載靜態文件（如果存在）
 if os.path.exists("static"):
     app.mount("/static", StaticFiles(directory="static"), name="static")
@@ -61,6 +68,71 @@ def _is_origin_allowed(request: Request) -> bool:
     # 在 Hugging Face Spaces 環境中，允許所有來源
     return True
 
+# 檔案清理函數
+def cleanup_old_files():
+    """清理過期的音頻檔案"""
+    try:
+        if not os.path.exists(OUTPUT_DIR):
+            return
+        
+        current_time = time.time()
+        files = []
+        
+        # 獲取所有檔案及其修改時間
+        for filename in os.listdir(OUTPUT_DIR):
+            if filename.endswith('.mp3'):
+                file_path = os.path.join(OUTPUT_DIR, filename)
+                file_mtime = os.path.getmtime(file_path)
+                files.append((file_path, file_mtime, filename))
+        
+        # 按修改時間排序（最新的在前）
+        files.sort(key=lambda x: x[1], reverse=True)
+        
+        deleted_count = 0
+        
+        # 刪除過期檔案
+        for file_path, file_mtime, filename in files:
+            file_age = current_time - file_mtime
+            
+            # 如果檔案超過最大存活時間，刪除它
+            if file_age > FILE_MAX_AGE:
+                try:
+                    os.remove(file_path)
+                    deleted_count += 1
+                    print(f"刪除過期檔案: {filename}")
+                except Exception as e:
+                    print(f"刪除檔案失敗 {filename}: {e}")
+        
+        # 如果檔案數量超過限制，刪除最舊的檔案
+        remaining_files = len(files) - deleted_count
+        if remaining_files > MAX_FILES:
+            files_to_delete = files[MAX_FILES:]
+            for file_path, _, filename in files_to_delete:
+                try:
+                    if os.path.exists(file_path):
+                        os.remove(file_path)
+                        deleted_count += 1
+                        print(f"刪除超量檔案: {filename}")
+                except Exception as e:
+                    print(f"刪除檔案失敗 {filename}: {e}")
+        
+        if deleted_count > 0:
+            print(f"清理完成，共刪除 {deleted_count} 個檔案")
+            
+    except Exception as e:
+        print(f"檔案清理過程中發生錯誤: {e}")
+
+# 背景清理任務
+async def background_cleanup():
+    """背景檔案清理任務"""
+    while True:
+        try:
+            cleanup_old_files()
+            await asyncio.sleep(FILE_CLEANUP_INTERVAL)
+        except Exception as e:
+            print(f"背景清理任務錯誤: {e}")
+            await asyncio.sleep(60)  # 錯誤時等待1分鐘再重試
+
 @app.get("/")
 async def root():
     """根路徑，返回 API 信息"""
@@ -71,9 +143,13 @@ async def root():
             "GET /voices": "獲取所有可用語音",
             "POST /tts": "文字轉語音",
             "GET /tts": "文字轉語音 (GET 方法)",
-            "GET /health": "健康檢查"
+            "GET /health": "健康檢查",
+            "GET /audio/{file_id}.mp3": "獲取音頻文件",
+            "DELETE /audio/{file_id}.mp3": "刪除音頻文件",
+            "POST /cleanup": "手動觸發檔案清理",
+            "GET /storage-info": "獲取儲存空間資訊"
         },
-        "note": "此服務部署在 Hugging Face Spaces 上"
+        "note": "此服務部署在 Hugging Face Spaces 上，具有自動檔案清理功能"
     }
 
 @app.get("/health")
@@ -235,6 +311,61 @@ async def delete_audio_file(file_id: str):
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"文件刪除失敗: {str(e)}")
 
+@app.post("/cleanup")
+async def manual_cleanup():
+    """手動觸發檔案清理"""
+    try:
+        cleanup_old_files()
+        return {"success": True, "message": "手動清理完成"}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"清理失敗: {str(e)}")
+
+@app.get("/storage-info")
+async def get_storage_info():
+    """獲取儲存空間資訊"""
+    try:
+        if not os.path.exists(OUTPUT_DIR):
+            return {"files": 0, "total_size": 0, "max_files": MAX_FILES, "max_age": FILE_MAX_AGE}
+        
+        files = []
+        total_size = 0
+        
+        for filename in os.listdir(OUTPUT_DIR):
+            if filename.endswith('.mp3'):
+                file_path = os.path.join(OUTPUT_DIR, filename)
+                file_size = os.path.getsize(file_path)
+                file_mtime = os.path.getmtime(file_path)
+                file_age = time.time() - file_mtime
+                
+                files.append({
+                    "filename": filename,
+                    "size": file_size,
+                    "age_seconds": int(file_age),
+                    "age_minutes": int(file_age / 60)
+                })
+                total_size += file_size
+        
+        return {
+            "files": len(files),
+            "total_size": total_size,
+            "total_size_mb": round(total_size / (1024 * 1024), 2),
+            "max_files": MAX_FILES,
+            "max_age_seconds": FILE_MAX_AGE,
+            "max_age_minutes": int(FILE_MAX_AGE / 60),
+            "cleanup_interval_seconds": FILE_CLEANUP_INTERVAL,
+            "cleanup_interval_minutes": int(FILE_CLEANUP_INTERVAL / 60),
+            "file_list": files[:10]  # 只顯示前10個檔案
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"獲取儲存資訊失敗: {str(e)}")
+
+# 應用啟動事件
+@app.on_event("startup")
+async def startup_event():
+    """應用啟動時啟動背景清理任務"""
+    asyncio.create_task(background_cleanup())
+    print("背景檔案清理任務已啟動")
+
 # 清理函數（可選）
 @app.on_event("shutdown")
 async def cleanup():