Update app.py

app.py

@@ -37,7 +37,68 @@ def animate_img(encoded_string):
             return sd_video[0]
 
 
-def flip_text(prompt, motion):
+def create_video(prompt):
+    url_sd3 = os.getenv("url_sd3")
+    url_sd4 = os.getenv("url_sd4")
+    
+    try:
+        with closing(create_connection(f"{url_sd3}", timeout=60)) 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:
+                status = json.loads(conn.recv())['msg']
+                if status == 'estimation':
+                    continue
+                if status == 'process_starts':
+                    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
+            
+    except:
+       with closing(create_connection(f"{url_sd4}", timeout=60)) 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()
+            conn.recv()
+            conn.recv()
+            conn.recv()
+            photo = json.loads(conn.recv())['output']['data'][0]
+            photo = 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
+
+
+    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]
+        
+            
+
+
+def flip_text1(prompt, motion):
     try:
         language = detect(prompt)
         if language == 'ru':
@@ -92,6 +153,7 @@ def flip_text(prompt, motion):
 
 
 
+
 def flip_text2(encoded_string, prompt, motion):
 
     url_video_g = os.getenv("url_video_g")
@@ -152,6 +214,7 @@ def flip_text2(encoded_string, prompt, motion):
                 
 
 
+
 css = """
 #generate {
     width: 100%;