highperfocused/nostrdvm
1
0
Fork 0
mirror of https://github.com/believethehype/nostrdvm.git synced 2025-06-24 11:01:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update utils.py

Browse Source
This commit is contained in:
Believethehype 2024-01-15 10:26:13 +01:00
parent 068130cfa4
commit 692eb6f38b
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
nostr_dvm/backends/nova_server/utils.py
View File

@ -6,6 +6,7 @@ import zipfile
import pandas as pd import pandas as pd
import requests import requests
import PIL.Image as Image import PIL.Image as Image
from moviepy.video.io.VideoFileClip import VideoFileClip
from nostr_dvm.utils.output_utils import upload_media_to_hoster from nostr_dvm.utils.output_utils import upload_media_to_hoster
@ -87,6 +88,15 @@ def check_server_status(jobID, address) -> str | pd.DataFrame:
result = upload_media_to_hoster("./outputs/image.jpg") result = upload_media_to_hoster("./outputs/image.jpg")
os.remove("./outputs/image.jpg") os.remove("./outputs/image.jpg")
return result return result
elif content_type == 'video/mp4':
with open('./outputs/video.mp4', 'wb') as f:
f.write(response.content)
clip = VideoFileClip("./outputs/video.mp4")
clip.write_videofile("./outputs/video2.mp4")
result = upload_media_to_hoster("./outputs/video2.mp4")
os.remove("./outputs/video.mp4")
os.remove("./outputs/video2.mp4")
return result
elif content_type == 'text/plain; charset=utf-8': elif content_type == 'text/plain; charset=utf-8':
return response.content.decode('utf-8') return response.content.decode('utf-8')
elif content_type == "application/x-zip-compressed": elif content_type == "application/x-zip-compressed":