128 lines
5.0 KiB
Python
128 lines
5.0 KiB
Python
from django.shortcuts import render, redirect, get_object_or_404
|
|
from django.http import JsonResponse
|
|
from django.views.decorators.http import require_POST
|
|
import json
|
|
from .models import Article, TodoItem, Conversation, Message
|
|
from .forms import TodoItemForm
|
|
import time
|
|
from ai.local_ai_api import LocalAIApi
|
|
|
|
def index(request):
|
|
if request.method == 'POST':
|
|
form = TodoItemForm(request.POST)
|
|
if form.is_valid():
|
|
form.save()
|
|
return redirect('index')
|
|
else:
|
|
form = TodoItemForm()
|
|
|
|
todo_list = TodoItem.objects.all().order_by('-created_at')
|
|
articles = Article.objects.all()
|
|
|
|
context = {
|
|
'articles': articles,
|
|
'todo_list': todo_list,
|
|
'form': form,
|
|
'timestamp': int(time.time()),
|
|
}
|
|
return render(request, "core/index.html", context)
|
|
|
|
def kanban_board(request):
|
|
tasks = TodoItem.objects.all().order_by('created_at')
|
|
tasks_by_status = {
|
|
status_value: list(filter(lambda t: t.status == status_value, tasks))
|
|
for status_value, status_display in TodoItem.STATUS_CHOICES
|
|
}
|
|
|
|
context = {
|
|
'tasks_by_status': tasks_by_status,
|
|
'status_choices': TodoItem.STATUS_CHOICES,
|
|
'timestamp': int(time.time()),
|
|
}
|
|
return render(request, "core/kanban.html", context)
|
|
|
|
def article_detail(request, article_id):
|
|
article = Article.objects.get(pk=article_id)
|
|
return render(request, "core/article_detail.html", {"article": article})
|
|
|
|
@require_POST
|
|
def update_task_status(request):
|
|
try:
|
|
data = json.loads(request.body)
|
|
task_id = data.get('task_id')
|
|
new_status = data.get('new_status')
|
|
|
|
task = get_object_or_404(TodoItem, id=task_id)
|
|
task.status = new_status
|
|
task.save()
|
|
|
|
return JsonResponse({'success': True})
|
|
except (json.JSONDecodeError, TypeError, ValueError) as e:
|
|
return JsonResponse({'success': False, 'error': str(e)}, status=400)
|
|
|
|
def chat_view(request, conversation_id=None):
|
|
if request.method == 'POST':
|
|
if 'title' in request.POST:
|
|
title = request.POST.get('title', 'New Conversation')
|
|
conversation = Conversation.objects.create(title=title)
|
|
return redirect('core:chat_detail', conversation_id=conversation.id)
|
|
|
|
elif 'text' in request.POST and conversation_id:
|
|
text = request.POST.get('text')
|
|
selected_conversation = get_object_or_404(Conversation, id=conversation_id)
|
|
if text:
|
|
Message.objects.create(conversation=selected_conversation, content=text, is_from_user=True)
|
|
|
|
# Construct the conversation history for the AI
|
|
history = []
|
|
for msg in selected_conversation.messages.order_by('created_at'):
|
|
role = "user" if msg.is_from_user else "assistant"
|
|
history.append({"role": role, "content": msg.content})
|
|
|
|
try:
|
|
system_message = {
|
|
"role": "system",
|
|
"content": "You are a helpful assistant for a project management application. Your purpose is to assist users with their tasks and provide information about the application. The application manages articles and a to-do list."
|
|
}
|
|
|
|
tasks = TodoItem.objects.all().order_by('created_at')
|
|
task_list_str = "\n".join([
|
|
f"- {task.title} (Status: {task.get_status_display()}, Tags: {task.tags or 'None'})"
|
|
for task in tasks
|
|
])
|
|
|
|
tasks_context = {
|
|
"role": "system",
|
|
"content": f"Here is the current list of tasks:\n{task_list_str}"
|
|
}
|
|
|
|
response = LocalAIApi.create_response({
|
|
"input": [
|
|
system_message,
|
|
tasks_context,
|
|
*history
|
|
]
|
|
})
|
|
ai_text = LocalAIApi.extract_text(response)
|
|
if not ai_text:
|
|
ai_text = "I couldn't process that. Please try again."
|
|
except Exception as e:
|
|
ai_text = f"An error occurred: {str(e)}"
|
|
|
|
Message.objects.create(conversation=selected_conversation, content=ai_text, is_from_user=False)
|
|
|
|
# Re-fetch the conversation to include the new messages for rendering
|
|
selected_conversation = get_object_or_404(Conversation, id=conversation_id)
|
|
|
|
conversations = Conversation.objects.order_by('-created_at')
|
|
if conversation_id:
|
|
selected_conversation = get_object_or_404(Conversation, id=conversation_id)
|
|
else:
|
|
selected_conversation = None
|
|
|
|
return render(request, 'core/chat.html', {
|
|
'conversation_list': conversations,
|
|
'selected_conversation': selected_conversation,
|
|
'timestamp': int(time.time()),
|
|
})
|