mobees-ai/chatbot/views.py

from pickle import decode_long
from django.shortcuts import render
from rest_framework.views import APIView, Response
from django.http import HttpResponse
from chatbot.main import instance as Chatbot

class ChatbotView (APIView):

    # def get (self, request):
    #     status, data = InsightsBdo().list(request)

    #     return Response(status=status, data=data)

    def post (self, request):
        # get anwser from a question
        question = request.data.get('question')

        status, data = Chatbot.chat(question)

        return Response(status=status, data=data)
    
    def put (self, request):
        # train chatbot with docs or chat instruction
        chat = request.data.get('chat', None)

        status, data = Chatbot.train(chat)

        return Response(status=status, data=data)
    
    # def delete (self, request, id=None):
    #     # delete insight
        
    #     status, data = InsightsBdo().delete(request, id)

    #     return Response(status=status, data=data)