Merge pull request #2 from east-and-west-magic/feature-file-service

.gitignore

@@ -1,3 +1,4 @@
 .venv
+.env
 __pycache__
 *.json

app.py

@@ -6,7 +6,7 @@ from zoneinfo import ZoneInfo
 import gradio as gr
 from extract import extract
 import app_util
-from pgsoft.pgconst.const import service_list, functionality_list, game_list
+from pgsoft.pgconst.const import service_list, functionality_list
 from pgsoft.pghost import ais
 from pgsoft.pgdate.date_utils import beijing
 import call_pgai
@@ -15,7 +15,7 @@ from str_util import normalize_text
 #######################
 # proxy version
 #######################
-proxy_version = "1.0.0-2024-01-02-a"  # reconstruct ai calling
+proxy_version = "1.0.0-2024-01-18-a"  # reconstruct ai calling
 
 t = datetime.now()
 t = t.astimezone(ZoneInfo("Asia/Shanghai"))
@@ -27,7 +27,7 @@ print(f"[Seattle]: {t.replace(microsecond=0)}")
 identity = os.environ.get("identity")
 print(f"identity: {identity}")
 if not identity:
-    identity = "pgdemo_dev"
+    identity = "watermelon"
 ai = "stevez-ai-dev"
 if identity in ais:
     ai = ais[identity]
@@ -67,7 +67,7 @@ def run(hf_token, service, game, functionality, nlp_command):
     print(f"<<<<<<<<<<<<<< service starts at {service_start} <<<<<<<<<<<<<<")
     if service in ["download game", "upload game", "list games"]:
         res = app_util.file_service(service, nlp_command, db_token)
-        if not res:
+        if res is None:
             outp = {"status": "Failure"}
         else:
             outp = {"status": "OK", "result": res}

app_util.py

@@ -37,7 +37,7 @@ def call_logger(log_info, caller, hf_token) -> None:
     print(f"calling logger ends at {calling_end}, costs {timecost:.2f}s")
 
 
-dataset_id = "hubei-hunan/games"
+dataset_id = "pgsoft/watermelon"
 local_dir = "game"
 if not os.path.exists(local_dir):
     os.mkdir(local_dir)
@@ -134,7 +134,7 @@ def file_service(service, arg: str, token: str):
             repo_type="dataset",
             token=token,
         )
-        if not games:
+        if games is None:
             return None
         games = {item.split(".")[0]: item for item in games if len(item) >= 32}
         print(f"[{service}] OK")

call_pgai.py

@@ -29,6 +29,8 @@ def post_to_pgai_helper(command: str) -> Any:
         return None
     
 
+from_cache = True
+# @cache
 def call_pgai(service, game, functionality, nlp_command, url, hf_token):
     calling_start = beijing()
     print(f"calling ai starts at {calling_start}")