Initial commit for Hugging Face Space

Dockerfile

@@ -9,6 +9,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 RUN mkdir -p /app/cache && chmod -R 777 /app/cache
 ENV HF_HOME=/app/cache
+ENV PYTHONUNBUFFERED=1
+ENV EVENTLET_NO_GREENDNS=yes
 
 COPY . /app
 

templates/index.html

@@ -396,7 +396,13 @@
     </div>
 
     <script>
-        let socket = io();
+        // Replace the socket initialization with:
+        let socket = io({
+            transports: ['websocket'],
+            reconnection: true,
+            reconnectionAttempts: 5,
+            reconnectionDelay: 1000
+        });
         let beams = {};
         let completedBeams = [];
         let isGenerating = false;
@@ -412,11 +418,22 @@
 
         function setupSocketListeners() {
             socket.on('beam_update', function(data) {
+                console.log('Received beam update:', data); // Add logging
                 const { beam_idx, text } = data;
-                if (!beams[beam_idx]) {
-                    createBeamContainer(beam_idx);
-                }
-                document.getElementById(`beam-${beam_idx}`).textContent = text;
+
+                requestAnimationFrame(() => {
+                    if (!beams[beam_idx]) {
+                        createBeamContainer(beam_idx);
+                    }
+                    const beamElement = document.getElementById(`beam-${beam_idx}`);
+                    if (beamElement) {
+                        beamElement.textContent = text;
+                        // Force a reflow to ensure the update is visible
+                        beamElement.style.display = 'none';
+                        beamElement.offsetHeight; // Force reflow
+                        beamElement.style.display = 'block';
+                    }
+                });
             });
 
             socket.on('beam_finished', function(data) {
@@ -475,20 +492,30 @@
 
         function resetConnection() {
             if (socket) {
-                socket.disconnect();
+                socket.removeAllListeners();
+                socket.close();
             }
-            socket = io();
+            socket = io({
+                transports: ['websocket'],
+                reconnection: true,
+                reconnectionAttempts: 5,
+                reconnectionDelay: 1000
+            });
             setupSocketListeners();
         }
 
         function generate() {
             if (isGenerating) return;
 
+            console.log('Starting generation...'); // Add logging
+
+            // Clear previous state
             document.getElementById('beams').innerHTML = '';
             document.getElementById('completed-list').innerHTML = '';
             beams = {};
             completedBeams = [];
 
+            // Reset connection before each generation
             resetConnection();
 
             const prompt = document.getElementById('prompt').value;
@@ -496,6 +523,10 @@
             const maxTokens = parseInt(document.getElementById('max_tokens').value);
             const sleepTime = parseInt(document.getElementById('sleep_time').value);
 
+            console.log('Emitting generate event with params:', {
+                prompt, numBeams, maxTokens, sleepTime
+            });
+
             socket.emit('generate', {
                 prompt: prompt,
                 num_beams: numBeams,