Update app.py

app.py

@@ -12,7 +12,9 @@ import json
 import time
 import requests
 import random
-from google_drive_upload import authenticate_google_drive, upload_to_google_drive
+from googleapiclient.discovery import build
+from google.oauth2 import service_account
+from googleapiclient.http import MediaFileUpload
 from io import BytesIO
 
 # Crear el archivo de credenciales de servicio desde los secretos
@@ -171,6 +173,38 @@ def combine_audio_video(audio_file, video_clip, music_clip=None):
             final_clip.close()
         return None
 
+def upload_to_google_drive(file_path, folder_id):
+    """Sube un archivo a Google Drive y devuelve el enlace público."""
+    try:
+        # Autenticación
+        creds = service_account.Credentials.from_service_account_file(
+            'service-account.json', scopes=['https://www.googleapis.com/auth/drive']
+        )
+        service = build('drive', 'v3', credentials=creds)
+
+        # Subir el archivo
+        file_metadata = {
+            'name': os.path.basename(file_path),
+            'parents': [folder_id]
+        }
+        media = MediaFileUpload(file_path, resumable=True)
+        file = service.files().create(body=file_metadata, media_body=media, fields='id').execute()
+
+        # Hacer el archivo público
+        permission = {
+            'type': 'anyone',
+            'role': 'reader'
+        }
+        service.permissions().create(fileId=file['id'], body=permission).execute()
+
+        # Generar el enlace de descarga
+        file_id = file['id']
+        download_link = f"https://drive.google.com/uc?export=download&id={file_id}"
+        return download_link
+    except Exception as e:
+        print(f"Error subiendo a Google Drive: {e}")
+        return None
+
 def process_input(text, txt_file, mp3_file, selected_voice, rate, pitch, keywords):
     """Procesa la entrada del usuario y genera el video final."""
     try:
@@ -197,14 +231,13 @@ def process_input(text, txt_file, mp3_file, selected_voice, rate, pitch, keyword
         final_video_path = combine_audio_video(audio_file, video_clip, music_clip)
         if not final_video_path:
             raise ValueError("Failed to combine audio and video")
-        # Subimos a Google Drive
-        video_id = upload_to_google_drive(final_video_path, folder_id=FOLDER_ID)
-        if video_id:
-            print(f"Video subido a Google Drive con ID: {video_id}")
+        # Subimos a Google Drive y obtenemos el enlace
+        download_link = upload_to_google_drive(final_video_path, folder_id=FOLDER_ID)
+        if download_link:
+            print(f"Video subido a Google Drive. Enlace de descarga: {download_link}")
+            return f"[Descargar video]({download_link})"
         else:
-            print("Error subiendo el video a Google Drive")
-        
-        return final_video_path
+            raise ValueError("Error subiendo el video a Google Drive")
     except Exception as e:
         print(f"Error durante el procesamiento: {e}")
         return None
@@ -228,13 +261,13 @@ with gr.Blocks() as demo:
             rate_slider = gr.Slider(minimum=-50, maximum=50, value=0, label="Speech Rate Adjustment (%)", step=1)
             pitch_slider = gr.Slider(minimum=-20, maximum=20, value=0, label="Pitch Adjustment (Hz)", step=1)
         with gr.Column():
-           output_video = gr.File(label="Generated Video")  # En lugar de gr.Video
+           output_link = gr.Markdown("")  # Mostrar el enlace de descarga
 
     btn = gr.Button("Generate Video")
     btn.click(
         process_input,
         inputs=[text_input, txt_file_input, mp3_file_input, voice_dropdown, rate_slider, pitch_slider, keyword_input],
-        outputs=output_video
+        outputs=output_link
     )
 
 # Leer el puerto asignado por Hugging Face