Update Dockerfile

Dockerfile

@@ -1,10 +1,8 @@
 FROM ollama/ollama:latest
 
-# Install Python & Dependencies
 RUN apt-get update && apt-get install -y python3 python3-pip && \
     pip3 install flask flask-cors requests --break-system-packages
 
-# Set up environment variables
 ENV OLLAMA_HOST=127.0.0.1:11434
 ENV HOME=/home/ollama
 RUN mkdir -p /home/ollama/.ollama && chmod -R 777 /home/ollama
@@ -16,7 +14,7 @@ import requests
 from flask_cors import CORS
 
 app = Flask(__name__)
-# Enable CORS for browser direct access
+# Enable CORS for direct browser access
 CORS(app, resources={r"/*": {"origins": "*", "allow_headers": ["Content-Type", "x-api-key"]}})
 
 UNLIMITED_KEY = "sk-ess4l0ri37"
@@ -31,13 +29,35 @@ def proxy():
         return jsonify({"error": "Unauthorized"}), 401
 
     try:
-        # Proxy to local Ollama
+        data = request.json
+        # Convert your 'prompt' into a 'messages' array for the /api/chat endpoint
+        ollama_payload = {
+            "model": data.get("model", "dolphin3:8b"),
+            "messages": [{"role": "user", "content": data.get("prompt", "")}],
+            "stream": False,
+            "options": {
+                "temperature": data.get("temperature", 0.7)
+            }
+        }
+
+        # TALKING TO /api/chat INSTEAD OF /api/generate
         resp = requests.post(
-            "http://127.0.0.1:11434/api/generate", 
-            json=request.json, 
-            timeout=300
+            "http://127.0.0.1:11434/api/chat", 
+            json=ollama_payload, 
+            timeout=180
         )
-        return Response(resp.content, status=resp.status_code, content_type=resp.headers.get('Content-Type'))
+        
+        if resp.status_code != 200:
+            return jsonify({"error": "Ollama Error", "details": resp.text}), resp.status_code
+
+        ollama_res = resp.json()
+        
+        # Flatten the response back to the format your frontend expects
+        return jsonify({
+            "response": ollama_res.get("message", {}).get("content", ""),
+            "done": True
+        })
+        
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
@@ -45,7 +65,7 @@ if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)
 EOF
 
-# --- Startup Script ---
+# --- start.sh stays the same ---
 RUN cat <<'EOF' > /start.sh
 #!/bin/bash
 ollama serve &