trying to launch ollama

mycoder/app.py

@@ -33,4 +33,11 @@ if st.button("Run Query2Plot"):
 
     # TO DO: Display the generated plot if it exists
     if os.path.exists("output/plot.png"):
-        st.image("output/plot.png", caption="Generated Figure")
+        st.image("output/plot.png", caption="Generated Figure")
+
+if st.button("DEMO START OLLAMA"):
+    from mycoder.ollamaini import start_ollama, start_model
+    start_ollama()
+    import time
+    time.sleep(20)  # Wait for Ollama to start
+    start_model()

mycoder/ollamaini.py

@@ -0,0 +1,13 @@
+import subprocess
+
+def download_ollama() -> None:
+    subprocess.run(["curl -fsSL https://ollama.com/install.sh | sh"], shell=True, check=True)
+
+def start_ollama() -> None:
+    subprocess.run(["ollama", "serve", "&"], shell=True, check=True)
+
+def start_model() -> None:
+    subprocess.run(["ollama", "run", "codellama:7b"], check=True)
+
+def stop_model() -> None:
+    subprocess.run(["ollama", "stop", "codellama:7b"], check=True)