Update utils.py

utils.py

@@ -9,6 +9,8 @@ import hashlib
 import csv
 import requests
 import re
+import sys
+import subprocess
 
 import gradio as gr
 from pypinyin import lazy_pinyin
@@ -387,3 +389,42 @@ def find_n(lst, max_num):
             return n - i -1
         total = total - lst[i]
     return 1
+
+
+def run(command, desc=None, errdesc=None, custom_env=None, live=False):
+     if desc is not None:
+         print(desc)
+     if live:
+         result = subprocess.run(command, shell=True, env=os.environ if custom_env is None else custom_env)
+         if result.returncode != 0:
+             raise RuntimeError(f"""{errdesc or 'Error running command'}.
+ Command: {command}
+ Error code: {result.returncode}""")
+
+         return ""
+     result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True, env=os.environ if custom_env is None else custom_env)
+     if result.returncode != 0:
+         message = f"""{errdesc or 'Error running command'}.
+ Command: {command}
+ Error code: {result.returncode}
+ stdout: {result.stdout.decode(encoding="utf8", errors="ignore") if len(result.stdout)>0 else '<empty>'}
+ stderr: {result.stderr.decode(encoding="utf8", errors="ignore") if len(result.stderr)>0 else '<empty>'}
+ """
+         raise RuntimeError(message)
+     return result.stdout.decode(encoding="utf8", errors="ignore")
+
+ def versions_html():
+     git = os.environ.get('GIT', "git")
+     python_version = ".".join([str(x) for x in sys.version_info[0:3]])
+     try:
+         commit_hash = run(f"{git} rev-parse HEAD").strip()
+     except Exception:
+         commit_hash = "<none>"
+     short_commit = commit_hash[0:7]
+     return f"""
+ Python: <span title="{sys.version}">{python_version}</span>
+  • 
+ Gradio: {gr.__version__}
+  • 
+ commit: <a style="text-decoration:none" href="https://github.com/GaiZhenbiao/ChuanhuChatGPT/commit/{short_commit}">{short_commit}</a>
+"""