From 6f2b6cccbda6b3cd7ec5e0d5ba8ebf23d7c53419 Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <hlohaus@users.noreply.github.com>
Date: Thu, 21 Nov 2024 07:14:36 +0100
Subject: Add upload cookie files

---
 g4f/gui/client/index.html           |  2 +-
 g4f/gui/client/static/js/chat.v1.js | 33 +++++++++++++++++++++++----------
 g4f/gui/server/backend.py           | 27 ++++++++++++++++-----------
 3 files changed, 40 insertions(+), 22 deletions(-)

(limited to 'g4f/gui')

diff --git a/g4f/gui/client/index.html b/g4f/gui/client/index.html
index 6c2ad8b6..116509d8 100644
--- a/g4f/gui/client/index.html
+++ b/g4f/gui/client/index.html
@@ -224,7 +224,7 @@
                         <i class="fa-solid fa-camera"></i>
                     </label>
                     <label class="file-label" for="file">
-                        <input type="file" id="file" name="file" accept="text/plain, text/html, text/xml, application/json, text/javascript, .sh, .py, .php, .css, .yaml, .sql, .log, .csv, .twig, .md" required/>
+                        <input type="file" id="file" name="file" accept="text/plain, text/html, text/xml, application/json, text/javascript, .har, .sh, .py, .php, .css, .yaml, .sql, .log, .csv, .twig, .md" required/>
                         <i class="fa-solid fa-paperclip"></i>
                     </label>
                     <label class="micro-label" for="micro">
diff --git a/g4f/gui/client/static/js/chat.v1.js b/g4f/gui/client/static/js/chat.v1.js
index 73c0de0f..8127fb9d 100644
--- a/g4f/gui/client/static/js/chat.v1.js
+++ b/g4f/gui/client/static/js/chat.v1.js
@@ -1338,17 +1338,25 @@ fileInput.addEventListener('click', async (event) => {
     delete fileInput.dataset.text;
 });
 
+async function upload_cookies() {
+    const file = fileInput.files[0];
+    const formData = new FormData();
+    formData.append('file', file);
+    response = await fetch("/backend-api/v2/upload_cookies", {
+        method: 'POST',
+        body: formData,
+    });
+    if (response.status == 200) {
+        inputCount.innerText = `${file.name} was uploaded successfully`;
+    }
+    fileInput.value = "";
+}
+
 fileInput.addEventListener('change', async (event) => {
     if (fileInput.files.length) {
-        type = fileInput.files[0].type;
-        if (type && type.indexOf('/')) {
-            type = type.split('/').pop().replace('x-', '')
-            type = type.replace('plain', 'plaintext')
-                       .replace('shellscript', 'sh')
-                       .replace('svg+xml', 'svg')
-                       .replace('vnd.trolltech.linguist', 'ts')
-        } else {
-            type = fileInput.files[0].name.split('.').pop()
+        type = fileInput.files[0].name.split('.').pop()
+        if (type == "har") {
+            return await upload_cookies();
         }
         fileInput.dataset.type = type
         const reader = new FileReader();
@@ -1357,14 +1365,19 @@ fileInput.addEventListener('change', async (event) => {
             if (type == "json") {
                 const data = JSON.parse(fileInput.dataset.text);
                 if ("g4f" in data.options) {
+                    let count = 0;
                     Object.keys(data).forEach(key => {
                         if (key != "options" && !localStorage.getItem(key)) {
                             appStorage.setItem(key, JSON.stringify(data[key]));
-                        } 
+                            count += 1;
+                        }
                     });
                     delete fileInput.dataset.text;
                     await load_conversations();
                     fileInput.value = "";
+                    inputCount.innerText = `${count} Conversations were imported successfully`;
+                } else {
+                    await upload_cookies();
                 }
             }
         });
diff --git a/g4f/gui/server/backend.py b/g4f/gui/server/backend.py
index 87da49e1..fe020bb8 100644
--- a/g4f/gui/server/backend.py
+++ b/g4f/gui/server/backend.py
@@ -1,12 +1,15 @@
 import json
 import asyncio
 import flask
+import os
 from flask import request, Flask
 from typing import AsyncGenerator, Generator
+from werkzeug.utils import secure_filename
 
 from g4f.image import is_allowed_extension, to_image
 from g4f.client.service import convert_to_provider
 from g4f.errors import ProviderNotFoundError
+from g4f.cookies import get_cookies_dir
 from .api import Api
 
 def safe_iter_generator(generator: Generator) -> Generator:
@@ -79,8 +82,8 @@ class Backend_Api(Api):
                 'function': self.handle_synthesize,
                 'methods': ['GET']
             },
-            '/backend-api/v2/error': {
-                'function': self.handle_error,
+            '/backend-api/v2/upload_cookies': {
+                'function': self.upload_cookies,
                 'methods': ['POST']
             },
             '/images/<path:name>': {
@@ -89,15 +92,17 @@ class Backend_Api(Api):
             }
         }
 
-    def handle_error(self):
-        """
-        Initialize the backend API with the given Flask application.
-
-        Args:
-            app (Flask): Flask application instance to attach routes to.
-        """
-        print(request.json)
-        return 'ok', 200
+    def upload_cookies(self):
+        file = None
+        if "file" in request.files:
+            file = request.files['file']
+            if file.filename == '':
+                return 'No selected file', 400
+        if file and file.filename.endswith(".json") or file.filename.endswith(".har"):
+            filename = secure_filename(file.filename)
+            file.save(os.path.join(get_cookies_dir(), filename))
+            return "File saved", 200
+        return 'Not supported file', 400
 
     def handle_conversation(self):
         """
-- 
cgit v1.2.3