Just don't ask at this point

.env.sample

@@ -1,9 +1,15 @@
+# (optional) if Google Signup needed
 GOOGLE_CLIENT_ID=YOUR-CLIENT-ID
 GOOGLE_SECRET_KEY=YOUR-SECRET-KEY
 
+# Where the app is being run
 ALLOWED_HOSTS=localhost,127.0.0.1,192.168.1.13,localhost,127.0.0.1,192.168.1.13,192.168.1.150
 CSRF_TRUSTED_ORIGINS=http://localhost,http://127.0.0.1,http://192.168.1.13,http://localhost:8000,http://127.0.0.1:8000,http://192.168.1.13:8000,http://192.168.1.150:8000
 
+# default admin credentials
 DJANGO_SUPERUSER_USERNAME=admin
 [email protected]
 DJANGO_SUPERUSER_PASSWORD=admin
+
+# specify your redis url
+CELERY_BROKER_URL=redis://redis:6379/0

.gitignore

@@ -180,4 +180,7 @@ node_modules
 *output.css
 
 # productions static files
-staticfiles
+staticfiles
+
+# algolyzer playground aiml models
+aiml_models

Algolyzer/Algolyzer/__init__.py

@@ -0,0 +1,3 @@
+from .celery_app import celery_app
+
+__all__ = ("celery_app",)

Algolyzer/Algolyzer/celery_app.py

@@ -0,0 +1,17 @@
+import os
+
+from celery import Celery
+
+# Use the existing DJANGO_SETTINGS_MODULE if it's already set, otherwise set a default
+os.environ.setdefault(
+    "DJANGO_SETTINGS_MODULE",
+    os.getenv("DJANGO_SETTINGS_MODULE", "Algolyzer.settings.development"),
+)
+
+celery_app = Celery("Algolyzer")
+
+# Load task modules from all registered Django app configs.
+celery_app.config_from_object("django.conf:settings", namespace="CELERY")
+
+# Auto-discover tasks in installed apps
+celery_app.autodiscover_tasks()

Algolyzer/Algolyzer/settings/base.py

@@ -32,6 +32,7 @@
     "quiz",
     "study",
     "community",
+    "playground",
 ]
 
 MIDDLEWARE = [
@@ -94,3 +95,16 @@
 STATIC_URL = "/static/"
 STATICFILES_DIRS = [BASE_DIR / "static"]
 STATIC_ROOT = BASE_DIR / "staticfiles"
+
+# Celery Configurations
+# settings.py
+
+# Celery Settings
+CELERY_BROKER_URL = os.getenv(
+    "CELERY_BROKER_URL", "redis://localhost:6379/0"
+)  # URL for Redis
+CELERY_ACCEPT_CONTENT = ["json"]  # Accept JSON formatted tasks
+CELERY_TASK_SERIALIZER = "json"  # Use JSON serialization
+
+# AIML downloaded models directory
+MODEL_DIR = "/aiml_models"

Algolyzer/Algolyzer/urls.py

@@ -26,6 +26,7 @@
     path("quiz/", include("quiz.urls")),
     path("study/", include("study.urls")),
     path("community/", include("community.urls")),
+    path("playground/", include("playground.urls")),
     # allauth urls
     path("accounts/", include("allauth.urls")),
 ]

Algolyzer/playground/admin.py

@@ -0,0 +1,3 @@
+# from django.contrib import admin
+
+# Register your models here.

Algolyzer/playground/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class PlaygroundConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "playground"

Algolyzer/playground/management/commands/download_models.py

@@ -0,0 +1,38 @@
+import os
+
+from django.core.management.base import BaseCommand
+from transformers import AutoModel, AutoTokenizer
+
+# Define models to download
+MODELS = {
+    "sentiment-analysis": "distilbert-base-uncased-finetuned-sst-2-english",
+    "text-classification": "facebook/bart-large-mnli",
+}
+
+# Define directory for saving models
+MODEL_DIR = os.path.join("playground", "aiml_models")
+
+
+class Command(BaseCommand):
+    help = "Download Hugging Face models for use in Django views"
+
+    def handle(self, *args, **kwargs):
+        os.makedirs(MODEL_DIR, exist_ok=True)
+        for task, model_name in MODELS.items():
+            self.stdout.write(
+                self.style.SUCCESS(f"Downloading {task} model: {model_name}...")
+            )
+
+            # Download model and tokenizer
+            model = AutoModel.from_pretrained(model_name)
+            tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+            # Save them in the specified directory
+            model.save_pretrained(os.path.join(MODEL_DIR, model_name.replace("/", "_")))
+            tokenizer.save_pretrained(
+                os.path.join(MODEL_DIR, model_name.replace("/", "_"))
+            )
+
+            self.stdout.write(
+                self.style.SUCCESS(f"✔ Model {model_name} downloaded successfully!\n")
+            )

Algolyzer/playground/migrations/0001_initial.py

@@ -0,0 +1,56 @@
+# Generated by Django 5.1.4 on 2025-02-10 16:23
+
+import django.db.models.deletion
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    initial = True
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="PlaygroundTask",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                ("celery_task_id", models.CharField(max_length=255, unique=True)),
+                ("model_name", models.CharField(max_length=100)),
+                ("input_data", models.TextField()),
+                (
+                    "status",
+                    models.CharField(
+                        choices=[
+                            ("PENDING", "Pending"),
+                            ("PROCESSING", "Processing"),
+                            ("COMPLETED", "Completed"),
+                            ("FAILED", "Failed"),
+                        ],
+                        default="PENDING",
+                        max_length=20,
+                    ),
+                ),
+                ("result", models.TextField(blank=True, null=True)),
+                ("created_at", models.DateTimeField(auto_now_add=True)),
+                ("completed_at", models.DateTimeField(blank=True, null=True)),
+                (
+                    "user",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        to=settings.AUTH_USER_MODEL,
+                    ),
+                ),
+            ],
+        ),
+    ]

Algolyzer/playground/models.py

@@ -0,0 +1,23 @@
+from django.contrib.auth.models import User
+from django.db import models
+
+
+class PlaygroundTask(models.Model):
+    STATUS_CHOICES = [
+        ("PENDING", "Pending"),
+        ("PROCESSING", "Processing"),
+        ("COMPLETED", "Completed"),
+        ("FAILED", "Failed"),
+    ]
+
+    user = models.ForeignKey(User, on_delete=models.CASCADE)
+    celery_task_id = models.CharField(max_length=255, unique=True)  # Celery's Task ID
+    model_name = models.CharField(max_length=100)  # Selected ML model
+    input_data = models.TextField()  # Store user input (can be JSON)
+    status = models.CharField(max_length=20, choices=STATUS_CHOICES, default="PENDING")
+    result = models.TextField(blank=True, null=True)  # ML output
+    created_at = models.DateTimeField(auto_now_add=True)
+    completed_at = models.DateTimeField(blank=True, null=True)
+
+    def __str__(self):
+        return f"Task {self.celery_task_id} - {self.status}"

Algolyzer/playground/tasks.py

@@ -0,0 +1,57 @@
+import os
+
+from celery import shared_task
+from django.conf import settings
+from django.utils.timezone import now
+from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
+
+from .models import PlaygroundTask
+
+# Load ML models at startup (avoid reloading for each request)
+MODEL_REGISTRY = {
+    "model_1": "distilbert-base-uncased-finetuned-sst-2-english",  # Replace with actual model paths
+    "model_2": "model_2",
+}
+
+
+@shared_task(bind=True)
+def process_ml_task(self, task_db_id, model_name, input_data):
+    """Celery task to process ML model input."""
+    try:
+        # Fetch the task from the database
+        task = PlaygroundTask.objects.get(id=task_db_id)
+        task.status = "PROCESSING"
+        task.save()
+
+        # Load the selected model
+        model_path = MODEL_REGISTRY.get(model_name)
+        MODEL_DIR = os.path.join(
+            settings.BASE_DIR, "playground", "aiml_models", model_path
+        )
+
+        # Load model from saved directory
+        tokenizer = AutoTokenizer.from_pretrained(MODEL_DIR)
+        model = AutoModelForSequenceClassification.from_pretrained(MODEL_DIR)
+        sentiment_pipeline = pipeline(
+            "sentiment-analysis", model=model, tokenizer=tokenizer
+        )
+
+        if not model:
+            raise ValueError(f"Model '{model_name}' not found.")
+
+        # Process the input with the model
+        result = sentiment_pipeline(input_data)[0]
+
+        # Update the task in the database
+        task.status = "COMPLETED"
+        task.result = str(result)  # Convert to JSON-friendly format if needed
+        task.completed_at = now()
+        task.save()
+
+    except Exception as e:
+        task.status = "FAILED"
+        task.result = str(e)
+        task.save()
+        # raise e
+
+    return task.result  # Celery stores the result

Algolyzer/playground/templates/playground/home.html

@@ -0,0 +1,23 @@
+{% extends "base.html" %}
+
+{% block content %}
+<div class="container mx-auto">
+    <h1 class="text-center">This is Playground App</h1>
+    {% if error %}
+        <p style="color: red;">{{ error }}</p>
+    {% endif %}
+    <form method="post" action="{% url 'submit_task_view' %}">
+        {% csrf_token %}
+        <label for="input_data">Enter Input:</label>
+        <input type="text" name="input_data" required>
+
+        <label for="model_name">Select Model:</label>
+        <select name="model_name">
+            <option value="model_1">Model 1</option>
+            <option value="model_2">Model 2</option>
+        </select>
+
+        <button type="submit">Submit Task</button>
+    </form>
+</div>
+{% endblock %}

Algolyzer/playground/templates/playground/task_status.html

@@ -0,0 +1,20 @@
+{% extends "base.html" %}
+
+{% block content %}
+    <h1>Task Status</h1>
+
+    {% if error %}
+        <p style="color: red;">{{ error }}</p>
+    {% else %}
+        <p><strong>Task ID:</strong> {{ task_id }}</p>
+        <p><strong>Status:</strong> {{ status }}</p>
+
+        {% if result %}
+            <p><strong>Result:</strong> {{ result }}</p>
+        {% else %}
+            <p>Processing... Refresh the page to check status.</p>
+        {% endif %}
+    {% endif %}
+
+    <a href="{% url 'playground_home' %}">Submit another task</a>
+{% endblock%}

Algolyzer/playground/tests.py

@@ -0,0 +1,3 @@
+# from django.test import TestCase
+
+# Create your tests here.

Algolyzer/playground/urls.py

@@ -0,0 +1,9 @@
+from django.urls import path
+
+from .views import check_task_status_view, playground_home, submit_task_view
+
+urlpatterns = [
+    path("", playground_home, name="playground_home"),
+    path("submit_task/", submit_task_view, name="submit_task_view"),
+    path("task_status/<str:task_id>/", check_task_status_view, name="task_status_view"),
+]