Update app.py

app.py

@@ -21,9 +21,9 @@ token = os.environ.get("HF_TOKEN")
 library_username = os.environ.get("OLLAMA_USERNAME").lower()
 
 def regenerate_pubkey(oauth_token: gr.OAuthToken | None):
+    if oauth_token.token is None:
+        raise ValueError("You must be logged in to use Ollamafy")
     try:
-        if oauth_token.token is None:
-            raise ValueError("You must be logged in to use Ollamafy")
         hash_oauth = hashlib.sha256(unicode_string.encode(oauth_token)).hexdigest()
         generate_ollama_host_file = f"echo $(ss -natp | grep (cat ollama.pid) | awk '{{print $4}}') > ollama.host"
         generate_ollama_pid_file = f"echo $! > ollama.pid"
@@ -59,6 +59,7 @@ def regenerate_pubkey(oauth_token: gr.OAuthToken | None):
 
 def ollamafy_model(login, account, model_id, ollama_library_username , ollama_q_method, latest, maintainer, oauth_token: gr.OAuthToken | None):
     ollama_library_username: library_username | None 
+    
     if oauth_token.token is None:
         raise ValueError("You must be logged in to use Ollamafy")
     hash_oauth = hashlib.sha256(unicode_string.encode(oauth_token)).hexdigest()
@@ -66,6 +67,8 @@ def ollamafy_model(login, account, model_id, ollama_library_username , ollama_q_
     # username = whoami(oauth_token.token)["name"]
     model_name = model_id.split('/')[-1]
     fp16 = f"{model_name}-fp16.gguf"
+    ollama_stop = f"kill -9 'cat ${HOME}/{hash_oauth}/ollama.pid'"
+    delete_home = f"rm -Rf {HOME}/{hash_oauth}/.ollama"
 
     try:
         api = HfApi(token=oauth_token.token)   
@@ -183,6 +186,18 @@ def ollamafy_model(login, account, model_id, ollama_library_username , ollama_q_
     finally:
         shutil.rmtree(model_name, ignore_errors=True)
         print("Folder cleaned up successfully!")
+        
+        result = subprocess.run(ollama_stop, shell=True, capture_output=True)
+        print(result)
+        if result.returncode != 0:
+            raise Exception(f"Error stoppping Ollama {result.stderr}")
+        print("Ollama stopped successfully!")
+        
+        result = subprocess.run(delete_home, shell=True, capture_output=True)
+        print(result)
+        if result.returncode != 0:
+            raise Exception(f"Error removing Ollama HOME folder {result.stderr}")
+        print("Ollama HOME fodler removed successfully!")
 
         
 css="""/* Custom CSS to allow scrolling */