Update app.py

app.py

@@ -31,7 +31,7 @@ def regenerate_pubkey(pubkey, oauth_token: gr.OAuthToken | None):
     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"
     ollama_pubkey = f"cat {HOME}/{hash_oauth}/.ollama/id_ed25519.pub"  
-    ollama_start = f"HOME={HOME}/{hash_oauth} ollama serve & {generate_ollama_pid_file} & sleep 5 & {generate_ollama_host_file}"    
+    ollama_start = f"HOME={HOME}/{hash_oauth} ollama serve & {generate_ollama_pid_file} & sleep 5"    
     ollama_stop = f"kill -9 'cat {HOME}/{hash_oauth}/ollama.pid'"
     delete_home = f"rm -Rf {HOME}/{hash_oauth}/.ollama"
     
@@ -58,8 +58,15 @@ def regenerate_pubkey(pubkey, oauth_token: gr.OAuthToken | None):
         print(result)
         if result.returncode != 0:
             raise Exception(f"Error starting Ollama {result.stderr}")
+        print(f"echo $(ss -natp | grep (cat ollama.pid) | awk '{{print $4}}')")
         print("Ollama Pubkey Obtained!")
 
+        result = subprocess.run(generate_ollama_host_file, shell=True, capture_output=True)
+        print(result)
+        if result.returncode != 0:
+            raise Exception(f"Error generating Ollama Host File {result.stderr}")
+        print("Ollama Host File Created!")
+
         
     except Exception as e:
         return (f"Error: {e}", "error.png")