Update app.py

app.py

@@ -1,5 +1,5 @@
 # app.py
-from flask import Flask, request, jsonify, Response
+from flask import Flask, request, jsonify, render_template_string, Response
 import os
 import requests
 import json
@@ -11,8 +11,10 @@ import re
 
 app = Flask(__name__)
 
-# Configure logging with file output
-log_handler = RotatingFileHandler('/home/ollama/openwebui.log', maxBytes=1000000, backupCount=5)
+# Configure logging with file output in a writable directory
+log_path = '/app/openwebui.log'
+os.makedirs(os.path.dirname(log_path), exist_ok=True)
+log_handler = RotatingFileHandler(log_path, maxBytes=1000000, backupCount=5)
 log_handler.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)s'))
 logging.getLogger().addHandler(log_handler)
 logging.getLogger().setLevel(logging.INFO)
@@ -42,14 +44,8 @@ class OllamaManager:
                 self.available_models = ALLOWED_MODELS
                 logging.warning("No allowed models found in API response, using ALLOWED_MODELS")
             logging.info(f"Available models: {self.available_models}")
-        except requests.exceptions.ConnectionError as e:
-            logging.error(f"Connection error refreshing models: {e}")
-            self.available_models = ALLOWED_MODELS
-        except requests.exceptions.HTTPError as e:
-            logging.error(f"HTTP error refreshing models: {e}")
-            self.available_models = ALLOWED_MODELS
         except Exception as e:
-            logging.error(f"Unexpected error refreshing models: {e}")
+            logging.error(f"Error refreshing models: {e}")
             self.available_models = ALLOWED_MODELS
     
     def list_models(self) -> List[str]:
@@ -59,7 +55,6 @@ class OllamaManager:
     def generate(self, model_name: str, prompt: str, stream: bool = False, **kwargs) -> Any:
         """Generate text using a model, with optional streaming."""
         if model_name not in self.available_models:
-            logging.warning(f"Attempted to generate with unavailable model: {model_name}")
             return {"status": "error", "message": f"Model {model_name} not available"}
         
         try:
@@ -77,21 +72,14 @@ class OllamaManager:
                 response = requests.post(f"{self.base_url}/api/generate", json=payload, timeout=120)
                 response.raise_for_status()
                 data = response.json()
-                logging.info(f"Generated response with model {model_name}")
                 return {
                     "status": "success",
                     "response": data.get('response', ''),
                     "model": model_name,
                     "usage": data.get('usage', {})
                 }
-        except requests.exceptions.ConnectionError as e:
-            logging.error(f"Connection error generating response: {e}")
-            return {"status": "error", "message": f"Connection error: {str(e)}"}
-        except requests.exceptions.HTTPError as e:
-            logging.error(f"HTTP error generating response: {e}")
-            return {"status": "error", "message": f"HTTP error: {str(e)}"}
         except Exception as e:
-            logging.error(f"Unexpected error generating response: {e}")
+            logging.error(f"Error generating response: {e}")
             return {"status": "error", "message": str(e)}
     
     def health_check(self) -> Dict[str, Any]:
@@ -99,22 +87,15 @@ class OllamaManager:
         try:
             response = requests.get(f"{self.base_url}/api/tags", timeout=10)
             response.raise_for_status()
-            logging.info("Health check successful")
             return {"status": "healthy", "available_models": len(self.available_models)}
-        except requests.exceptions.ConnectionError as e:
-            logging.error(f"Health check connection error: {e}")
-            return {"status": "unhealthy", "error": f"Connection error: {str(e)}"}
-        except requests.exceptions.HTTPError as e:
-            logging.error(f"Health check HTTP error: {e}")
-            return {"status": "unhealthy", "error": f"HTTP error: {str(e)}"}
         except Exception as e:
-            logging.error(f"Health check unexpected error: {e}")
+            logging.error(f"Health check failed: {e}")
             return {"status": "unhealthy", "error": str(e)}
 
 # Initialize Ollama manager
 ollama_manager = OllamaManager(OLLAMA_BASE_URL)
 
-# HTML template for the chat interface with comprehensive, mobile-optimized UI
+# HTML template for the chat interface with improved UI and Sandpack
 HTML_TEMPLATE = '''
 <!DOCTYPE html>
 <html lang="en">
@@ -134,32 +115,28 @@ HTML_TEMPLATE = '''
     </script>
     <style>
         :root {
-            --primary-color: #5a4bff;
-            --secondary-color: #7b3fe4;
+            --primary-color: #667eea;
+            --secondary-color: #764ba2;
             --text-color: #333;
-            --bg-color: #f8fafc;
+            --bg-color: #fafbfc;
             --message-bg-user: var(--primary-color);
             --message-bg-assistant: white;
             --avatar-user: var(--primary-color);
-            --avatar-assistant: #2ea44f;
-            --border-color: #e2e8f0;
+            --avatar-assistant: #28a745;
+            --border-color: #e9ecef;
             --input-bg: white;
-            --sidebar-bg: #ffffff;
-            --sidebar-border: #e2e8f0;
         }
         .dark-mode {
             --primary-color: #3b4a8c;
             --secondary-color: #4a2e6b;
-            --text-color: #e2e8f0;
-            --bg-color: #1a202c;
+            --text-color: #f0f0f0;
+            --bg-color: #1a1a1a;
             --message-bg-user: var(--primary-color);
-            --message-bg-assistant: #2d3748;
+            --message-bg-assistant: #2a2a2a;
             --avatar-user: var(--primary-color);
-            --avatar-assistant: #276749;
-            --border-color: #4a5568;
-            --input-bg: #2d3748;
-            --sidebar-bg: #2d3748;
-            --sidebar-border: #4a5568;
+            --avatar-assistant: #1a7a3a;
+            --border-color: #4a4a4a;
+            --input-bg: #3a3a3a;
         }
         * {
             margin: 0;
@@ -171,110 +148,55 @@ HTML_TEMPLATE = '''
             background: linear-gradient(135deg, var(--primary-color) 0%, var(--secondary-color) 100%);
             color: var(--text-color);
             min-height: 100vh;
-            overflow-x: hidden;
+            padding: 20px;
         }
         .container {
-            display: flex;
-            max-width: 100%;
-            min-height: 100vh;
+            max-width: 1200px;
+            margin: 0 auto;
             background: var(--bg-color);
+            border-radius: 20px;
+            box-shadow: 0 20px 40px rgba(0,0,0,0.1);
+            overflow: hidden;
+            position: relative;
         }
-        .sidebar {
-            width: 250px;
-            background: var(--sidebar-bg);
-            border-right: 1px solid var(--sidebar-border);
-            padding: 20px;
-            position: fixed;
-            height: 100%;
-            transform: translateX(-100%);
-            transition: transform 0.3s ease;
-            z-index: 1000;
-        }
-        .sidebar.open {
-            transform: translateX(0);
-        }
-        .sidebar-toggle {
-            position: fixed;
+        .theme-toggle {
+            position: absolute;
             top: 10px;
-            left: 10px;
-            background: var(--primary-color);
-            color: white;
+            right: 10px;
+            background: none;
             border: none;
-            padding: 10px;
-            border-radius: 8px;
-            cursor: pointer;
-            z-index: 1100;
-        }
-        .sidebar h2 {
-            font-size: 1.5rem;
-            margin-bottom: 20px;
-        }
-        .chat-history {
-            list-style: none;
-            overflow-y: auto;
-            max-height: calc(100vh - 100px);
-        }
-        .chat-history-item {
-            padding: 10px;
-            border-radius: 8px;
-            margin-bottom: 10px;
             cursor: pointer;
-            transition: background 0.2s;
-        }
-        .chat-history-item:hover {
-            background: var(--border-color);
-        }
-        .chat-history-item.active {
-            background: var(--primary-color);
+            font-size: 1.2rem;
             color: white;
         }
-        .main-content {
-            flex: 1;
-            display: flex;
-            flex-direction: column;
-            min-height: 100vh;
-        }
         .header {
             background: linear-gradient(135deg, var(--primary-color) 0%, var(--secondary-color) 100%);
             color: white;
-            padding: 20px;
+            padding: 30px;
             text-align: center;
-            position: relative;
-        }
-        .theme-toggle {
-            position: absolute;
-            top: 20px;
-            right: 20px;
-            background: none;
-            border: none;
-            cursor: pointer;
-            font-size: 1.5rem;
-            color: white;
         }
         .header h1 {
-            font-size: 2rem;
+            font-size: 2.5rem;
             margin-bottom: 10px;
             font-weight: 700;
         }
         .header p {
-            font-size: 1rem;
+            font-size: 1.1rem;
             opacity: 0.9;
         }
         .controls {
-            padding: 15px 20px;
+            padding: 20px 30px;
             background: var(--bg-color);
             border-bottom: 1px solid var(--border-color);
             display: flex;
-            gap: 10px;
+            gap: 15px;
+            align-items: center;
             flex-wrap: wrap;
-            justify-content: center;
         }
         .control-group {
             display: flex;
             align-items: center;
             gap: 8px;
-            flex: 1;
-            min-width: 200px;
         }
         .control-group label {
             font-weight: 600;
@@ -283,8 +205,7 @@ HTML_TEMPLATE = '''
         }
         .control-group select,
         .control-group input {
-            flex: 1;
-            padding: 10px;
+            padding: 8px 12px;
             border: 2px solid var(--border-color);
             border-radius: 8px;
             font-size: 14px;
@@ -298,7 +219,7 @@ HTML_TEMPLATE = '''
             border-color: var(--primary-color);
         }
         .chat-container {
-            flex: 1;
+            height: 500px;
             overflow-y: auto;
             padding: 20px;
             background: var(--bg-color);
@@ -333,7 +254,7 @@ HTML_TEMPLATE = '''
             background: var(--message-bg-assistant);
             padding: 15px 20px;
             border-radius: 18px;
-            max-width: 80%;
+            max-width: 70%;
             box-shadow: 0 2px 10px rgba(0,0,0,0.1);
             line-height: 1.5;
             color: var(--text-color);
@@ -357,7 +278,6 @@ HTML_TEMPLATE = '''
             cursor: pointer;
             font-size: 12px;
             color: var(--text-color);
-            transition: background 0.2s;
         }
         .code-button:hover {
             background: rgba(0,0,0,0.2);
@@ -369,25 +289,24 @@ HTML_TEMPLATE = '''
             overflow: hidden;
         }
         .input-container {
-            padding: 15px 20px;
+            padding: 20px 30px;
             background: var(--bg-color);
             border-top: 1px solid var(--border-color);
         }
         .input-form {
             display: flex;
-            gap: 10px;
-            align-items: center;
+            gap: 15px;
         }
         .input-field {
             flex: 1;
-            padding: 12px 15px;
+            padding: 15px 20px;
             border: 2px solid var(--border-color);
             border-radius: 25px;
             font-size: 16px;
             transition: border-color 0.3s;
             resize: none;
             min-height: 50px;
-            max-height: 150px;
+            max-height: 120px;
             background: var(--input-bg);
             color: var(--text-color);
         }
@@ -396,7 +315,7 @@ HTML_TEMPLATE = '''
             border-color: var(--primary-color);
         }
         .send-button {
-            padding: 12px 20px;
+            padding: 15px 30px;
             background: linear-gradient(135deg, var(--primary-color) 0%, var(--secondary-color) 100%);
             color: white;
             border: none;
@@ -405,7 +324,7 @@ HTML_TEMPLATE = '''
             font-weight: 600;
             cursor: pointer;
             transition: transform 0.2s;
-            min-width: 80px;
+            min-width: 100px;
         }
         .send-button:hover {
             transform: translateY(-2px);
@@ -434,173 +353,107 @@ HTML_TEMPLATE = '''
             border-radius: 18px;
             color: #6c757d;
             font-style: italic;
-            margin: 20px;
         }
         @keyframes fadeIn {
             from { opacity: 0; transform: translateY(10px); }
             to { opacity: 1; transform: translateY(0); }
         }
         @media (max-width: 768px) {
-            .container {
-                flex-direction: column;
-            }
-            .sidebar {
-                width: 100%;
-                height: auto;
-                max-height: 80vh;
-                position: fixed;
-                top: 0;
-                left: 0;
-                transform: translateY(-100%);
-                border-right: none;
-                border-bottom: 1px solid var(--sidebar-border);
-            }
-            .sidebar.open {
-                transform: translateY(0);
-            }
-            .sidebar-toggle {
-                top: 10px;
-                left: 10px;
-                z-index: 1100;
-            }
-            .main-content {
-                margin-top: 60px;
-            }
             .controls {
-                flex-direction: column;
-                gap: 15px;
-            }
-            .control-group {
                 flex-direction: column;
                 align-items: stretch;
             }
-            .control-group select,
-            .control-group input {
-                width: 100%;
+            .control-group {
+                justify-content: space-between;
             }
             .message-content {
-                max-width: 90%;
-            }
-            .header {
-                padding: 15px;
-            }
-            .header h1 {
-                font-size: 1.8rem;
-            }
-            .header p {
-                font-size: 0.9rem;
-            }
-            .input-container {
-                padding: 10px 15px;
-            }
-            .send-button {
-                padding: 10px 15px;
-                min-width: 60px;
+                max-width: 85%;
             }
         }
     </style>
 </head>
 <body>
-    <button class="sidebar-toggle" id="sidebar-toggle">☰</button>
     <div class="container">
-        <div class="sidebar" id="sidebar">
-            <h2>Chat History</h2>
-            <ul class="chat-history" id="chat-history">
-                <!-- Chat history items will be populated here -->
-            </ul>
+        <div class="header">
+            <h1>🤖 OpenWebUI</h1>
+            <p>Chat with your local Ollama models through Hugging Face Spaces</p>
         </div>
-        <div class="main-content">
-            <div class="header">
-                <button class="theme-toggle" id="theme-toggle">🌙</button>
-                <h1>🤖 OpenWebUI</h1>
-                <p>Chat with AI models powered by Ollama on Hugging Face Spaces</p>
-            </div>
-            
-            <div class="controls">
-                <div class="control-group">
-                    <label for="model-select">Model:</label>
-                    <select id="model-select">
-                        <option value="">Select a model...</option>
-                    </select>
-                </div>
-                <div class="control-group">
-                    <label for="temperature">Temperature:</label>
-                    <div style="flex: 1; display: flex; align-items: center; gap: 8px;">
-                        <input type="range" id="temperature" min="0" max="2" step="0.1" value="0.7">
-                        <span id="temp-value">0.7</span>
-                    </div>
-                </div>
-                <div class="control-group">
-                    <label for="max-tokens">Max Tokens:</label>
-                    <input type="number" id="max-tokens" min="1" max="4096" value="2048">
-                </div>
+        
+        <div class="controls">
+            <div class="control-group">
+                <label for="model-select">Model:</label>
+                <select id="model-select">
+                    <option value="">Select a model...</option>
+                </select>
             </div>
-            
-            <div class="chat-container" id="chat-container">
-                <div class="message assistant">
-                    <div class="message-avatar">AI</div>
-                    <div class="message-content">
-                        Hello! I'm your AI assistant powered by Ollama. How can I help you today?
-                    </div>
-                </div>
+            <div class="control-group">
+                <label for="temperature">Temperature:</label>
+                <input type="range" id="temperature" min="0" max="2" step="0.1" value="0.7">
+                <span id="temp-value">0.7</span>
             </div>
-            
-            <div class="typing-indicator" id="typing-indicator">
-                AI is thinking...
+            <div class="control-group">
+                <label for="max-tokens">Max Tokens:</label>
+                <input type="number" id="max-tokens" min="1" max="4096" value="2048">
             </div>
-            
-            <div class="input-container">
-                <form class="input-form" id="chat-form">
-                    <textarea 
-                        class="input-field" 
-                        id="message-input" 
-                        placeholder="Type your message here..."
-                        rows="1"
-                    ></textarea>
-                    <button type="submit" class="send-button" id="send-button">
-                        Send
-                    </button>
-                </form>
+        </div>
+        
+        <div class="chat-container" id="chat-container">
+            <div class="message assistant">
+                <div class="message-avatar">AI</div>
+                <div class="message-content">
+                    Hello! I'm your AI assistant powered by Ollama. How can I help you today?
+                </div>
             </div>
-            
-            <div class="status" id="status"></div>
         </div>
+        
+        <div class="typing-indicator" id="typing-indicator">
+            AI is thinking...
+        </div>
+        
+        <div class="input-container">
+            <form class="input-form" id="chat-form">
+                <textarea 
+                    class="input-field" 
+                    id="message-input" 
+                    placeholder="Type your message here..."
+                    rows="1"
+                ></textarea>
+                <button type="submit" class="send-button" id="send-button">
+                    Send
+                </button>
+            </form>
+        </div>
+        
+        <div class="status" id="status"></div>
     </div>
 
     <script type="module">
         import { Sandpack } from 'https://esm.sh/@codesandbox/sandpack-react@latest';
 
-        let conversationHistory = JSON.parse(localStorage.getItem('chatHistory')) || [];
-        let currentConversationId = null;
+        let conversationHistory = [];
         let currentMessageDiv = null;
         let currentCodeBlocks = [];
 
         document.addEventListener('DOMContentLoaded', function() {
             loadModels();
-            loadChatHistory();
             setupEventListeners();
             autoResizeTextarea();
         });
-
+        
         function toggleTheme() {
             document.body.classList.toggle('dark-mode');
             const themeToggle = document.getElementById('theme-toggle');
             themeToggle.textContent = document.body.classList.contains('dark-mode') ? '☀️' : '🌙';
             localStorage.setItem('theme', document.body.classList.contains('dark-mode') ? 'dark' : 'light');
         }
-
+        
         function loadTheme() {
             if (localStorage.getItem('theme') === 'dark') {
                 document.body.classList.add('dark-mode');
                 document.getElementById('theme-toggle').textContent = '☀️';
             }
         }
-
-        function toggleSidebar() {
-            const sidebar = document.getElementById('sidebar');
-            sidebar.classList.toggle('open');
-        }
-
+        
         async function loadModels() {
             const modelSelect = document.getElementById('model-select');
             modelSelect.innerHTML = '<option value="">Loading models...</option>';
@@ -632,51 +485,7 @@ HTML_TEMPLATE = '''
                 showStatus('Failed to load models: ' + error.message, 'error');
             }
         }
-
-        function loadChatHistory() {
-            const chatHistoryList = document.getElementById('chat-history');
-            chatHistoryList.innerHTML = '';
-            conversationHistory.forEach((conv, index) => {
-                const li = document.createElement('li');
-                li.className = 'chat-history-item';
-                li.textContent = `Chat ${index + 1} - ${new Date(conv.timestamp).toLocaleString()}`;
-                li.dataset.convId = index;
-                li.addEventListener('click', () => loadConversation(index));
-                chatHistoryList.appendChild(li);
-            });
-            if (conversationHistory.length > 0) {
-                loadConversation(conversationHistory.length - 1);
-            }
-        }
-
-        function loadConversation(convId) {
-            currentConversationId = convId;
-            const chatContainer = document.getElementById('chat-container');
-            chatContainer.innerHTML = '';
-            const conversation = conversationHistory[convId];
-            conversation.messages.forEach(msg => {
-                const messageDiv = addMessage(msg.content, msg.role, false, false);
-                if (msg.role === 'assistant') {
-                    processCodeBlocks(messageDiv, msg.content);
-                }
-            });
-            const historyItems = document.querySelectorAll('.chat-history-item');
-            historyItems.forEach(item => item.classList.remove('active'));
-            historyItems[convId].classList.add('active');
-        }
-
-        function saveConversation() {
-            if (!currentConversationId && currentConversationId !== 0) {
-                conversationHistory.push({
-                    timestamp: Date.now(),
-                    messages: []
-                });
-                currentConversationId = conversationHistory.length - 1;
-            }
-            localStorage.setItem('chatHistory', JSON.stringify(conversationHistory));
-            loadChatHistory();
-        }
-
+        
         function setupEventListeners() {
             document.getElementById('chat-form').addEventListener('submit', handleSubmit);
             document.getElementById('temperature').addEventListener('input', function() {
@@ -684,14 +493,13 @@ HTML_TEMPLATE = '''
             });
             document.getElementById('message-input').addEventListener('input', autoResizeTextarea);
             document.getElementById('theme-toggle').addEventListener('click', toggleTheme);
-            document.getElementById('sidebar-toggle').addEventListener('click', toggleSidebar);
             loadTheme();
         }
         
         function autoResizeTextarea() {
             const textarea = document.getElementById('message-input');
             textarea.style.height = 'auto';
-            textarea.style.height = Math.min(textarea.scrollHeight, 150) + 'px';
+            textarea.style.height = Math.min(textarea.scrollHeight, 120) + 'px';
         }
         
         async function handleSubmit(e) {
@@ -712,12 +520,6 @@ HTML_TEMPLATE = '''
             }
             
             addMessage(message, 'user');
-            if (currentConversationId === null) {
-                saveConversation();
-            }
-            conversationHistory[currentConversationId].messages.push({ role: 'user', content: message });
-            localStorage.setItem('chatHistory', JSON.stringify(conversationHistory));
-            
             messageInput.value = '';
             autoResizeTextarea();
             showTypingIndicator(true);
@@ -764,15 +566,11 @@ HTML_TEMPLATE = '''
                 }
                 
                 processCodeBlocks(currentMessageDiv, accumulatedResponse);
-                conversationHistory[currentConversationId].messages.push({ role: 'assistant', content: accumulatedResponse });
-                localStorage.setItem('chatHistory', JSON.stringify(conversationHistory));
                 showStatus(`Response generated using ${model}`, 'success');
             } catch (error) {
                 showTypingIndicator(false);
                 if (currentMessageDiv) {
                     updateMessage(currentMessageDiv, 'Sorry, I encountered a network error.');
-                    conversationHistory[currentConversationId].messages.push({ role: 'assistant', content: 'Sorry, I encountered a network error.' });
-                    localStorage.setItem('chatHistory', JSON.stringify(conversationHistory));
                 } else {
                     addMessage('Sorry, I encountered a network error.', 'assistant');
                 }
@@ -780,7 +578,7 @@ HTML_TEMPLATE = '''
             }
         }
         
-        function addMessage(content, sender, isStreaming = false, save = true) {
+        function addMessage(content, sender, isStreaming = false) {
             const chatContainer = document.getElementById('chat-container');
             const messageDiv = document.createElement('div');
             messageDiv.className = `message ${sender}`;
@@ -798,12 +596,8 @@ HTML_TEMPLATE = '''
             chatContainer.appendChild(messageDiv);
             chatContainer.scrollTop = chatContainer.scrollHeight;
             
-            if (!isStreaming && save) {
-                if (currentConversationId === null) {
-                    saveConversation();
-                }
-                conversationHistory[currentConversationId].messages.push({ role: sender, content: content });
-                localStorage.setItem('chatHistory', JSON.stringify(conversationHistory));
+            if (!isStreaming) {
+                conversationHistory.push({ role: sender, content: content });
             }
             return messageDiv;
         }
@@ -827,19 +621,23 @@ HTML_TEMPLATE = '''
                 const code = match[2].trim();
                 const startIndex = match.index;
                 
+                // Add text before the code block
                 if (startIndex > lastIndex) {
                     fragments.push({ type: 'text', content: content.slice(lastIndex, startIndex) });
                 }
                 
+                // Add code block
                 fragments.push({ type: 'code', language, content: code });
                 currentCodeBlocks.push({ language, content: code });
                 lastIndex = codeBlockRegex.lastIndex;
             }
             
+            // Add remaining text
             if (lastIndex < content.length) {
                 fragments.push({ type: 'text', content: content.slice(lastIndex) });
             }
             
+            // Clear message content and rebuild with fragments
             messageContent.innerHTML = '';
             fragments.forEach((fragment, index) => {
                 if (fragment.type === 'text') {
@@ -883,7 +681,7 @@ HTML_TEMPLATE = '''
                     script.textContent = `
                         import { Sandpack } from '@codesandbox/sandpack-react';
                         import { createRoot } from 'react-dom';
-                        const root = createRoot(document.getElementById('sandpack-${currentConversationId}-${index}'));
+                        const root = createRoot(document.getElementById('sandpack-${index}'));
                         root.render(
                             React.createElement(Sandpack, {
                                 template: "${fragment.language === 'javascript' ? 'react' : fragment.language}",
@@ -901,12 +699,14 @@ HTML_TEMPLATE = '''
                     `;
                     
                     const sandboxDiv = document.createElement('div');
-                    sandboxDiv.id = `sandpack-${currentConversationId}-${index}`;
+                    sandboxDiv.id = `sandpack-${index}`;
                     codeContainer.appendChild(sandboxDiv);
                     codeContainer.appendChild(script);
                     messageContent.appendChild(codeContainer);
                 }
             });
+            
+            conversationHistory.push({ role: 'assistant', content: content });
         }
         
         function showTypingIndicator(show) {
@@ -943,7 +743,6 @@ def chat():
     try:
         data = request.get_json()
         if not data or 'prompt' not in data or 'model' not in data:
-            logging.warning("Chat request missing 'prompt' or 'model' field")
             return jsonify({"status": "error", "message": "Prompt and model are required"}), 400
         
         prompt = data['prompt']
@@ -956,15 +755,10 @@ def chat():
         
         if stream and isinstance(result, requests.Response):
             def generate_stream():
-                try:
-                    for chunk in result.iter_content(chunk_size=None):
-                        yield chunk
-                except Exception as e:
-                    logging.error(f"Streaming error: {e}")
-                    yield json.dumps({"status": "error", "message": str(e)}).encode()
+                for chunk in result.iter_content(chunk_size=None):
+                    yield chunk
             return Response(generate_stream(), content_type='application/json')
         else:
-            logging.info(f"Non-streaming chat response generated with model {model}")
             return jsonify(result), 200 if result["status"] == "success" else 500
     except Exception as e:
         logging.error(f"Chat endpoint error: {e}")
@@ -975,7 +769,6 @@ def get_models():
     """Get available models."""
     try:
         models = ollama_manager.list_models()
-        logging.info(f"Returning models: {models}")
         return jsonify({
             "status": "success",
             "models": models,
@@ -1001,7 +794,7 @@ def health_check():
             "status": "unhealthy",
             "error": str(e),
             "timestamp": time.time()
-        }), 503
+        }), 500
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=False)