Update server.py

server.py

@@ -5,7 +5,7 @@ import os
 import pickle
 import base64
 
-# Load YouTube token from environment
+# Load YouTube token (Channel B - plipStory)
 token_b64 = os.environ.get('YOUTUBE_TOKEN')
 if token_b64:
     try:
@@ -17,6 +17,7 @@ if token_b64:
         print(f"Token error: {e}")
 
 class Handler(BaseHTTPRequestHandler):
+
     def do_GET(self):
         self.send_response(200)
         self.end_headers()
@@ -24,70 +25,50 @@ class Handler(BaseHTTPRequestHandler):
 
     def do_POST(self):
         try:
-            content_length = self.headers.get('Content-Length')
-            if not content_length:
-                self.send_response(400)
-                self.end_headers()
-                self.wfile.write(b'{"error": "No Content-Length"}')
-                return
-            length = int(content_length)
+            length = int(self.headers['Content-Length'])
             body = json.loads(self.rfile.read(length))
+            path = self.path
+
+            # ── Existing: generate Short from script ──────────────────
+            if path == '/' or path == '':
+                script      = body.get('script', '')
+                title       = body.get('title', 'YouTube Short')
+                description = body.get('description', '')
+
+                print(f"Generating video for: {title}")
+                subprocess.Popen(['python', '/app/generate.py', script, title, description])
+
+                self._respond(200, '{"status": "video generation started"}')
+
+            # ── NEW: mirror a YouTube video to Channel B ───────────────
+            elif path == '/mirror':
+                video_id    = body.get('video_id', '')
+                title       = body.get('title', 'Mirrored Video')
+                description = body.get('description', '')
 
-            # FB Mirror endpoint
-            if self.path == '/fb-mirror':
-                video_url = body.get('video_url', '')
-                title = body.get('title', 'Facebook Short')
-                description = body.get('description', '') + '\n#Shorts'
-
-                fb_path = '/tmp/fb_input.mp4'
-
-                result = subprocess.run([
-                    'yt-dlp',
-                    '--no-check-certificate',
-                    '--user-agent', 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
-                    '-o', fb_path,
-                    '-f', 'best[ext=mp4]/best',
-                    '--no-playlist',
-                    video_url
-                ], capture_output=True, text=True)
-
-                print("yt-dlp stdout:", result.stdout)
-                print("yt-dlp stderr:", result.stderr)
-
-                if result.returncode != 0:
-                    self.send_response(500)
-                    self.end_headers()
-                    self.wfile.write(f'{{"error": "yt-dlp failed: {result.stderr[:200]}"}}'.encode())
+                if not video_id:
+                    self._respond(400, '{"error": "video_id is required"}')
                     return
 
+                print(f"Mirroring video_id: {video_id}")
                 subprocess.Popen([
-                    'python', '/app/upload.py',
-                    fb_path, title, description
+                    'python', '/app/mirror.py',
+                    video_id, title, description
                 ])
 
-                self.send_response(200)
-                self.end_headers()
-                self.wfile.write(b'{"status": "fb mirror started"}')
-                return
+                self._respond(200, f'{{"status": "mirror started", "video_id": "{video_id}"}}')
 
-            # Original flow
-            script = body.get('script', '')
-            title = body.get('title', 'YouTube Short')
-            description = body.get('description', '')
-
-            subprocess.Popen([
-                'python', '/app/generate.py',
-                script, title, description
-            ])
-
-            self.send_response(200)
-            self.end_headers()
-            self.wfile.write(b'{"status": "video generation started"}')
+            else:
+                self._respond(404, '{"error": "not found"}')
 
         except Exception as e:
-            self.send_response(500)
-            self.end_headers()
-            self.wfile.write(f'{{"error": "{str(e)}"}}'.encode())
+            self._respond(500, f'{{"error": "{str(e)}"}}')
+
+    def _respond(self, code, body):
+        self.send_response(code)
+        self.send_header('Content-Type', 'application/json')
+        self.end_headers()
+        self.wfile.write(body.encode())
 
     def log_message(self, format, *args):
         pass