Update app.py

app.py

@@ -97,7 +97,7 @@ async def generate_video(client, prompt, loop=False, aspect_ratio="16:9", camera
                         fd.write(chunk)
     
     progress(1.0, desc="비디오 생성 완료!")
-    return file_name
+    return file_name, generation.id
 
 async def translate_prompt(prompt):
     try:
@@ -127,13 +127,13 @@ async def text_to_video(prompt, loop, aspect_ratio, camera_motion, extend_id, re
         if interpolate_id1 and interpolate_id2:
             interpolate_ids = [interpolate_id1, interpolate_id2]
 
-        video_path = await generate_video(
+        video_path, video_id = await generate_video(
             client, translated_prompt, loop, aspect_ratio, camera_motion, 
             extend_id, reverse_extend_id, interpolate_ids, progress
         )
-        return video_path, ""
+        return video_path, video_id, ""
     except Exception as e:
-        return None, f"오류 발생: {str(e)}"
+        return None, None, f"오류 발생: {str(e)}"
 
 async def image_to_video(prompt, image_url, loop, aspect_ratio, camera_motion, progress=gr.Progress()):
     api_key = os.getenv("LMGEN_KEY")
@@ -200,9 +200,9 @@ async def image_to_video(prompt, image_url, loop, aspect_ratio, camera_motion, p
                             fd.write(chunk)
         
         progress(1.0, desc="비디오 생성 완료!")
-        return file_name, ""
+        return file_name, generation.id, ""
     except Exception as e:
-        return None, f"오류 발생: {str(e)}"
+        return None, None, f"오류 발생: {str(e)}"
 
 with gr.Blocks() as demo:
     gr.Markdown("# Luma AI 텍스트-비디오 생성 데모")
@@ -211,6 +211,7 @@ with gr.Blocks() as demo:
         prompt = gr.Textbox(label="프롬프트 (비디오에 대한 설명을 입력하세요)")
         generate_btn = gr.Button("비디오 생성")
         video_output = gr.Video(label="생성된 비디오")
+        video_id_output = gr.Textbox(label="생성된 비디오 ID", visible=True)
         error_output = gr.Textbox(label="오류 메시지", visible=True)
         
         with gr.Accordion("고급 옵션", open=False):
@@ -230,7 +231,7 @@ with gr.Blocks() as demo:
         generate_btn.click(
             text_to_video,
             inputs=[prompt, loop, aspect_ratio, camera_motion, extend_id, reverse_extend_id, interpolate_id1, interpolate_id2],
-            outputs=[video_output, error_output]
+            outputs=[video_output, video_id_output, error_output]
         )
     
     with gr.Tab("이미지로 비디오 만들기"):
@@ -238,6 +239,7 @@ with gr.Blocks() as demo:
         img_url = gr.Textbox(label="이미지 URL (변환하고자 하는 이미지의 웹 주소)")
         img_generate_btn = gr.Button("이미지로 비디오 생성")
         img_video_output = gr.Video(label="생성된 비디오")
+        img_video_id_output = gr.Textbox(label="생성된 비디오 ID", visible=True)
         img_error_output = gr.Textbox(label="오류 메시지", visible=True)
         
         with gr.Accordion("고급 옵션", open=False):
@@ -252,7 +254,7 @@ with gr.Blocks() as demo:
         img_generate_btn.click(
             image_to_video,
             inputs=[img_prompt, img_url, img_loop, img_aspect_ratio, img_camera_motion],
-            outputs=[img_video_output, img_error_output]
+            outputs=[img_video_output, img_video_id_output, img_error_output]
         )
 
     async def update_camera_motions():