Update app.py

app.py

@@ -1,11 +1,28 @@
 import gradio as gr
 import os
 import subprocess
+from fastapi import FastAPI, Request
+from starlette.responses import JSONResponse
+import uvicorn
 
-# Función para iniciar la transmisión
+app = FastAPI()
+
+# Ruta para recibir el stream de video
+@app.post("/save_stream")
+async def save_stream(request: Request):
+    # Recibe el stream de video y lo almacena o lo procesa en tiempo real con FFmpeg
+    body = await request.body()
+    with open("temp_stream.webm", "ab") as f:
+        f.write(body)
+    
+    # Aquí podrías procesar el archivo con FFmpeg en tiempo real para enviarlo al servidor RTMP
+
+    return JSONResponse({"status": "ok"})
+
+# Función para iniciar la transmisión (ejecutada al presionar el botón en Gradio)
 def start_streaming(rtmp_server, stream_key):
-    # Comando FFmpeg para la transmisión
-    command = f"ffmpeg -f x11grab -i :0.0 -c:v libx264 -b:v 1000k -f flv {rtmp_server}/{stream_key}"
+    # Comando FFmpeg para procesar el archivo temporal y transmitirlo
+    command = f"ffmpeg -re -i temp_stream.webm -c:v libx264 -b:v 1000k -f flv {rtmp_server}/{stream_key}"
     process = subprocess.Popen(command, shell=True)
     return "Transmisión en vivo iniciada"
 
@@ -14,10 +31,17 @@ def build_interface():
     with gr.Blocks() as demo:
         gr.Markdown("# Transmisión en Vivo")
 
+        # Insertar el script para capturar pantalla
+        gr.HTML("""
+        <script src="screen_capture.js"></script>
+        <video id="screenPreview" style="width:100%;"></video>  <!-- Previsualización de pantalla -->
+        <button onclick="startScreenCapture()">Compartir Pantalla</button>
+        """)
+
         # Entrada para el servidor RTMP
         rtmp_server = gr.Textbox(label="Servidor RTMP (YouTube, Twitch, etc.)", placeholder="Ingresa el servidor RTMP")
 
-        # Entrada para la clave de transmisión
+        # Entrada para la clave de transmisión (ocultada)
         stream_key = gr.Textbox(label="Clave de transmisión", placeholder="Ingresa tu clave de transmisión", type="password")
 
         # Botón para iniciar la transmisión
@@ -31,6 +55,8 @@ def build_interface():
 
     return demo
 
-# Ejecuta la interfaz
+# Inicia Gradio y el backend de FastAPI
 demo = build_interface()
-demo.launch()
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)