Update src/api/main.py

src/api/main.py

@@ -1,70 +1,22 @@
-import uuid
-import re
-import logging
-from datetime import datetime
-from typing import Dict, Optional
-from enum import Enum
 from contextlib import asynccontextmanager
-
-from fastapi import FastAPI, HTTPException, BackgroundTasks, Depends
+from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from pydantic import BaseModel, HttpUrl
-
-# --- استيراد الأدوات الجديدة (POT Solver & New Downloader) ---
-from src.api.downloader import YouTubeDownloader 
 from src.api.pot_server import pot_server
-
-# --- استيراد مكونات النظام ---
-from src.transcription.transcript_fetcher import TranscriptFetcher
-from src.transcription.whisper_transcriber import WhisperTranscriber
-from src.summarization.note_generator import NoteGenerator
-from src.utils.logger import setup_logger
-from src.db.firebase import get_firebase_db
-from src.db.models import Note, User
-from src.auth.dependencies import get_current_user
-from src.api.auth_routes import router as auth_router
 from src.api.notes_routes import router as notes_router
+from src.api.auth_routes import router as auth_router
+# استورد أي routers تانية عندك هنا
 
-logger = setup_logger(__name__)
-
-# --- Models ---
-class TaskStatus(str, Enum):
-    PENDING = "pending"
-    DOWNLOADING = "downloading"
-    TRANSCRIBING = "transcribing"
-    GENERATING_NOTES = "generating_notes"
-    COMPLETED = "completed"
-    FAILED = "failed"
-
-class GenerateNotesRequest(BaseModel):
-    youtube_url: HttpUrl
-    language: str = "en"
-
-class TaskResponse(BaseModel):
-    task_id: str
-    status: TaskStatus
-    message: str
-
-# مخزن المهام المؤقت
-tasks: Dict[str, Dict] = {}
-
-# --- إدارة دورة حياة السيرفر (Lifespan) ---
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    logger.info("🚀 Lifespan: Initializing Firebase...")
-    get_firebase_db()
-    
-    logger.info("🚀 Lifespan: Starting POT solver server (bgutil v1.3.1)...")
-    # تشغيل سيرفر الـ Node.js في الخلفية لحل تحديات يوتيوب
-    pot_server.start() 
-    
+    print("🚀 Starting POT solver server...")
+    pot_server.start()
     yield
-    
-    logger.info("🛑 Lifespan: Stopping POT solver server...")
+    print("🛑 Stopping POT solver server...")
     pot_server.stop()
 
-app = FastAPI(title="YouTube Study Notes AI", lifespan=lifespan)
+app = FastAPI(title="AIdea API", lifespan=lifespan)
 
+# تفعيل الـ CORS عشان الـ Flutter Web يشتغل
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -73,163 +25,10 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# --- تسجيل الـ Routers ---
-app.include_router(auth_router)
+# تسجيل المسارات (مهم جداً عشان الـ 404 تختفي)
 app.include_router(notes_router)
-
-# --- Endpoints الأساسية ---
+app.include_router(auth_router)
 
 @app.get("/")
 def read_root():
-    return {"message": "Welcome to AIdea API! The server is running perfectly with POT Solver."}
-
-@app.get("/health")
-async def health_check():
-    """اختبار حالة السيرفر والاتصال بيوتيوب"""
-    import socket
-    import httpx
-    dns_results = {}
-    for domain in ["www.youtube.com", "google.com"]:
-        try:
-            dns_results[domain] = socket.gethostbyname(domain)
-        except:
-            dns_results[domain] = "Error"
-            
-    return {
-        "status": "online",
-        "pot_server_running": pot_server._proc is not None,
-        "dns": dns_results,
-        "timestamp": datetime.now()
-    }
-
-@app.post("/generate", response_model=TaskResponse)
-async def generate(
-    request: GenerateNotesRequest,
-    background_tasks: BackgroundTasks,
-    current_user: User = Depends(get_current_user),
-):
-    task_id = str(uuid.uuid4())
-    user_id = current_user.id
-
-    tasks[task_id] = {
-        "status": TaskStatus.PENDING,
-        "message": "Initializing...",
-        "youtube_url": str(request.youtube_url),
-        "user_id": user_id,
-        "created_at": datetime.now(),
-    }
-
-    background_tasks.add_task(
-        process_video_and_save,
-        task_id,
-        str(request.youtube_url),
-        request.language,
-        user_id,
-    )
-
-    return TaskResponse(
-        task_id=task_id,
-        status=TaskStatus.PENDING,
-        message="Generation started successfully.",
-    )
-
-@app.get("/status/{task_id}")
-async def get_task_status(task_id: str):
-    if task_id not in tasks:
-        raise HTTPException(status_code=404, detail="Task not found")
-    return tasks[task_id]
-
-# --- منطق معالجة الفيديو (Logic) ---
-
-async def transcribeAudio(task_id: str, youtube_url: str, language: str):
-    audio_file = None
-    downloader = YouTubeDownloader() # يستخدم النسخة الجديدة في src/api/downloader.py
-    
-    # استخراج الـ Video ID للتعامل مع الملفات
-    video_id_match = re.search(r"(?:v=|youtu\.be/)([A-Za-z0-9_-]{11})", youtube_url)
-    video_id = video_id_match.group(1) if video_id_match else task_id
-
-    video_info = {"title": "YouTube Video", "duration": 0}
-
-    tasks[task_id]["status"] = TaskStatus.TRANSCRIBING
-    fetcher = TranscriptFetcher()
-    
-    try:
-        # محاولة جلب الترجمة الجاهزة أولاً (أسرع)
-        transcript_text = fetcher.fetch_transcript(youtube_url, languages=[language, 'ar', 'en'])
-    except Exception as e:
-        logger.error(f"Transcript fetching failed: {e}")
-        transcript_text = None
-    
-    if transcript_text:
-        logger.info(f"✅ Using direct YouTube transcript for {video_id}")
-    else:
-        # لو فشل نجلب الصوت باستخدام yt-dlp + POT Solver
-        logger.info(f"🚀 Falling back to audio download + Whisper for {video_id}")
-        tasks[task_id]["status"] = TaskStatus.DOWNLOADING
-        
-        audio_file = downloader.download_audio(youtube_url, video_id)
-        
-        if not audio_file:
-            raise Exception("Failed to download audio even with POT Solver.")
-
-        tasks[task_id]["status"] = TaskStatus.TRANSCRIBING
-        transcriber = WhisperTranscriber()
-        transcript_data = transcriber.transcribe(str(audio_file), language=language)
-        transcript_text = transcript_data["text"]
-
-    return transcript_text, video_info, audio_file, downloader
-
-async def process_video_and_save(task_id: str, youtube_url: str, language: str, user_id: str):
-    audio_file = None
-    downloader = None
-    try:
-        # 1. الحصول على النص (Transcribe)
-        transcript_text, video_info, audio_file, downloader = await transcribeAudio(
-            task_id, youtube_url, language
-        )
-
-        # 2. توليد الملخص (Generate Summary)
-        tasks[task_id]["status"] = TaskStatus.GENERATING_NOTES
-        note_gen = NoteGenerator()
-        summary_json = note_gen.generateSummary(transcript_text, video_info["title"])
-        
-        # 3. استخراج أهم النقاط (Action Items)
-        action_items_json = note_gen.extractActionItems(transcript_text, video_info["title"])
-
-        combined_json = {
-            **summary_json,
-            "action_items": action_items_json.get("action_items", [])
-        }
-        
-        final_notes = note_gen.format_final_notes(
-            note_gen.format_notes_to_markdown(combined_json),
-            video_info["title"],
-            youtube_url,
-            video_info["duration"],
-        )
-
-        # 4. الحفظ في Firebase
-        db = get_firebase_db()
-        if db:
-            note_data = {
-                "user_id": user_id,
-                "video_url": youtube_url,
-                "video_title": video_info.get("title", "YouTube Video"),
-                "summary_content": final_notes,
-                "created_at": datetime.utcnow()
-            }
-            db.collection("notes").add(note_data)
-
-        tasks[task_id]["notes"] = final_notes
-        tasks[task_id]["status"] = TaskStatus.COMPLETED
-        logger.info(f"✅ Task {task_id} completed successfully!")
-
-    except Exception as e:
-        logger.error(f"❌ Task {task_id} failed: {str(e)}")
-        tasks[task_id]["status"] = TaskStatus.FAILED
-        tasks[task_id]["message"] = f"Error: {str(e)}"
-    finally:
-        # تنظيف الملفات المؤقتة
-        if downloader and audio_file:
-            downloader.cleanup(audio_file)
+    return {"message": "AIdea API is Live and POT is working!"}