Update app.py

app.py

@@ -18,9 +18,8 @@ from gradio_client import Client
 
 
 def animate_img(encoded_string):
-    try:    
+    try:   
         r = requests.post("https://stable-video-diffusion.com/api/upload", files={"file": open(encoded_string, 'rb')})
-        print(r.text)
         hash_ = r.json()['hash']
         time.sleep(10)
         c = 0
@@ -37,17 +36,21 @@ def animate_img(encoded_string):
                 sd_video = []
                 for match in matches:
                     sd_video.append(f"https://storage.stable-video-diffusion.com/{match}.mp4")
-                print(sd_video[0])
                 if len(sd_video) != 0:
+                    print("s_1")
                     return sd_video[0]
                 else:
                     _ = 1/0
+        print("f_1")
     except:
+        print("2")
         client1 = Client("https://emmadrex-stable-video-diffusion.hf.space")
         result1 = client1.predict(encoded_string, api_name="/resize_image")        
         client = Client("https://emmadrex-stable-video-diffusion.hf.space")
         result = client.predict(result1, 0, True, 1, 15, api_name="/video")
-        return result[0]['video']
+        res = result[0]['video']
+        print("s_2")
+        return res
 
 
 def create_video(prompt):
@@ -69,14 +72,17 @@ def create_video(prompt):
                     break
             photo = json.loads(conn.recv())['output']['data'][0][0]
             base64_string = photo.replace('data:image/jpeg;base64,', '').replace('data:image/png;base64,', '')
-    
+        
             image_bytes = base64.b64decode(base64_string)
             with tempfile.NamedTemporaryFile(delete=False) as temp:
-               temp.write(image_bytes)
-               temp_file_path = temp.name
+                temp.write(image_bytes)
+                temp_file_path = temp.name
+                print("cs_1")
+
             
     except:
-       with closing(create_connection(f"{url_sd4}", timeout=120)) as conn:
+        print("c_2")
+        with closing(create_connection(f"{url_sd4}", timeout=120)) as conn:
             conn.send('{"fn_index":0,"session_hash":""}')
             conn.send(f'{{"data":["{prompt}","[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry","dreamshaperXL10_alpha2.safetensors [c8afe2ef]",30,"DPM++ 2M Karras",7,1024,1024,-1],"event_data":null,"fn_index":0,"session_hash":""}}')
             conn.recv()
@@ -85,11 +91,12 @@ def create_video(prompt):
             conn.recv()
             photo = json.loads(conn.recv())['output']['data'][0]
             base64_string = photo.replace('data:image/jpeg;base64,', '').replace('data:image/png;base64,', '')
-
+        
             image_bytes = base64.b64decode(base64_string)
             with tempfile.NamedTemporaryFile(delete=False) as temp:
                temp.write(image_bytes)
                temp_file_path = temp.name
+                print("cs_2")
 
     try:
         r = requests.post("https://stable-video-diffusion.com/api/upload", files={"file": open(temp_file_path, 'rb')})