Save local changes (GSC/Bing integrations) before merging PR #354

frontend/src/api/videoStudioApi.ts

@@ -3,6 +3,8 @@
  */
 
 import { aiApiClient } from './client';
+// Import TaskStatusResponse from blogWriterApi to ensure compatibility with usePolling
+import type { TaskStatusResponse } from '../services/blogWriterApi';
 
 const API_BASE = '/api/video-studio';
 
@@ -18,6 +20,17 @@ export interface PromptOptimizeResponse {
   success: boolean;
 }
 
+export interface CreateAvatarVideoResponse {
+  task_id: string;
+  status: string;
+  message: string;
+  error?: string;
+  result?: {
+      video_url: string;
+      [key: string]: any;
+  };
+}
+
 /**
  * Optimize a prompt using WaveSpeed prompt optimizer
  */
@@ -30,3 +43,77 @@ export async function optimizePrompt(
   );
   return response.data;
 }
+
+/**
+ * Create a talking avatar video asynchronously
+ * Uses dedicated Video Studio endpoint for generic avatar generation
+ */
+export async function createAvatarVideoAsync(
+  imageFile: File,
+  audioFile: File,
+  resolution: '480p' | '720p' = '720p',
+  model: 'infinitetalk' | 'hunyuan-avatar' = 'infinitetalk'
+): Promise<CreateAvatarVideoResponse> {
+  const formData = new FormData();
+  formData.append('image', imageFile);
+  formData.append('audio', audioFile);
+  formData.append('resolution', resolution);
+  formData.append('model', model);
+
+  try {
+    const response = await aiApiClient.post<CreateAvatarVideoResponse>(
+      `${API_BASE}/avatar/create-async`,
+      formData,
+      {
+        headers: {
+          'Content-Type': 'multipart/form-data',
+        },
+      }
+    );
+    return response.data;
+  } catch (error: any) {
+    console.error('Error creating avatar video:', error);
+    throw new Error(error.response?.data?.detail || 'Failed to create avatar video');
+  }
+}
+
+/**
+ * Get the status of a video generation task
+ */
+export async function getVideoTaskStatus(taskId: string): Promise<CreateAvatarVideoResponse> {
+    try {
+        const response = await aiApiClient.get<CreateAvatarVideoResponse>(
+            `${API_BASE}/task/${taskId}`
+        );
+        return response.data;
+    } catch (error: any) {
+        console.error('Error fetching video task status:', error);
+        throw new Error(error.response?.data?.detail || 'Failed to fetch task status');
+    }
+}
+
+/**
+ * Poll video task status compatible with usePolling hook
+ */
+export async function pollVideoTaskStatus(taskId: string): Promise<TaskStatusResponse<{ video_url: string; [key: string]: any }>> {
+  const data = await getVideoTaskStatus(taskId);
+  
+  // Map CreateAvatarVideoResponse to TaskStatusResponse
+  // Ensure we map 'processing' to 'running' for frontend consistency
+  let status: 'pending' | 'running' | 'completed' | 'failed' = 'pending';
+  
+  if (data.status === 'completed') status = 'completed';
+  else if (data.status === 'failed') status = 'failed';
+  else if (data.status === 'running' || data.status === 'processing') status = 'running';
+  else status = 'pending';
+
+  return {
+    task_id: data.task_id,
+    status: status,
+    progress_messages: [], // Video Studio currently doesn't return progress messages
+    result: data.result,
+    error: data.error,
+    // Add default values for missing fields
+    created_at: new Date().toISOString(),
+  };
+}