import gradio as gr
import subprocess
import re
import tempfile
import webbrowser
import os
def run_ollama_prompt(prompt, model):
"""Runs a prompt using Ollama CLI and returns the response."""
command = ["ollama", "run", model, prompt]
try:
result = subprocess.run(command, capture_output=True, text=True, check=True, encoding="utf-8")
response = result.stdout.strip()
# Remove ... tags if present
response = re.sub(r'.*?', '', response, flags=re.DOTALL).strip()
return response
except subprocess.CalledProcessError as e:
return f"Error running Ollama: {e}"
# Function to handle HTML output
def save_and_open_html(response):
if response.strip().lower().startswith("") or "