Para que podamos crear una acción dependiendo de la opción del usuario, debemos identificar el valor de la variable “opcion”. Para esto, podemos implementar una estructura condicional “if else”.
Si el usuario ingresa el valor “1”, le solicitaremos que ingrese el nombre del video. Si el usuario ingresa el valor “2”, le solicitaremos que ingrese la URL del video de YouTube para posteriormente llamar y enviarle la URL como parámetro a una función llamada “descargar_youtube(url)”.
Si el usuario ingresa un dato incorrecto, se le mostrará en pantalla “Opción no válida” y el programa se cerrará.
Aquí está el código:
if opcion == 1:
archivo_video = input("Ingrese el nombre del video: ")
elif opcion == 2:
url = input("Ingrese la URL del video de YouTube: ")
archivo_video = descargar_youtube(url)
else:
print("Opción no válida.")
sys.exit()
Todo nuestro código que llevamos hasta el momento se debería de ver así:
import openai
import sys
import os
import whisper
import pytube
print("¿Qué tipo de video desea analizar?")
print("1. Video descargado")
print("2. Video de YouTube")
opcion = int(input("Elija una opción: "))
if opcion == 1:
archivo_video = input("Ingrese el nombre del video: ")
elif opcion == 2:
url = input("Ingrese la URL del video de YouTube: ")
archivo_video = descargar_youtube(url)
else:
print("Opción no válida.")
sys.exit()
Nota del maestro: Recuerda que este curso es gratuito y no cuenta con soporte. Si tienes problemas o dudas, puedes escribirme en Twitter haciendo clic aquí y esperar a que tenga un tiempo libre para poder contestarte. También puedes considerar tomar el curso profesional en nuestra plataforma de Azul School, que incluye soporte prioritario para todos nuestros cursos y muchos más beneficios. Haz clic aquí para comprar tu membresía anual u obtener más información al respecto.