Update app.py

app.py

@@ -2,26 +2,25 @@ import gradio as gr
 import requests
 import subprocess
 import time
-import os
+import sys
 
-# Start the Ollama server and wait for it to output the URL
-subprocess.Popen(['bash', 'run_ollama.sh'])
-ollama_url_file = 'ollama_url.txt'
+# Start the Ollama server and capture its URL output
+process = subprocess.Popen(['bash', 'run_ollama.sh'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+stdout, stderr = process.communicate()
 
-# Wait for the server URL file to be created
-while not os.path.exists(ollama_url_file):
-    time.sleep(1)
+# Check for errors or retrieve the URL from stdout
+if process.returncode != 0:
+    print(f"Error starting Ollama: {stderr.decode()}")
+    sys.exit(1)
 
-# Read the URL from the file
-with open(ollama_url_file, 'r') as file:
-    OLLAMA_URL = file.read().strip()
+ollama_url = stdout.decode().strip().split('\n')[-1].split()[-1]  # Extract the URL
 
 console_output = []
 
 def call_ollama_api(prompt):
     try:
         response = requests.post(
-            f'{OLLAMA_URL}/generate',
+            f'{ollama_url}/generate',
             json={"prompt": prompt}
         )
         response.raise_for_status()  # Raise an exception for HTTP errors