Update app.py

app.py

@@ -14,27 +14,37 @@ import os
 import random
 import tempfile
 import re
+from gradio_client import Client
 
 
 def animate_img(encoded_string):
-    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)
-    while True:
-        r2 = requests.get(f"https://stable-video-diffusion.com/result?hash={hash_}")
-        source_string = r2.text
-        if "Generation has been in progress for" in source_string:
-            time.sleep(10)
-            continue
-        if "Generation has been in progress for" not in source_string:
-            pattern = r'https://storage.stable-video-diffusion.com/([a-f0-9]{32})\.mp4'
-            matches = re.findall(pattern, source_string)
-            sd_video = []
-            for match in matches:
-                sd_video.append(f"https://storage.stable-video-diffusion.com/{match}.mp4")
-            print(sd_video[0])
-            return sd_video[0]
+    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
+        while c < 10:
+            r2 = requests.get(f"https://stable-video-diffusion.com/result?hash={hash_}")
+            source_string = r2.text
+            if "Generation has been in progress for" in source_string:
+                time.sleep(10)
+                c += 1
+                continue
+            if "Generation has been in progress for" not in source_string:
+                pattern = r'https://storage.stable-video-diffusion.com/([a-f0-9]{32})\.mp4'
+                matches = re.findall(pattern, source_string)
+                sd_video = []
+                for match in matches:
+                    sd_video.append(f"https://storage.stable-video-diffusion.com/{match}.mp4")
+                print(sd_video[0])
+                return sd_video[0]
+    except:
+        client1 = Client("https://emmadrex-stable-video-diffusion.hf.space/--replicas/c2gnj/")
+        result1 = client1.predict(encoded_string, api_name="/resize_image")        
+        client = Client("https://emmadrex-stable-video-diffusion.hf.space/--replicas/c2gnj/")
+        result = client.predict(result1, 0, True, 1, 15, api_name="/video")
+        return result[0]['video']
 
 
 def create_video(prompt):
@@ -42,7 +52,7 @@ def create_video(prompt):
     url_sd4 = os.getenv("url_sd4")
     
     try:
-        with closing(create_connection(f"{url_sd3}", timeout=60)) as conn:
+        with closing(create_connection(f"{url_sd3}", timeout=120)) as conn:
             conn.send('{"fn_index":3,"session_hash":""}')
             conn.send(f'{{"data":["{prompt}","[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry",7.5,"(No style)"],"event_data":null,"fn_index":3,"session_hash":""}}')
             while True:
@@ -60,7 +70,7 @@ def create_video(prompt):
                temp_file_path = temp.name
             
     except:
-       with closing(create_connection(f"{url_sd4}", timeout=60)) as conn:
+       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()
@@ -75,29 +85,35 @@ def create_video(prompt):
                temp.write(image_bytes)
                temp_file_path = temp.name
 
-
-    r = requests.post("https://stable-video-diffusion.com/api/upload", files={"file": open(temp_file_path, 'rb')})
-    print(r.text)
-    hash_ = r.json()['hash']
-    time.sleep(10)
-    while True:
-        r2 = requests.get(f"https://stable-video-diffusion.com/result?hash={hash_}")
-        source_string = r2.text
-        if "Generation has been in progress for" in source_string:
-            time.sleep(10)
-            continue
-        if "Generation has been in progress for" not in source_string:
-            pattern = r'https://storage.stable-video-diffusion.com/([a-f0-9]{32})\.mp4'
-            matches = re.findall(pattern, source_string)
-            sd_video = []
-            for match in matches:
-                sd_video.append(f"https://storage.stable-video-diffusion.com/{match}.mp4")
-            print(sd_video[0])
-            return sd_video[0]
+    try:
+        r = requests.post("https://stable-video-diffusion.com/api/upload", files={"file": open(temp_file_path, 'rb')})
+        print(r.text)
+        hash_ = r.json()['hash']
+        time.sleep(10)
+        c = 0
+        while c < 10:
+            r2 = requests.get(f"https://stable-video-diffusion.com/result?hash={hash_}")
+            source_string = r2.text
+            if "Generation has been in progress for" in source_string:
+                time.sleep(10)
+                c += 1
+                continue
+            if "Generation has been in progress for" not in source_string:
+                pattern = r'https://storage.stable-video-diffusion.com/([a-f0-9]{32})\.mp4'
+                matches = re.findall(pattern, source_string)
+                sd_video = []
+                for match in matches:
+                    sd_video.append(f"https://storage.stable-video-diffusion.com/{match}.mp4")
+                print(sd_video[0])
+                return sd_video[0]
+    except:
+        client1 = Client("https://emmadrex-stable-video-diffusion.hf.space/--replicas/c2gnj/")
+        result1 = client1.predict(encoded_string, api_name="/resize_image")        
+        client = Client("https://emmadrex-stable-video-diffusion.hf.space/--replicas/c2gnj/")
+        result = client.predict(result1, 0, True, 1, 15, api_name="/video")
+        return result[0]['video']
         
             
-
-
 def flip_text1(prompt, motion):
     try:
         language = detect(prompt)