streaming_view.py

from django.http import JsonResponse, StreamingHttpResponse
from django.views.decorators.csrf import csrf_exempt
import json

def streaming_view(view_func):
    
    """
    Decorator that converts a Django view function into a streaming response view.
    The view function needs to accept a 'prompt' and return a generator yielding data chunks.
    """
    def wrapper(request, *args, **kwargs):
        # Extract prompt from the request
        if request.method == 'POST':
            data = json.loads(request.body)
            prompt = data.get('prompt')
        elif request.method == 'GET':
            prompt = request.GET.get('prompt')
        else:
            return JsonResponse({"error": "Invalid request method"}, status=405)

        if not prompt:
            return JsonResponse({"error": "Prompt is required"}, status=400)

        # Set the OpenAI API key

        # Call the original view function, which should return a generator
        response_generator = view_func(prompt)

        # Return a StreamingHttpResponse using the generator
        return StreamingHttpResponse(response_generator, content_type='text/event-stream')

    return wrapper