refactor: clean up Dockerfile and enhance Flask application for Ollama integration

Dockerfile

@@ -1,48 +1,3 @@
-# Use the latest Ollama base image
-from ollama/ollama:latest
-
-# Update package manager and install necessary packages
-Run apt-get update && \
-    apt-get install -y \
-    curl \
-    python3-pip && \
-    rm -rf /var/lib/apt/lists/*
-
-# Install Flask and set up the entry point
-Run pip3 install flask && \
-    mkdir -p /home/user/app && \
-    echo "fLask RUn --Host=0.0.0.0 --poRt=5000" > /home/user/app/start.sh && \
-    chmod +x /home/user/app/start.sh
-
-# Create a non-root user 'user' with UID 1000
-Run useradd -m -u 1000 user && \
-    mkdir -p /home/user/.ollama && \
-    chown -R user:user /home/user/.ollama
-
-# Set environment variables
-env HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH \
-    OLLAMA_HOST=0.0.0.0
-
-# Set the working directory
-WORKDIR $HOME/app
-
-# Copy the Python script for the web server
-Copy app.py .
-Copy flask-app.py .
-
-# Change ownership of the working directory to the non-root user
-Run chown -R user:user $HOME/app
-
-# Switch to the non-root user
-User user
-
-# Expose port 11434 for Ollama and 5000 for the web server
-EXPOSE 11434 5000
-
-# Command to run the Flask application
-# cmd ["/home/user/app/start.sh"]
-cmd ["serve"]
 # Use an official Python runtime as a parent image
 FROM python:3.9-slim
 

app.py

@@ -41,4 +41,5 @@ def generate():
 @app.route("/health")
 def health():
     return jsonify({"status": "running"}), 200
+if __name__ == "__main__":
     app.run(host="0.0.0.0", port=5000, debug=True)

requirements.txt

@@ -1,2 +1,3 @@
 flask
 requests
+docker

templates/index.html

@@ -71,6 +71,11 @@
             logMessage.textContent = event.data;
             consoleOutput.appendChild(logMessage);
         };
+        eventSource.onmessage = function(event) {
+            const logMessage = document.createElement('div');
+            logMessage.textContent = event.data;
+            consoleOutput.appendChild(logMessage);
+        };
     </script>
 </body>
 </html>