Disable Celery on HF Spaces, run simulations synchronously

simulations/views_views.py

@@ -1,8 +1,11 @@
 from django.shortcuts import render, get_object_or_404, redirect
 from django.http import JsonResponse
 from .models import SimulationMethod, SimulationRun
-from .tasks import run_simulation
+from .tasks import run_simulation, clean_for_json, SIMULATION_METHODS
 import json
+import os
+
+HF_SPACES = os.environ.get('HF_SPACES', '').lower() in ('true', '1', 'yes')
 
 
 def home(request):
@@ -51,7 +54,29 @@ def run_create(request, method_slug=None):
             parameters=parameters,
         )
         
-        run_simulation.delay(run.id)
+        if HF_SPACES:
+            run.set_running()
+            method_func = SIMULATION_METHODS.get(run.method.slug)
+            
+            if method_func:
+                gen = method_func(run.parameters)
+                result = None
+                try:
+                    while True:
+                        progress = next(gen)
+                        run.progress = progress
+                        run.save(update_fields=['progress'])
+                except StopIteration as e:
+                    result = e.value
+                
+                if result:
+                    result = clean_for_json(result)
+                
+                run.set_success(result)
+            else:
+                run.set_failure(f"Méthode inconnue: {run.method.slug}")
+        else:
+            run_simulation.delay(run.id)
         
         return redirect('run_detail', run_id=run.id)
     

simulationserver/celery.py

@@ -1,11 +1,13 @@
 import os
 from celery import Celery
 
-os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'simulationserver.settings')
+HF_SPACES = os.environ.get('HF_SPACES', '').lower() in ('true', '1', 'yes')
 
-app = Celery('simulationserver')
-app.config_from_object('django.conf:settings', namespace='CELERY')
-app.autodiscover_tasks()
+if not HF_SPACES:
+    os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'simulationserver.settings')
+    app = Celery('simulationserver')
+    app.config_from_object('django.conf:settings', namespace='CELERY')
+    app.autodiscover_tasks()
 
 
 @app.task(bind=True, ignore_result=True)