Update app.py

app.py

@@ -2,29 +2,35 @@ import gradio as gr
 import requests
 import subprocess
 import time
+import os
 
 # Start the Ollama server and wait for it to output the URL
 subprocess.Popen(['bash', 'run_ollama.sh'])
-time.sleep(10)  # Adjust sleep time based on how long it takes Ollama to start
+ollama_url_file = 'ollama_url.txt'
+
+# Wait for the server URL file to be created
+while not os.path.exists(ollama_url_file):
+    time.sleep(1)
 
 # Read the URL from the file
-with open('ollama_url.txt', 'r') as file:
+with open(ollama_url_file, 'r') as file:
     OLLAMA_URL = file.read().strip()
 
 console_output = []
 
 def call_ollama_api(prompt):
-    response = requests.post(
-        f'{OLLAMA_URL}/generate',
-        json={"prompt": prompt}
-    )
-    if response.status_code == 200:
+    try:
+        response = requests.post(
+            f'{OLLAMA_URL}/generate',
+            json={"prompt": prompt}
+        )
+        response.raise_for_status()  # Raise an exception for HTTP errors
         result = response.json()
         output = result.get('text', '')
         console_output.append(f"Prompt: {prompt}\nResponse: {output}")
         return output
-    else:
-        error_message = f"Error: {response.status_code} - {response.text}"
+    except requests.exceptions.RequestException as e:
+        error_message = f"Error: {e}"
         console_output.append(error_message)
         return error_message
 
@@ -51,4 +57,5 @@ console_iface = gr.Interface(
     description="See the console outputs here."
 )
 
+# Launch a tabbed interface
 gr.TabbedInterface([iface, console_iface], ["Chat", "Console"]).launch()