Update app.py

app.py

@@ -48,19 +48,37 @@ def cleanup_directory(directory_path: str):
         shutil.rmtree(directory_path, ignore_errors=True)
         print(f"--- CLEANUP: Successfully removed {directory_path} ---")
 
-async def run_subprocess(command: str):
+async def run_subprocess(command: str) -> (str, str):
+    """
+    Executes a shell command asynchronously and returns its outputs.
+    Raises an exception if the command fails.
+    
+    Args:
+        command (str): The shell command to execute.
+
+    Returns:
+        tuple[str, str]: A tuple containing (stdout, stderr).
+
+    Raises:
+        RuntimeError: If the command returns a non-zero exit code.
+    """
     print(f"Executing command: {command}")
     process = await asyncio.create_subprocess_shell(
-        command, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE
+        command,
+        stdout=asyncio.subprocess.PIPE,
+        stderr=asyncio.subprocess.PIPE
     )
     stdout, stderr = await process.communicate()
     
     if process.returncode != 0:
         error_message = stderr.decode(errors='ignore')
-        print(f"FFmpeg Error: {error_message}")
+        print(f"ERROR: Command failed with exit code {process.returncode}\nStderr: {error_message}")
+        # Levanta uma exceção para que a função que a chamou possa tratá-la.
         raise RuntimeError(f"FFmpeg command failed: {error_message}")
     
-    print("Command executed successfully.")
+    print("SUCCESS: Command executed.")
+    # Retorna os outputs para a função que a chamou.
+    return stdout.decode(errors='ignore'), stderr.decode(errors='ignore')
 
 # ==============================================================================
 # 3. API ENDPOINTS
@@ -143,82 +161,37 @@ async def process_video_from_urls(payload: Dict = Body(...)):
 async def get_duration(file: UploadFile = File(...)):
     """
     Calculates the duration of a media file using FFprobe.
-
-    This endpoint accepts a binary file upload, saves it temporarily, runs
-    ffprobe to get its duration, and returns the duration in seconds.
-    It ensures temporary files are always cleaned up.
-
-    Args:
-        file (UploadFile): The media file (audio or video) uploaded by the client.
-
-    Returns:
-        JSONResponse: A JSON object containing the duration in seconds,
-                      e.g., {"duration_seconds": 146.5}.
-    
-    Raises:
-        HTTPException(400): If the duration cannot be parsed from ffprobe's output.
-        HTTPException(500): If ffprobe fails or an internal error occurs.
     """
-    # --- Setup Temporary Environment ---
-    # Create a unique filename to prevent conflicts
     unique_id = str(uuid.uuid4())
     temp_input_path = os.path.join(TEMP_DIR, f"{unique_id}_{file.filename}")
 
     try:
-        # --- Step 1: Save Uploaded File ---
-        print(f"Saving uploaded file to {temp_input_path}...")
-        try:
-            with open(temp_input_path, "wb") as buffer:
-                shutil.copyfileobj(file.file, buffer)
-            print("File saved successfully.")
-        except Exception as e:
-            # Handle potential file writing errors
-            raise HTTPException(status_code=500, detail=f"Failed to write uploaded file to disk: {e}")
-
-        # --- Step 2: Construct and Execute FFprobe Command ---
-        # Using shlex.quote for security against command injection
+        with open(temp_input_path, "wb") as buffer:
+            shutil.copyfileobj(file.file, buffer)
+
         quoted_path = shlex.quote(temp_input_path)
-        
-        # This ffprobe command is optimized to return ONLY the duration number
         command = f"ffprobe -v error -show_entries format=duration -of default=noprint_wrappers=1:nokey=1 {quoted_path}"
         
-        if not await run_subprocess(command):
-             # The run_subprocess helper now handles logging the error
-             raise HTTPException(status_code=500, detail="FFprobe command failed. Check server logs.")
-
-        # Re-run the command to capture the output, as the helper only returns success/fail
-        process = await asyncio.create_subprocess_shell(
-            command,
-            stdout=asyncio.subprocess.PIPE,
-            stderr=asyncio.subprocess.PIPE
-        )
-        stdout, stderr = await process.communicate()
+        # Agora, run_subprocess retorna o stdout se for bem-sucedido.
+        stdout, _ = await run_subprocess(command)
         
-        duration_str = stdout.decode().strip()
+        duration_str = stdout.strip()
         print(f"FFprobe output (duration): '{duration_str}'")
 
-        # --- Step 3: Parse and Return Duration ---
         if not duration_str:
-            raise ValueError("FFprobe returned an empty string.")
+            raise ValueError("FFprobe returned an empty string for duration.")
 
         duration = float(duration_str)
         return JSONResponse(content={"duration_seconds": duration})
 
     except ValueError:
-        # This catches errors from float() if ffprobe's output is not a number
         raise HTTPException(status_code=400, detail="Could not parse a valid duration from the media file.")
-    
     except Exception as e:
-        # Catch-all for any other unexpected errors
-        print(f"An unexpected error occurred in get_duration: {e}")
+        # Captura qualquer outro erro, incluindo o RuntimeError do run_subprocess.
         raise HTTPException(status_code=500, detail=f"An internal server error occurred: {str(e)}")
-    
     finally:
-        # --- Final Step: Cleanup ---
-        # This block ALWAYS runs, ensuring no temporary files are left behind.
         if os.path.exists(temp_input_path):
             os.remove(temp_input_path)
-            print(f"Cleaned up temporary file: {temp_input_path}")
 
 # ==============================================================================
 # 4. ENDPOINT PARA ADICIONAR LEGENDAS