Update app.py

app.py

@@ -7,6 +7,8 @@ import json
 import time
 
 API_URL = os.environ.get("API_URL")
+LEPTON_API_TOKEN = os.environ.get('LEPTON_API_TOKEN')
+
 supported_languages = ['zh', 'en']
 
 output_dir = 'outputs'
@@ -102,36 +104,57 @@ def predict(prompt, style, audio_file_pth, agree):
     }
     
     start = time.time()
-    # Send the data as a POST request
-    response = requests.post(API_URL, json=data, timeout=60)
+    # # Send the data as a POST request
+    # response = requests.post(API_URL, json=data, timeout=60)
+    
+    from leptonai.client import Client
+    c = Client(API_URL, token=LEPTON_API_TOKEN)
+    response = c.run(**data)
+    
     print(f'Get response successfully within {time.time() - start}')
 
-    # Check the response
-    if response.status_code == 200:
-        try:
-            json_data = json.loads(response.content)
-            text_hint += f"[ERROR] {json_data['error']} \n"
-            gr.Warning(
-                f"[ERROR] {json_data['error']} \n"
-            )
-            return (
-                text_hint,
-                None,
-                None,
-            )
-        except:
-            with open(save_path, 'wb') as f:
-                f.write(response.content)
-    else:
-        text_hint += f"[HTTP ERROR] {response.status_code} - {response.text} \n"
+    try:
+        json_data = json.loads(response)
+        text_hint += f"[ERROR] {json_data['error']} \n"
         gr.Warning(
-            f"[HTTP ERROR] {response.status_code} - {response.text} \n"
+            f"[ERROR] {json_data['error']} \n"
         )
         return (
             text_hint,
             None,
             None,
         )
+    except:
+        with open(save_path, 'wb') as f:
+            f.write(response)
+    
+    # # Check the response
+    # if response.status_code == 200:
+    #     try:
+    #         json_data = json.loads(response.content)
+    #         text_hint += f"[ERROR] {json_data['error']} \n"
+    #         gr.Warning(
+    #             f"[ERROR] {json_data['error']} \n"
+    #         )
+    #         return (
+    #             text_hint,
+    #             None,
+    #             None,
+    #         )
+    #     except:
+    #         with open(save_path, 'wb') as f:
+    #             f.write(response.content)
+    # else:
+    #     text_hint += f"[HTTP ERROR] {response.status_code} - {response.text} \n"
+    #     gr.Warning(
+    #         f"[HTTP ERROR] {response.status_code} - {response.text} \n"
+    #     )
+    #     return (
+    #         text_hint,
+    #         None,
+    #         None,
+    #     )
+            
     text_hint += f'''Get response successfully \n'''
     return (
         text_hint,