Update app.py

app.py

@@ -107,22 +107,36 @@ def generate_status(content_id):
     url = API_HOME + END_STATUS
     payload = {"id": content_id}
     headers = set_header()
-    response = requests.request("POST", url, json=payload, headers=headers).json()
 
-    # HTTP 상태코드 확인
-    if response['success'] == True:
-        print(f"정상: {response}")
-        if response['data']['status'] == 'queued':
-            print('아직 컨텐츠 URL이 생성되지 않았습니다.')
-            return 'PENDING'
-        if response['data']['status'] == 'processing':
-            print('컨텐츠 URL이 아직 생성 중 입니다.')
-            return 'PENDING' 
-        elif response['data']['status'] == 'success':
-            return response['data']['file']
+    while True:
+        response = requests.request("POST", url, json=payload, headers=headers).json()
+
+        if response['success']:
+            if response['data']['status'] == 'queued' or response['data']['status'] == 'processing':
+                print('컨텐츠 생성 중... 잠시 후 다시 시도하세요.')
+                time.sleep(10)  # 10초 대기
+            elif response['data']['status'] == 'success':
+                print('컨텐츠 생성 완료!')
+                return response['data']['file']
+        else:
+            print(f"request 실패: {response}")
+            break
+    return 'FAILED'
 
+def gradio_generate_status(content_id):
+    download_url = generate_status(content_id)
+    if download_url != 'FAILED':
+        return f"다운로드 링크: {download_url}"
     else:
-        print(f"request 실패")
+        return "컨텐츠 생성 실패 또는 ID가 잘못되었습니다."
+
+status_iface = gr.Interface(
+    fn=gradio_generate_status,  # 컨텐츠 상태 확인 및 다운로드 링크 제공 함수
+    inputs=gr.Textbox(label="컨텐츠 ID"),
+    outputs=gr.Textbox(label="다운로드 링크 또는 메시지")
+)
+
+status_iface.launch()
 
 
 def download_content(url):