Update main.py

main.py

@@ -455,10 +455,18 @@ EXEMPLOS
                     # Usar o vídeo cortado se disponível, senão o original
                     video_for_export = cropped_video_path if cropped_video_path and os.path.exists(cropped_video_path) else temp_file.name
                     
-                    # 1. Screenshot do vídeo cortado (frame no segundo 1)
+                    # 1. Screenshot do vídeo cortado (frame do MEIO)
                     screenshot_path = tempfile.NamedTemporaryFile(delete=False, suffix='.jpg').name
+                    # Obter duração do vídeo para calcular o meio
+                    duration_probe = subprocess.run(
+                        ["ffprobe", "-v", "quiet", "-show_entries", "format=duration", "-of", "csv=p=0", video_for_export],
+                        capture_output=True, text=True
+                    )
+                    video_duration = float(duration_probe.stdout.strip()) if duration_probe.returncode == 0 and duration_probe.stdout.strip() else 10.0
+                    middle_time = str(video_duration / 2)
+                    
                     ffmpeg_ss = subprocess.run(
-                        ["ffmpeg", "-y", "-i", video_for_export, "-ss", "1", "-frames:v", "1", screenshot_path],
+                        ["ffmpeg", "-y", "-i", video_for_export, "-ss", middle_time, "-frames:v", "1", screenshot_path],
                         capture_output=True, text=True
                     )
                     if ffmpeg_ss.returncode != 0:
@@ -494,7 +502,20 @@ EXEMPLOS
                             # URL do vídeo para o export: cortado se disponível, senão original
                             export_video_url = cropped_video_url if cropped_video_url else video_url
                             
-                            # 4. Montar title_url
+                            # 4. Obter dimensões do vídeo para cálculo horizontal/vertical
+                            import cv2
+                            cap = cv2.VideoCapture(video_for_export)
+                            crop_w = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)) if cap.isOpened() else 1080
+                            crop_h = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)) if cap.isOpened() else 1920
+                            cap.release()
+                            
+                            canvas_width = 1080
+                            image_height = 616
+                            video_x = int((canvas_width - crop_w) / 2)
+                            video_template = "horizontal" if crop_w > crop_h else "vertical"
+                            print(f"📐 Dimensões do vídeo: {crop_w}x{crop_h} | Template: {video_template}")
+                            
+                            # 5. Montar title_url
                             import urllib.parse
                             title_params = urllib.parse.urlencode({
                                 "text": title_text,
@@ -503,11 +524,19 @@ EXEMPLOS
                             })
                             title_url = f"https://habulaj-recurve-api-img.hf.space/cover/title?{title_params}"
                             
-                            # 5. Chamar API de video export
+                            # 6. Chamar API de video export
                             print(f"🎬 Chamando video export API...")
                             export_payload = {
                                 "video_url": export_video_url,
-                                "title_url": title_url
+                                "title_url": title_url,
+                                "image_height": image_height,
+                                "cut_start": 0,
+                                "cut_end": video_duration,
+                                "video_x": video_x,
+                                "video_y": 0,
+                                "video_width": crop_w,
+                                "video_height": crop_h,
+                                "video_template": video_template
                             }
                             export_resp = requests.post(
                                 "https://habulaj-recurve-videos-export.hf.space/video/export",