import gradio as gr
import subprocess
import re
import tempfile
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 save HTML and return a file path
def save_html_file(response):
if "```html" in response.lower():
match = re.search(r'```html\n(.*?)```', response, re.DOTALL)
if match:
html_content = match.group(1).strip()
else:
return None
elif response.strip().lower().startswith("") or "