Update app.py

app.py

@@ -11,7 +11,7 @@ app = FastAPI(title="StudyLoop API", description="Extract YouTube transcripts")
 # Enable CORS for your frontend
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # For development, restrict in production
+    allow_origins=["*"],
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -20,12 +20,6 @@ app.add_middleware(
 class YouTubeRequest(BaseModel):
     url: str
 
-class TranscriptResponse(BaseModel):
-    success: bool
-    transcript: Optional[str] = None
-    error: Optional[str] = None
-    video_id: Optional[str] = None
-
 def extract_video_id(url: str) -> Optional[str]:
     patterns = [
         r'(?:youtube\.com\/watch\?v=)([^&]+)',
@@ -51,22 +45,21 @@ async def health():
 async def get_transcript(request: YouTubeRequest):
     video_id = extract_video_id(request.url)
     if not video_id:
-        return TranscriptResponse(success=False, error="Invalid YouTube URL", video_id=None)
+        return {"success": False, "error": "Invalid YouTube URL", "transcript": None}
     
     try:
         transcript_list = YouTubeTranscriptApi.get_transcript(video_id, languages=['en'])
         formatter = TextFormatter()
         text = formatter.format_transcript(transcript_list)
-        return TranscriptResponse(success=True, transcript=text, error=None, video_id=video_id)
+        return {"success": True, "transcript": text, "video_id": video_id}
     except Exception as e:
-        # Try without language filter
         try:
             transcript_list = YouTubeTranscriptApi.get_transcript(video_id)
             formatter = TextFormatter()
             text = formatter.format_transcript(transcript_list)
-            return TranscriptResponse(success=True, transcript=text, error=None, video_id=video_id)
+            return {"success": True, "transcript": text, "video_id": video_id}
         except Exception as e2:
-            return TranscriptResponse(success=False, error=f"Could not fetch transcript: {str(e2)}", video_id=video_id)
+            return {"success": False, "error": f"Could not fetch transcript: {str(e2)}", "transcript": None}
 
 @app.post("/extract")
 async def extract(request: YouTubeRequest):