From fb58feae87d62056af08ac0ea481f39f389b04ef Mon Sep 17 00:00:00 2001
From: hlohaus <983577+hlohaus@users.noreply.github.com>
Date: Wed, 5 Feb 2025 19:30:47 +0100
Subject: Check for exception in generate of G4F

---
 g4f/Provider/hf_space/G4F.py  | 13 +++++--------
 g4f/gui/server/backend_api.py |  2 ++
 2 files changed, 7 insertions(+), 8 deletions(-)

(limited to 'g4f')

diff --git a/g4f/Provider/hf_space/G4F.py b/g4f/Provider/hf_space/G4F.py
index bac40182..2c16c032 100644
--- a/g4f/Provider/hf_space/G4F.py
+++ b/g4f/Provider/hf_space/G4F.py
@@ -83,14 +83,11 @@ class G4F(Janus_Pro_7B):
                 "x-zerogpu-uuid": zerogpu_uuid,
             }
             async def generate():
-                try:
-                    async with session.post(cls.url_flux, json=payload, proxy=proxy, headers=headers) as response:
-                        await raise_for_status(response)
-                        response_data = await response.json()
-                        image_url = response_data["data"][0]['url']
-                        return ImageResponse(images=[image_url], alt=prompt)
-                except Exception as e:
-                    return Reasoning(status=f"Error: {e.__class__.__name__}: {e}")
+                async with session.post(cls.url_flux, json=payload, proxy=proxy, headers=headers) as response:
+                    await raise_for_status(response)
+                    response_data = await response.json()
+                    image_url = response_data["data"][0]['url']
+                    return ImageResponse(images=[image_url], alt=prompt)
             background_tasks = set()
             started = time.time()
             task = asyncio.create_task(generate())
diff --git a/g4f/gui/server/backend_api.py b/g4f/gui/server/backend_api.py
index 4b499408..b293a839 100644
--- a/g4f/gui/server/backend_api.py
+++ b/g4f/gui/server/backend_api.py
@@ -147,6 +147,8 @@ class Backend_Api(Api):
                 limiter_check()
             if "images" in json_data:
                 kwargs["images"] = json_data["images"]
+            if "zerogpu_token" in json_data:
+                kwargs["zerogpu_token"] = json_data["zerogpu_token"]
             kwargs = self._prepare_conversation_kwargs(json_data, kwargs)
             return self.app.response_class(
                 self._create_response_stream(
-- 
cgit v1.2.3