Add Error Message for not login, Change the readme format, Change the repo name

app.py

@@ -11,7 +11,6 @@ from gradio_huggingfacehub_search import HuggingfaceHubSearch
 from apscheduler.schedulers.background import BackgroundScheduler
 from datetime import datetime
 import numpy as np
-import shutil
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
 os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
@@ -195,7 +194,11 @@ def process_model(model_id, q_method, use_imatrix, imatrix_q_method, private_rep
                   export_to_org, repo_owner, org_token, oauth_token: gr.OAuthToken | None):
     if oauth_token is None or oauth_token.token is None:
         raise gr.Error("You must be logged in to use GGUF-my-repo")
-
+    try:
+        whoami(oauth_token.token)
+    except Exception as e:
+        raise gr.Error("You must be logged in to use GGUF-my-repo")
+    
     user_info = whoami(oauth_token.token)
     username = user_info["name"]
     user_orgs = user_info.get("orgs", [])
@@ -205,7 +208,7 @@ def process_model(model_id, q_method, use_imatrix, imatrix_q_method, private_rep
 
     current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     logger.info(f"Time {current_time}, Username {username}, Model_ID, {model_id}, q_method {','.join(q_method)}")
-    
+
     repo_namespace = get_repo_namespace(repo_owner, username, user_orgs)
     model_name = model_id.split('/')[-1]
     try:
@@ -226,8 +229,7 @@ def process_model(model_id, q_method, use_imatrix, imatrix_q_method, private_rep
           fp16 = str(Path(outdir)/f"{model_name}.fp16.gguf")
 
           with tempfile.TemporaryDirectory(dir=downloads_dir) as tmpdir:
-              print(datetime.now().strftime("%Y-%m-%d %H:%M:%S") + " Start download")
-              logger.info(datetime.now().strftime("%Y-%m-%d %H:%M:%S") + " Start download")
+              print("Downloading")
               local_dir = Path(tmpdir)/model_name
               api.snapshot_download(repo_id=model_id, local_dir=local_dir, local_dir_use_symlinks=False, allow_patterns=dl_pattern)
               
@@ -236,16 +238,12 @@ def process_model(model_id, q_method, use_imatrix, imatrix_q_method, private_rep
               if os.path.exists(adapter_config_dir) and not os.path.exists(config_dir):
                   raise Exception("adapter_config.json is present. If converting LoRA, use GGUF-my-lora.")
 
-              print(datetime.now().strftime("%Y-%m-%d %H:%M:%S") + " Download successfully")
-              logger.info(datetime.now().strftime("%Y-%m-%d %H:%M:%S") + " Download successfully")
+              print("Download successfully")
               result = subprocess.run(["python", CONVERSION_SCRIPT, local_dir, "--outtype", "f16", "--outfile", fp16], shell=False, capture_output=True)
-              print(datetime.now().strftime("%Y-%m-%d %H:%M:%S") + " Converted to f16")
-              logger.info(datetime.now().strftime("%Y-%m-%d %H:%M:%S") + " Converted to f16")
+              print("Converted to f16")
               if result.returncode != 0:
                   raise Exception(f"Error converting to fp16: {result.stderr.decode()}")
 
-          shutil.rmtree(downloads_dir)
-
           imatrix_path = Path(outdir)/"imatrix.dat"
           if use_imatrix:
               train_data_path = train_data_file.name if train_data_file else "llama.cpp/groups_merged.txt"
@@ -258,8 +256,7 @@ def process_model(model_id, q_method, use_imatrix, imatrix_q_method, private_rep
 
           gguf_files = []
           for method in quant_methods:
-              print(datetime.now().strftime("%Y-%m-%d %H:%M:%S") + " Begin quantize")
-              logger.info(datetime.now().strftime("%Y-%m-%d %H:%M:%S") + " Begin quantize")
+              print("Begin quantize")
               name = f"{model_name.lower()}-{method.lower()}-{suffix}.gguf" if suffix else f"{model_name.lower()}-{method.lower()}.gguf"
               path = str(Path(outdir)/name)
               quant_cmd = ["./llama.cpp/llama-quantize", "--imatrix", imatrix_path, fp16, path, method] if use_imatrix else ["./llama.cpp/llama-quantize", fp16, path, method]
@@ -269,10 +266,9 @@ def process_model(model_id, q_method, use_imatrix, imatrix_q_method, private_rep
               size = os.path.getsize(path)/1024/1024/1024
               gguf_files.append((name, path, size, method))
 
-          print(datetime.now().strftime("%Y-%m-%d %H:%M:%S") + " Quantize successfully!")
-          logger.info(datetime.now().strftime("%Y-%m-%d %H:%M:%S") + " Quantize successfully!")
+          print("Quantize successfully!")
           suffix_for_repo = f"{imatrix_q_method}-imat" if use_imatrix else "-".join(quant_methods)
-          repo_id = f"{repo_namespace}/{model_name}-{suffix_for_repo}-GGUF"
+          repo_id = f"{repo_namespace}/{model_name}-GGUF"
           new_repo_url = api.create_repo(repo_id=repo_id, exist_ok=True, private=private_repo)
 
           try:
@@ -420,6 +416,7 @@ with gr.Blocks(css=".gradio-container {overflow-y: auto;}") as demo:
     iface.render()
 
 
+
 def restart_space():
     HfApi().restart_space(repo_id="Antigma/quantize-my-repo", token=HF_TOKEN, factory_reboot=True)