From ba60296677cdddb622fad71f957ecf1e442a0fca Mon Sep 17 00:00:00 2001
From: hlohaus <983577+hlohaus@users.noreply.github.com>
Date: Sat, 22 Feb 2025 13:19:52 +0100
Subject: Fix unittest, update model lists

---
 g4f/gui/server/api.py         | 4 ++++
 g4f/gui/server/backend_api.py | 8 ++++----
 2 files changed, 8 insertions(+), 4 deletions(-)

(limited to 'g4f/gui/server')

diff --git a/g4f/gui/server/api.py b/g4f/gui/server/api.py
index f8e81700..23ad2691 100644
--- a/g4f/gui/server/api.py
+++ b/g4f/gui/server/api.py
@@ -19,6 +19,8 @@ from ... import version, models
 from ... import ChatCompletion, get_model_and_provider
 from ... import debug
 
+logger = logging.getLogger(__name__)
+
 conversations: dict[dict[str, BaseConversation]] = {}
 
 class Api:
@@ -184,6 +186,7 @@ class Api:
                         else:
                             yield self._format_json("conversation_id", conversation_id)
                 elif isinstance(chunk, Exception):
+                    logger.exception(chunk)
                     debug.error(chunk)
                     yield self._format_json('message', get_error_message(chunk), error=type(chunk).__name__)
                 elif isinstance(chunk, PreviewResponse):
@@ -219,6 +222,7 @@ class Api:
                     yield self._format_json("content", str(chunk))
                 yield from self._yield_logs()
         except Exception as e:
+            logger.exception(e)
             debug.error(e)
             yield from self._yield_logs()
             yield self._format_json('error', type(e).__name__, message=get_error_message(e))
diff --git a/g4f/gui/server/backend_api.py b/g4f/gui/server/backend_api.py
index 287e728b..28c74251 100644
--- a/g4f/gui/server/backend_api.py
+++ b/g4f/gui/server/backend_api.py
@@ -124,9 +124,9 @@ class Backend_Api(Api):
                 Response: A Flask response object for streaming.
             """
             kwargs = {}
-            if "files[]" in request.files:
+            if "files" in request.files:
                 images = []
-                for file in request.files.getlist('files[]'):
+                for file in request.files.getlist('files'):
                     if file.filename != '' and is_allowed_extension(file.filename):
                         images.append((to_image(file.stream, file.filename.endswith('.svg')), file.filename))
                 kwargs['images'] = images
@@ -135,7 +135,7 @@ class Backend_Api(Api):
             else:
                 json_data = request.json
 
-            if app.demo and json_data.get("provider") not in ["Custom", "Feature", "HuggingFace", "HuggingSpace", "HuggingChat", "G4F"]:
+            if app.demo and json_data.get("provider") not in ["Custom", "Feature", "HuggingFace", "HuggingSpace", "HuggingChat", "G4F", "PollinationsAI"]:
                 model = json_data.get("model")
                 if model != "default" and model in models.demo_models:
                     json_data["provider"] = random.choice(models.demo_models[model][1])
@@ -329,7 +329,7 @@ class Backend_Api(Api):
             bucket_dir = get_bucket_dir(bucket_id)
             os.makedirs(bucket_dir, exist_ok=True)
             filenames = []
-            for file in request.files.getlist('files[]'):
+            for file in request.files.getlist('files'):
                 try:
                     filename = secure_filename(file.filename)
                     if supports_filename(filename):
-- 
cgit v1.2.3