Update time.tzset()

app.py

@@ -1,7 +1,9 @@
 """See https://huggingface.co/spaces/Gradio-Blocks/Story-to-video/blob/main/app.py."""
 import base64
 import io
+import os
 import re
+import time
 from random import choice
 
 import gradio as gr
@@ -9,8 +11,9 @@ import translators as ts
 from fastlid import fastlid
 from logzero import logger
 from PIL import Image  # opencv-python
+from random import choices
 from tenacity import retry
-from tenacity.stop import stop_after_attempt, stop_after_delay
+from tenacity.stop import stop_after_attempt  # , stop_after_delay
 
 # from PIL import Image
 # from transformers import AutoTokenizer, AutoModelForSeq2SeqLM,pipeline
@@ -18,11 +21,19 @@ from tenacity.stop import stop_after_attempt, stop_after_delay
 # import torch
 
 image_gen = gr.Interface.load("spaces/multimodalart/latentdiffusion")
+# image_gen = gr.Interface.load("huggingface/multimodalart/latentdiffusion")
+
+os.environ["TZ"] = "Asia/Shanghai"
+try:
+    time.tzset()
+except Exception:
+    ...  # Windows
+
 
 examples_ = [
-    "蓝色的夜，森林中好多萤火虫",
     "黑云压城城欲摧 ，甲光向日金鳞开。",
     "黄金在河里流淌，宝石遍地，空中铺满巨大的彩虹。",
+    "蓝色的夜，森林中好多萤火虫",
     "季姬寂，集鸡，鸡即棘鸡。棘鸡饥叽，季姬及箕稷济鸡。",
     "an apple",
     "a cat",
@@ -31,10 +42,15 @@ examples_ = [
 ]
 
 
-@retry(stop=(stop_after_delay(10) | stop_after_attempt(5)))
+# @retry(stop=(stop_after_delay(10) | stop_after_attempt(5)))
+@retry(stop=stop_after_attempt(5))
 def tr_(text: str) -> str:
-    """Wrap [ts.deepl, ts.sogou, ts.baidu, ts.google] with tenacity."""
-    for tr in [ts.deepl, ts.sogou, ts.baidu, ts.google]:
+    """Wrap [ts.deepl, ts.baidu, ts.google] with tenacity.
+
+    not working: sogou;  ?tencent
+    """
+    cand = [ts.baidu, ts.youdao, ts.google]
+    for tr in [ts.deepl] + choices(cand, k=len(cand)):
         try:
             res = tr(text)
             logger.info(" api used: %s", tr.__name__)
@@ -44,6 +60,7 @@ def tr_(text: str) -> str:
             continue
     else:
         res = "Something is probably wong, ping dev to fix it if you like."
+
     return res
 
 
@@ -53,20 +70,20 @@ def generate_images(phrase: str, steps: int = 125):
 
     generated_text = phrase
     detected = "en"
-    api_used = ""
+    extra_info = ""
     try:
         detected = fastlid(phrase)[0]
     except Exception as exc:
         logger.error(exc)
 
-    # safe guard short Chinese phrases
+    # safeguard short Chinese phrases
     if len(phrase) < 10 and re.search(r"[一-龟]+", phrase):
         detected = "zh"
 
     if detected not in ["en"]:
         try:
             generated_text = tr_(phrase)
-            api_used = f"({tr_.api_used})"
+            extra_info = f"({tr_.api_used}: {generated_text})"
         except Exception as exc:
             logger.error(exc)
             return None, f"{phrase:}, errors: {str(exc)}"
@@ -90,7 +107,7 @@ def generate_images(phrase: str, steps: int = 125):
     # generated_images = []
 
     img = None
-    err_msg = f"{phrase} {api_used}"
+    err_msg = f"{phrase} {extra_info}"
     for image in image_bytes[1]:
         image_str = image[0]
         try:

flagged/log.csv

@@ -0,0 +1,2 @@
+'phrase','steps','component 2','phrase','flag','username','timestamp'
+'黑云压城城欲摧 ，甲光向日金鳞开。',125,'component 2/0.png','黑云压城城欲摧 ，甲光向日金鳞开。 (sogou_api)','','','2022-07-20 06:24:49.026071'