mirrors/gpt4free
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-12-06 02:30:41 -08:00
Code Issues Projects Releases Packages Wiki Activity

Update handle media uploads

Browse source
This commit is contained in:
hlohaus 2025-03-30 00:04:13 +01:00
parent 714d02006e
commit 6c8387f045
7 changed files with 37 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

8
g4f/gui/server/api.py
View file

@ -182,15 +182,15 @@ class Api:
elif isinstance(chunk, PreviewResponse):
yield self._format_json("preview", chunk.to_string())
elif isinstance(chunk, ImagePreview):
yield self._format_json("preview", chunk.to_string(), images=chunk.images, alt=chunk.alt)
elif isinstance(chunk, (ImageResponse, VideoResponse)):
yield self._format_json("preview", chunk.to_string(), urls=chunk.urls, alt=chunk.alt)
elif isinstance(chunk, MediaResponse):
media = chunk
if download_media or chunk.get("cookies"):
chunk.alt = format_image_prompt(kwargs.get("messages"), chunk.alt)
tags = [model, kwargs.get("aspect_ratio")]
tags = [model, kwargs.get("aspect_ratio"), kwargs.get("resolution"), kwargs.get("width"), kwargs.get("height")]
media = asyncio.run(copy_media(chunk.get_list(), chunk.get("cookies"), chunk.get("headers"), proxy=proxy, alt=chunk.alt, tags=tags))
media = ImageResponse(media, chunk.alt) if isinstance(chunk, ImageResponse) else VideoResponse(media, chunk.alt)
yield self._format_json("content", str(media), images=chunk.get_list(), alt=chunk.alt)
yield self._format_json("content", str(media), urls=chunk.urls, alt=chunk.alt)
elif isinstance(chunk, SynthesizeData):
yield self._format_json("synthesize", chunk.get_dict())
elif isinstance(chunk, TitleGeneration):