Django 프로젝트로 챗봇 기능 만들기 - Part 2

Django 프로젝트에서 챗봇 기능을 구현하는 방법 2번째 글입니다다. 템플릿 작성, 회원가입 및 로그인 기능, 게시판 기능, 그리고 OpenAI API를 활용한 챗봇 기능을 포함합니다.

Posted Jan 24, 2025 Updated Feb 12, 2025

By mminzy22

views 10 min read

이번 글에서는 Django 프로젝트에서 템플릿과 뷰를 구성하는 방법을 다룹니다.

1. `base.html` 작성하기

먼저 프로젝트에서 사용할 공통 템플릿을 작성합니다.

`base.html` 템플릿 설정

  
TEMPLATES = [
    {
        'BACKEND': 'django.template.backends.django.DjangoTemplates',
        'DIRS': [ BASE_DIR / 'templates' ],
        'APP_DIRS': True,
        'OPTIONS': {
            'context_processors': [
                'django.template.context_processors.debug',
                'django.template.context_processors.request',
                'django.contrib.auth.context_processors.auth',
                'django.contrib.messages.context_processors.messages',
            ],
        },
    },
]

`base.html` 코드

  
<!DOCTYPE html>
<html lang="en">
  <head>
    <meta charset="UTF-8" />
    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
    <title>Django 프로젝트</title>
  </head>
  <body>

    <header>
      {% block header %}
      <h2>Django 프로젝트</h2>
      <nav>
        <a href="{% url 'index' %}">Home</a>
        <a href="{% url 'chatbot:chat' %}">챗봇</a>

        {% if user.is_authenticated %}
        <!-- 로그인한 사용자에게만 표시 -->
        <form method="post" action="{% url 'accounts:logout' %}" style="display: inline;">
          {% csrf_token %}
          <button type="submit">Logout</button>
        </form>
        {% else %}
        <!-- 비로그인 사용자에게만 표시 -->
        <a href="{% url 'accounts:signup' %}">회원가입</a>
        <a href="{% url 'accounts:login' %}">로그인</a>
        {% endif %}
        <a href="{% url 'posts:post_list' %}">게시판</a>
      </nav>
      {% endblock %}
    </header>

    <main>
      {% block content %}Default Content{% endblock %}
    </main>

    <footer>
      {% block footer %}Footer 입니다.{% endblock %}
    </footer>

  </body>
</html>

프로젝트 홈으로 사용할 `index.html` 생성합니다.

{% extends "base.html" %}

{% block content %}
<h1>Home</h1>
{% endblock %}

2. 회원가입, 로그인, 로그아웃 템플릿 작성

회원가입 뷰 및 템플릿

뷰 설정

  
from django.shortcuts import render
from django.views.generic.edit import CreateView
from django.contrib.auth.views import LoginView, LogoutView
from .forms import CustomUserForm

class SignupView(CreateView):
    template_name = 'accounts/signup.html'
    form_class = CustomUserForm
    success_url = '/accounts/login/'

`signup.html` 템플릿

  
{% extends "base.html" %}

{% block content %}
<h2>Sign Up</h2>
<form method="post">
    {% csrf_token %}
    {{ form.as_p }}
    <button type="submit">Sign Up</button>
</form>
{% endblock %}

로그인, 로그아웃 뷰 및 템플릿

뷰 설정

  
class LoginView(LoginView):
    template_name = 'accounts/login.html'

class LogoutView(LogoutView):
    pass

`login.html` 템플릿

  
{% extends "base.html" %}

{% block content %}
<h2>Login</h2>
<form method="post">
    {% csrf_token %}
    {{ form.as_p }}
    <button type="submit">Login</button>
</form>
{% endblock %}

3. 게시판 기능 구성

게시판의 목록, 작성, 상세 조회, 수정, 삭제, 좋아요 기능을 구성합니다.

게시글 `views.py` 작성

  
from django.shortcuts import get_object_or_404, redirect
from django.http import HttpResponseRedirect
from django.urls import reverse_lazy, reverse
from django.views.generic import ListView, DetailView, CreateView, UpdateView, DeleteView, View
from django.contrib.auth.mixins import LoginRequiredMixin
from .models import Post, Comment
from .forms import PostForm, CommentForm


class PostListView(ListView):
    model = Post
    template_name = "posts/post_list.html"
    context_object_name = "posts"
    ordering = ['-created_at']


class PostCreateView(LoginRequiredMixin, CreateView):
    model = Post
    form_class = PostForm
    template_name = "posts/post_form.html"
    success_url = reverse_lazy('posts:post_list')

    def form_valid(self, form):
        form.instance.author = self.request.user
        return super().form_valid(form)


class PostDetailView(DetailView):
    model = Post
    template_name = "posts/post_detail.html"
    context_object_name = "post"

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['comment_form'] = CommentForm()
        return context

    def post(self, request, *args, **kwargs):
        if not request.user.is_authenticated:
            return redirect(f"{reverse('accounts:login')}?next={request.path}")

        self.object = self.get_object()
        form = CommentForm(request.POST)
        if form.is_valid():
            comment = form.save(commit=False)
            comment.post = self.object
            comment.author = request.user
            comment.save()
            return redirect('posts:post_detail', pk=self.object.pk)

        return self.get(request, *args, **kwargs)


class PostUpdateView(LoginRequiredMixin, UpdateView):
    model = Post
    form_class = PostForm
    template_name = "posts/post_form.html"
    success_url = reverse_lazy('posts:post_list')

    def get_queryset(self):
        return Post.objects.filter(author=self.request.user)


class PostDeleteView(LoginRequiredMixin, DeleteView):
    model = Post
    success_url = reverse_lazy('posts:post_list')

    def get_queryset(self):
        return Post.objects.filter(author=self.request.user)
    
    def form_valid(self, form):
        self.object.delete()
        return HttpResponseRedirect(self.success_url)


class PostLikeToggleView(View):
    def post(self, request, pk):
        post = get_object_or_404(Post, pk=pk)
        
        if not request.user.is_authenticated:
            login_url = f"{reverse_lazy('accounts:login')}?next={reverse_lazy('posts:post_detail', kwargs={'pk': pk})}"
            return redirect(login_url)

        if request.user in post.likes.all():
            post.likes.remove(request.user)
        else:
            post.likes.add(request.user)
        
        return redirect('posts:post_detail', pk=pk)


class CommentUpdateView(LoginRequiredMixin, UpdateView):
    model = Comment
    form_class = CommentForm
    template_name = "posts/post_detail.html"

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['post'] = self.object.post
        return context

    def form_valid(self, form):
        self.object = form.save()
        return redirect('posts:post_detail', pk=self.object.post.pk)

    def get_queryset(self):
        return Comment.objects.filter(author=self.request.user)


class CommentDeleteView(LoginRequiredMixin, DeleteView):
    model = Comment

    def get_queryset(self):
        return Comment.objects.filter(author=self.request.user)

    def get_success_url(self):
        try:
            post = self.object.post
            return reverse_lazy('posts:post_detail', kwargs={'pk': post.pk})
        except AttributeError:
            return reverse_lazy('posts:post_list')

    def get(self, request, *args, **kwargs):
        return self.post(request, *args, **kwargs)

게시글 목록 템플릿

  
{% extends "base.html" %}


{% block content %}
<h2>게시글 목록</h2>

<!-- 새 게시글 작성 버튼 -->
<a href="{% url 'posts:post_create' %}" class="btn btn-primary">새 게시글 작성</a>

<!-- 게시글 목록 -->
<table border="1">
    <thead>
        <tr>
            <th>번호</th>
            <th>제목</th>
            <th>내용</th>
            <th>작성자</th>
            <th>좋아요 수</th>
        </tr>
    </thead>
    <tbody>
        {% for post in posts %}
        <tr>
            <td>{{ forloop.counter }}</td>
            <td><a href="{% url 'posts:post_detail' post.pk %}">{{ post.title }}</a></td>
            <td>{{ post.content|truncatechars:50 }}</td>
            <td>{{ post.author }}</td>
            <td>{{ post.total_likes }}</td>
        </tr>
        {% empty %}
        <tr>
            <td colspan="6" style="text-align: center;">게시글이 없습니다.</td>
        </tr>
        {% endfor %}
    </tbody>
</table>

{% endblock %}

게시글 생성 템플릿

  
{% extends "base.html" %}

{% block content %}
  <h2>글쓰기</h2>
  <form method="post">
    {% csrf_token %}
    {{ form.as_p }}
    <button type="submit">글쓰기</button>
  </form>
{% endblock %}

게시글 상세 페이지 템플릿

  
{% extends "base.html" %}


{% block content %}
<h2>게시글 상세 조회</h2>

<h1>{{ post.title }}</h1>
<p>{{ post.content }}</p>
<p>작성자: {{ post.author }}</p>
<p>좋아요 수: {{ post.total_likes }}</p>

<!-- 좋아요 버튼 -->
<form method="post" action="{% url 'posts:post_like_toggle' post.pk %}">
    {% csrf_token %}
    <button type="submit">
        {% if user in post.likes.all %}
            Unlike
        {% else %}
            Like
        {% endif %}
    </button>
</form>

<!-- 게시글 수정/삭제 버튼 (작성자만 표시) -->
{% if post.author == user %}
    <a href="{% url 'posts:post_update' post.pk %}">게시글 수정</a>
    <form action="{% url 'posts:post_delete' post.pk %}" method="post" style="display:inline;">
        {% csrf_token %}
        <button type="submit" onclick="return confirm('게시글을 정말 삭제하시겠습니까?')">삭제</button>
    </form>
{% endif %}

<hr>
<h2>댓글</h2>

<form method="post" action="">
    {% csrf_token %}
    {{ comment_form.as_p }}
    <button type="submit">댓글 작성</button>
</form>
<hr>
<ul>
    {% for comment in post.comments.all %}
    <li>
        <p>{{ comment.author }}: {{ comment.content }}</p>
        <p>{{ comment.created_at }}</p>
        {% if comment.author == user %}
            <!-- 댓글 수정/삭제 버튼 -->
            <a href="{% url 'posts:comment_update' comment.pk %}">수정</a>
            <form method="post" action="{% url 'posts:comment_delete' comment.pk %}" style="display:inline;">
                {% csrf_token %}
                <button type="submit" onclick="return confirm('댓글을 정말 삭제하시겠습니까?')">삭제</button>
            </form>

            <!-- 댓글 수정 폼 -->
            {% if request.resolver_match.url_name == 'comment_update' and comment.pk == request.resolver_match.kwargs.pk %}
            <form method="post" action="{% url 'posts:comment_update' comment.pk %}">
                {% csrf_token %}
                {{ form.as_p }}
                <button type="submit">댓글 수정</button>
            </form>
            {% endif %}
        {% endif %}
    </li>
    {% endfor %}
</ul>

{% endblock %}

4. 챗봇 기능 구현

OpenAI API를 활용해 챗봇 기능을 구현합니다.

챗봇 Open API Key

루트 폴더에 .env 파일 생성:
1 touch .env

.env 파일 내용:

  
OPENAI_API_KEY="오픈 API 키 입력하기"

.env 파일은 민감한 정보를 담고 있으므로 .gitignore에 반드시 추가.

챗봇 관련 유틸 함수

  
import openai
import os
from dotenv import load_dotenv

load_dotenv()

openai.api_key = os.getenv("OPENAI_API_KEY")

def get_ai_response(user_input, messages):

    messages.append({"role": "user", "content": user_input})

    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=messages
    )

    ai_response = response['choices'][0]['message']['content']
    messages.append({"role": "assistant", "content": ai_response})

    return ai_response, messages

챗봇 뷰

  
from django.contrib.auth.mixins import LoginRequiredMixin
from django.views.generic import FormView
from django.shortcuts import render
from .models import ChatMessage
from .forms import ChatMessageForm
from .utils import get_ai_response


class ChatView(LoginRequiredMixin, FormView):
    template_name = 'chatbot/chat.html'
    form_class = ChatMessageForm

    def form_valid(self, form):
        user_input = self.request.POST.get('user_input')
        prompt = "내가 하는 말이랑 최대한 연관지어서 오늘 하루 힘낼 수 있는 명언해줘"

        ai_response, _ = get_ai_response(user_input, [{"role": "system", "content": prompt}])

        # db 저장
        ChatMessage.objects.create(
            user_input=self.request.user,
            user_text=user_input,
            bot_response=ai_response
        )

        return render(self.request, self.template_name, {
            "form": form,
            "ai_response": ai_response,
            "user_input": user_input,
        })

`chat.html` 템플릿

  
{% extends "base.html" %}

{% block content %}
<h1>Chatbot</h1>
<form method="post">
    {% csrf_token %}
    <label for="user_input">대화 입력:</label>
    <input type="text" id="user_input" name="user_input" required>
    <button type="submit">전송</button>
</form>
<hr>
{% if ai_response %}
    <p><strong>사용자:</strong> {{ user_input }}</p>
    <p><strong>AI:</strong> {{ ai_response }}</p>
{% endif %}
{% endblock %}

Django, LLM

This post is licensed under CC BY 4.0 by the author.

1. base.html 작성하기

base.html 템플릿 설정

base.html 코드

프로젝트 홈으로 사용할 index.html 생성합니다.

2. 회원가입, 로그인, 로그아웃 템플릿 작성

회원가입 뷰 및 템플릿

뷰 설정

signup.html 템플릿

로그인, 로그아웃 뷰 및 템플릿

뷰 설정

login.html 템플릿

3. 게시판 기능 구성

게시글 views.py 작성

게시글 목록 템플릿

게시글 생성 템플릿

게시글 상세 페이지 템플릿

4. 챗봇 기능 구현

챗봇 Open API Key

챗봇 관련 유틸 함수

챗봇 뷰

chat.html 템플릿

Trending Tags

1. `base.html` 작성하기

`base.html` 템플릿 설정

`base.html` 코드

프로젝트 홈으로 사용할 `index.html` 생성합니다.

`signup.html` 템플릿

`login.html` 템플릿

게시글 `views.py` 작성

`chat.html` 템플릿