sync ms

app.py

@@ -1,6 +1,16 @@
+import os
+import shutil
+import requests
 import gradio as gr
-from utils import download_file, EN_US, API_QR, TMP_DIR
 
+EN_US = os.getenv("LANG") != "zh_CN.UTF-8"
+API_QR = os.getenv("api_qr")
+if not API_QR:
+    print("请检查环境变量")
+    exit()
+
+
+TMP_DIR = "./__pycache__"
 ZH2EN = {
     "二维码输出尺寸": "Image size",
     "输入文本": "Input text",
@@ -15,6 +25,24 @@ def _L(zh_txt: str):
     return ZH2EN[zh_txt] if EN_US else zh_txt
 
 
+def clean_dir(dir_path: str):
+    if os.path.exists(dir_path):
+        shutil.rmtree(dir_path)
+
+    os.makedirs(dir_path)
+
+
+def download_file(url, local_filename):
+    clean_dir(os.path.dirname(local_filename))
+    response = requests.get(url, stream=True)
+    response.raise_for_status()
+    with open(local_filename, "wb") as f:
+        for chunk in response.iter_content(chunk_size=8192):
+            f.write(chunk)
+
+    return local_filename
+
+
 def infer(img_size: int, input_txt: str):
     status = "Success"
     img = None
@@ -23,7 +51,7 @@ def infer(img_size: int, input_txt: str):
             raise ValueError("Please input valid text!")
 
         img = download_file(
-            f"{API_QR}/?size={img_size}x{img_size}&data={input_txt}",
+            f"{API_QR}/?size={img_size}x{img_size}&data={input_txt}" if EN_US else f"{API_QR}/?text={input_txt}&size={img_size}",
             f"{TMP_DIR}/qrcode.jpg",
         )
 

utils.py

@@ -1,30 +0,0 @@
-import os
-import shutil
-import requests
-
-EN_US = os.getenv("LANG") != "zh_CN.UTF-8"
-API_QR = os.getenv("api_qr")
-if not API_QR:
-    print("请检查环境变量")
-    exit()
-
-
-TMP_DIR = "./__pycache__"
-
-
-def clean_dir(dir_path: str):
-    if os.path.exists(dir_path):
-        shutil.rmtree(dir_path)
-
-    os.makedirs(dir_path)
-
-
-def download_file(url, local_filename):
-    clean_dir(os.path.dirname(local_filename))
-    response = requests.get(url, stream=True)
-    response.raise_for_status()
-    with open(local_filename, "wb") as f:
-        for chunk in response.iter_content(chunk_size=8192):
-            f.write(chunk)
-
-    return local_filename