added: ollama_host_file , ollama_pid_file , ollama_pubkey ollama_start , ollama_stop , delete_home

ollama_host_file = f"echo $(ss -natp | grep (cat ollama.pid) | awk '{{print $4}}') > ollama.host"
ollama_pid_file = f"echo $! > ollama.pid"
ollama_pubkey = open(f"{HOME}/.ollama/{hash_oauth}/id_ed25519.pub", "r")
ollama_start = f"HOME={HOME}/{hash_oauth} ollama serve & {ollama_pid_file} & sleep 5 & {ollama_host_file}"
ollama_stop = f"kill -9 'cat ${HOME}/{hash_oauth}/ollama.pid'"
delete_home = f"rm -Rf {HOME}/{hash_oauth}/.ollama"

app.py

@@ -20,8 +20,7 @@ HOME = os.environ.get("HOME")
 # token = os.environ.get("HF_TOKEN")
 library_username = os.environ.get("OLLAMA_USERNAME").lower()
 
-ollama_start = f"ollama serve & echo $! > ${HOME}/ollama.pid & sleep 5"
-ollama_stop = f"kill -9 'cat ${HOME}/ollama.pid'"
+
 
 
 def regenerate_pubkey(login, oauth_token: gr.OAuthToken | None):
@@ -29,24 +28,26 @@ def regenerate_pubkey(login, oauth_token: gr.OAuthToken | 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()
-    ollama_pubkey = open(f"{HOME}/.ollama/{hash_oauth}/id_ed25519.pub", "r")
-    delete_ihome = f"rm -Rf {HOME}/{hash_oauth}/.ollama"
-    ollama_istart = f"OLLAMA_HOST=127.0.0.1:0 OLLAMA_MODELS={HOME}/.ollama/models HOME={HOME}/{hash_oauth} ollama serve & echo $! > ${HOME}/{hash_oauth}/ollama.pid & sleep 5"
-    ollama_istop = f"kill -9 'cat ${HOME}/{hash_oauth}/ollama.pid'"
+    ollama_host_file = f"echo $(ss -natp | grep (cat ollama.pid) | awk '{{print $4}}') > ollama.host"
+    ollama_pid_file = f"echo $! > ollama.pid"
+    ollama_pubkey = open(f"{HOME}/.ollama/{hash_oauth}/id_ed25519.pub", "r")    
+    ollama_start = f"HOME={HOME}/{hash_oauth} ollama serve & {ollama_pid_file} & sleep 5 & {ollama_host_file}"    
+    ollama_stop = f"kill -9 'cat ${HOME}/{hash_oauth}/ollama.pid'"
+    delete_home = f"rm -Rf {HOME}/{hash_oauth}/.ollama"
         
-    result = subprocess.run(ollama_istop, shell=True, capture_output=True)
+    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_ihome, shell=True, capture_output=True)
+    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!")
 
-    result = subprocess.run(ollama_istart, shell=True, capture_output=True)
+    result = subprocess.run(ollama_start, shell=True, capture_output=True)
     print(result)
     if result.returncode != 0:
         raise Exception(f"Error starting Ollama {result.stderr}")
@@ -191,7 +192,7 @@ with gr.Blocks(css=css) as demo:
                 min_width=250,
             )
         with gr.Column(scale=1):
-            generate_pubkey = gr.Button (,
+            generate_pubkey = gr.Button (
                 fn=regenerate_pubkey,
                 value=regenerate_pubkey,
                 min_width=250,