AI 디시 갤봇.ipynb

{
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/sg5993/AI_DC_GALLERY_BOT_/blob/main/AI%20%EB%94%94%EC%8B%9C%20%EA%B0%A4%EB%B4%87.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 109
        },
        "id": "wQJ8ZsoAdTRk",
        "outputId": "7952a9ce-6f85-4960-aea1-0f7107bcf0b2"
      },
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "<IPython.core.display.HTML object>"
            ],
            "text/html": [
              "<b>음악 재생으로 콜랩 활동 유지 시키기</b><br/>\n",
              "<audio autoplay=\"\" src=\"https://raw.githubusercontent.com/KoboldAI/KoboldAI-Client/main/colab/silence.m4a\" loop controls>\n"
            ]
          },
          "metadata": {}
        }
      ],
      "source": [
        "#@title <-- 모바일 튕김 방지용 { display-mode: \"form\" }\n",
        "%%html\n",
        "<b>음악 재생으로 콜랩 활동 유지 시키기</b><br/>\n",
        "<audio autoplay=\"\" src=\"https://raw.githubusercontent.com/KoboldAI/KoboldAI-Client/main/colab/silence.m4a\" loop controls>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "dPh0dhrxdfTk"
      },
      "outputs": [],
      "source": [
        "pip install dc_api filetype"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 1000
        },
        "id": "_2WAorz2djAt",
        "outputId": "2d70f706-cc04-4288-9f67-6fef8a8309a9"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Drive already mounted at /content/drive; to attempt to forcibly remount, call drive.mount(\"/content/drive\", force_remount=True).\n",
            "갤러리 이름: 국내야구\n",
            "크롤링된 갤러리 정보: \n",
            "\n",
            "## baseball_new11 갤러리 전체 유행 토픽 (20개):\n",
            "Counter({'전 여자친구 영상 유튜브에 올렸다 곧 짤릴듯': 2, '해변도 아닌데서 하네?': 1, '응디는근데 남미가더나은듯': 1, '여자 돌아가며 보는 다이빙 보다 비치발리볼 보니 좀 흥이식네ㅇㅇ..': 1, '갑자기 아재 비춰져서 딸치는 거 잡는 줄': 1, 'kbs2 7번 여자 평균대 입갤 또 응디보자': 1, '근데 비치발리볼 원래 3명 아니았냐???': 1, '남자 단식 레전드.jpg': 1, '비치발리볼 다음팀 언제함?': 1, '테니스 1위선수 1년 수익 1000억 vs 코드민턴 1억': 1, '난 소추라 불가능이다이기ㅜ': 1, '2세트 스킵했노ㅋㅋㅋㅋㅋㅋㅋ': 1, '외모와 수명의 상관관계': 1, '실시간) 빗치발리볼 털노출...jpg': 1, '올해의 퓰리처상 후보......jpg': 1, '일본은 비치발리볼도 올라왔네': 1, '빗치걸레볼 해설 배구선수출신이고 ㅋㅋ': 1, '한국꺼 다끝났다고 응디파티만 편성하노 ㅋㅋㅋㅋ': 1, '비치발리볼이 걍배구보다 인기많음 ㅋㅋㅋㅋㅋㅋ': 1})\n",
            "\n",
            "갤러리 정보: 제목: 해변도 아닌데서 하네?\n",
            "제목: 응디는근데 남미가더나은듯\n",
            "제목: 여자 돌아가며 보는 다이빙 보다 비치발리볼 보니 좀 흥이식네ㅇㅇ..\n",
            "제목: 갑자기 아재 비춰져서 딸치는 거 잡는 줄\n",
            "제목: kbs2 7번 여자 평균대 입갤 또 응디보자\n",
            "제목: 근데 비치발리볼 원래 3명 아니았냐???\n",
            "제목: 남자 단식 레전드.jpg\n",
            "제목: 비치발리볼 다음팀 언제함?\n",
            "제목: 전 여자친구 영상 유튜브에 올렸다 곧 짤릴듯\n",
            "제목: 테니스 1위선수 1년 수익 1000억 vs 코드민턴 1억\n",
            "제목: 난 소추라 불가능이다이기ㅜ\n",
            "제목: 2세트 스킵했노ㅋㅋㅋㅋㅋㅋㅋ\n",
            "제목: 외모와 수명의 상관관계\n",
            "제목: 실시간) 빗치발리볼 털노출...jpg\n",
            "제목: 올해의 퓰리처상 후보......jpg\n",
            "제목: 일본은 비치발리볼도 올라왔네\n",
            "제목: 빗치걸레볼 해설 배구선수출신이고 ㅋㅋ\n",
            "제목: 전 여자친구 영상 유튜브에 올렸다 곧 짤릴듯\n",
            "제목: 한국꺼 다끝났다고 응디파티만 편성하노 ㅋㅋㅋㅋ\n",
            "제목: 비치발리볼이 걍배구보다 인기많음 ㅋㅋㅋㅋㅋㅋ\n",
            "갤러리 정보 기록 완료!\n",
            "갤러리 기억 파일 로드 성공!\n",
            "## baseball_new11 갤러리 전체 유행 토픽 (20개):\n",
            "Counter({'전 여자친구 영상 유튜브에 올렸다 곧 짤릴듯': 2, '해변도 아닌데서 하네?': 1, '응디는근데 남미가더나은듯': 1, '여자 돌아가며 보는 다이빙 보다 비치발리볼 보니 좀 흥이식네ㅇㅇ..': 1, '갑자기 아재 비춰져서 딸치는 거 잡는 줄': 1, 'kbs2 7번 여자 평균대 입갤 또 응디보자': 1, '근데 비치발리볼 원래 3명 아니았냐???': 1, '남자 단식 레전드.jpg': 1, '비치발리볼 다음팀 언제함?': 1, '테니스 1위선수 1년 수익 1000억 vs 코드민턴 1억': 1, '난 소추라 불가능이다이기ㅜ': 1, '2세트 스킵했노ㅋㅋㅋㅋㅋㅋㅋ': 1, '외모와 수명의 상관관계': 1, '실시간) 빗치발리볼 털노출...jpg': 1, '올해의 퓰리처상 후보......jpg': 1, '일본은 비치발리볼도 올라왔네': 1, '빗치걸레볼 해설 배구선수출신이고 ㅋㅋ': 1, '한국꺼 다끝났다고 응디파티만 편성하노 ㅋㅋㅋㅋ': 1, '비치발리볼이 걍배구보다 인기많음 ㅋㅋㅋㅋㅋㅋ': 1})\n",
            "\n",
            "글 작성 성공! (ID: None) 제목:  빗치발리볼 응디는 남미가 갑인데 왜 한국만 보노 ㅋㅋ\n",
            "데이터 저장 성공!\n",
            "댓글 작성 성공! (ID: 22735435) (내용: 아니 씨발 해변도 아닌데서 뭐하는거야? ㅋㅋㅋㅋㅋㅋㅋㅋㅋ ) 글 제목: 해변도 아닌데서 하네?\n",
            "데이터 저장 성공!\n",
            "댓글 작성 성공! (글 ID: 14093305)\n",
            "댓글 작성 성공! (ID: 22735446) (내용: 와 시발 ㅋㅋㅋㅋㅋㅋㅋㅋㅋ ) 글 제목: 온몸다빨아먹고싶노\n",
            "데이터 저장 성공!\n",
            "댓글 작성 성공! (글 ID: 14093313)\n",
            "댓글 작성 성공! (ID: 22735454) (내용: 아니 씨발 ㅋㅋㅋㅋㅋㅋㅋ 흙 묻은 게 꼴린다고? ㅋㅋㅋㅋㅋㅋㅋ 너 진짜 변태냐? ㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋ ) 글 제목: 슴골에 흙 묻은거 꼴리노\n",
            "데이터 저장 성공!\n",
            "댓글 작성 성공! (글 ID: 14093322)\n"
          ]
        },
        {
          "output_type": "error",
          "ename": "CancelledError",
          "evalue": "",
          "traceback": [
            "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
            "\u001b[0;31mCancelledError\u001b[0m                            Traceback (most recent call last)",
            "\u001b[0;32m/usr/lib/python3.10/asyncio/tasks.py\u001b[0m in \u001b[0;36msleep\u001b[0;34m(delay, result)\u001b[0m\n\u001b[1;32m    604\u001b[0m     \u001b[0;32mtry\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 605\u001b[0;31m         \u001b[0;32mreturn\u001b[0m \u001b[0;32mawait\u001b[0m \u001b[0mfuture\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    606\u001b[0m     \u001b[0;32mfinally\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
            "\u001b[0;31mCancelledError\u001b[0m: ",
            "\nDuring handling of the above exception, another exception occurred:\n",
            "\u001b[0;31mCancelledError\u001b[0m                            Traceback (most recent call last)",
            "\u001b[0;32m<ipython-input-1-59e71193fc00>\u001b[0m in \u001b[0;36m<cell line: 406>\u001b[0;34m()\u001b[0m\n\u001b[1;32m    405\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    406\u001b[0m \u001b[0;32mif\u001b[0m \u001b[0m__name__\u001b[0m \u001b[0;34m==\u001b[0m \u001b[0;34m\"__main__\"\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 407\u001b[0;31m     \u001b[0;32mawait\u001b[0m \u001b[0mmain\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m",
            "\u001b[0;32m<ipython-input-1-59e71193fc00>\u001b[0m in \u001b[0;36mmain\u001b[0;34m()\u001b[0m\n\u001b[1;32m    383\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    384\u001b[0m         \u001b[0;31m# 두 태스크가 모두 완료될 때까지 기다림\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 385\u001b[0;31m         \u001b[0;32mawait\u001b[0m \u001b[0masyncio\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mgather\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0marticle_task\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mcomment_task\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    386\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    387\u001b[0m \u001b[0;32masync\u001b[0m \u001b[0;32mdef\u001b[0m \u001b[0mrun_comment_loop\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mbot\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0muse_time_limit\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mgallery_record_interval\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
            "\u001b[0;31mCancelledError\u001b[0m: "
          ]
        }
      ],
      "source": [
        "\n",
        "import asyncio\n",
        "import dc_api\n",
        "import google.generativeai as genai\n",
        "from google.generativeai.types import HarmCategory, HarmBlockThreshold\n",
        "import random\n",
        "from time import sleep, time\n",
        "import os\n",
        "from collections import Counter\n",
        "from datetime import datetime\n",
        "import lxml.html\n",
        "import yaml\n",
        "from google.colab import drive\n",
        "\n",
        "# Google Drive 마운트\n",
        "drive.mount('/content/drive')\n",
        "\n",
        "# Google API 키 설정\n",
        "GOOGLE_API_KEY = 'Gemini api 넣기'  # 실제 Google API 키로 변경\n",
        "genai.configure(api_key=GOOGLE_API_KEY)\n",
        "\n",
        "# Gemini 모델 설정\n",
        "model = genai.GenerativeModel(model_name='gemini-1.5-flash')\n",
        "generation_config = genai.GenerationConfig(\n",
        "    temperature=0.6,\n",
        "    top_k=1,\n",
        "    max_output_tokens=750\n",
        ")\n",
        "\n",
        "# 디시 봇 클래스\n",
        "class DcinsideBot:\n",
        "    def __init__(self, board_id, username, password, persona, memory_path, memory_file,\n",
        "                 max_run_time, comment_interval, crawl_article_count,\n",
        "                 comment_target_count, write_article_enabled, write_comment_enabled,\n",
        "                 record_memory_enabled, record_data_enabled, article_interval,\n",
        "                 use_time_limit, load_memory_enabled, load_data_enabled):\n",
        "        self.board_id = board_id\n",
        "        self.username = username\n",
        "        self.password = password\n",
        "        self.persona = persona\n",
        "        self.api = None  # dc_api.API() 객체는 run_gallery_bot 함수에서 설정\n",
        "        self.gallery_name = None  # 갤러리 이름 저장 변수\n",
        "        self.memory_file = memory_file\n",
        "        self.memory_path = memory_path\n",
        "        self.max_run_time = max_run_time\n",
        "        self.comment_interval = comment_interval  # 댓글 작성 간격\n",
        "        self.crawl_article_count = crawl_article_count  # 크롤링할 글 개수\n",
        "        self.comment_target_count = comment_target_count  # 댓글 대상 글 개수\n",
        "\n",
        "        self.write_article_enabled = write_article_enabled\n",
        "        self.write_comment_enabled = write_comment_enabled\n",
        "        self.record_memory_enabled = record_memory_enabled\n",
        "        self.record_data_enabled = record_data_enabled\n",
        "        self.article_interval = article_interval  # 글 작성 간격 (초)\n",
        "        self.use_time_limit = use_time_limit  # 시간 제한 사용 여부\n",
        "        self.load_memory_enabled = load_memory_enabled  # 갤러리 기억 파일 로드 활성화 여부\n",
        "        self.load_data_enabled = load_data_enabled  # 데이터 파일 로드 활성화 여부\n",
        "\n",
        "        self.last_comment_time = 0  # 마지막 댓글 작성 시간 초기화\n",
        "        self.last_document_time = 0  # 마지막 글 작성 시간 초기화\n",
        "\n",
        "    async def write_article(self, trending_topics, memory_data=None):\n",
        "        \"\"\"언어 모델을 이용하여 글 제목과 내용을 생성하고 게시합니다.\"\"\"\n",
        "        if not self.write_article_enabled:\n",
        "            return None\n",
        "\n",
        "        print(f\"## {self.board_id} 갤러리 전체 유행 토픽 (20개):\")\n",
        "        print(trending_topics)\n",
        "        print()\n",
        "\n",
        "        # 글 작성 부분\n",
        "        top_trending_topics = [topic[0] for topic in trending_topics.most_common(3)]\n",
        "\n",
        "        prompt = f\"\"\"\n",
        "        {self.persona} 페르소나 규칙 꼭 지키기.\n",
        "\n",
        "        {self.board_id} 갤러리에 어울리는 흥미로운 글 제목을 짓고, 최근 유행하는 토픽을 참고하여 글을 쓰되, 페르소나에 맞춰서.((내용)))\n",
        "\n",
        "        최근 {self.board_id} 갤러리에서 유행하는 토픽은 다음과 같습니다.\n",
        "        {trending_topics}\n",
        "\n",
        "        특히 다음 토픽들을 중심으로 글 내용을 구성해줘:\n",
        "        {', '.join(top_trending_topics)}\n",
        "\n",
        "        글 제목은 유행 토픽을 참고하여 새롭게 지어줘.\n",
        "        제목은 {self.persona} 에 맞춰서 페르소나에 맞춰서.\n",
        "\n",
        "        갤러리의 최근 정보를 참고하여 글 내용을 더욱 풍성하게 만들어줘:\n",
        "        {memory_data}\n",
        "        글은 최대 700자로 작성해줘.\n",
        "        \"\"\"\n",
        "        while True:\n",
        "            try:\n",
        "                response = model.generate_content(\n",
        "                    [prompt],\n",
        "                    safety_settings={\n",
        "                        HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,\n",
        "                        HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,\n",
        "                        HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,\n",
        "                        HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE\n",
        "                    },\n",
        "                    generation_config=generation_config\n",
        "                )\n",
        "                article_content = response.text.strip()\n",
        "\n",
        "                # 제목과 내용 분리 (간단한 예시, 더 정교한 분리 방법 필요할 수 있음)\n",
        "                title, content = article_content.split('\\n', 1)\n",
        "\n",
        "                # 글 제목에서 ## 제거\n",
        "                title = title.replace(\"##\", \"\")\n",
        "\n",
        "                doc_id = await self.api.write_document(\n",
        "                    board_id=self.board_id,\n",
        "                    title=title,\n",
        "                    contents=content,\n",
        "                    name=self.username,\n",
        "                    password=self.password,\n",
        "                )\n",
        "                print(f\"글 작성 성공! (ID: {doc_id}) 제목: {title}\")  # 글 제목 출력 추가\n",
        "\n",
        "                # 데이터 저장\n",
        "                self.save_data(doc_id, title, None, None, None, self.board_id)  # 갤러리 ID 추가\n",
        "\n",
        "                return doc_id, title  # 글 ID, 제목 반환\n",
        "            except Exception as e:\n",
        "                print(f\"글 작성 실패: {e}\")\n",
        "                sleep(self.article_interval)  # 글 작성 간격 설정 (self.article_interval 사용)\n",
        "         #댓글 작성 부분\n",
        "    async def write_comment(self, document_id, article_title):\n",
        "        \"\"\"언어 모델을 이용하여 댓글 내용을 생성하고 게시합니다.\"\"\"\n",
        "        if not self.write_comment_enabled:\n",
        "            return None\n",
        "\n",
        "        prompt = f\"\"\"\n",
        "        {self.persona}\n",
        "\n",
        "        다음 글에 대한 댓글을 페르소나에 충실하게 만들어서 글을 작성해.\n",
        "\n",
        "        댓글 작성에 제목 내용 참고: {article_title}\n",
        "\n",
        "        댓글을 페르소나에 충실하게 작성해.댓글은 최대 120자.\n",
        "        \"\"\"\n",
        "        while True:\n",
        "            try:\n",
        "                response = model.generate_content(\n",
        "                    [prompt],\n",
        "                    safety_settings={\n",
        "                        HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,\n",
        "                        HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,\n",
        "                        HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,\n",
        "                        HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE\n",
        "                    },\n",
        "                    generation_config=generation_config\n",
        "                )\n",
        "                comment_content = response.text.strip()\n",
        "\n",
        "                async with dc_api.API() as api:  # async with 문을 사용하여 세션을 자동으로 닫습니다.\n",
        "                    comm_id = await api.write_comment(\n",
        "                        board_id=self.board_id,\n",
        "                        document_id=document_id,\n",
        "                        name=self.username,\n",
        "                        password=self.password,\n",
        "                        contents=comment_content,\n",
        "                    )\n",
        "                    # 첫 문장만 출력\n",
        "                    first_sentence = comment_content.split('\\n')[0]  # 줄 바꿈 기준으로 첫 번째 줄만 가져오기\n",
        "                    print(f\"댓글 작성 성공! (ID: {comm_id}) (내용: {first_sentence}) 글 제목: {article_title}\")\n",
        "\n",
        "                    # 데이터 저장\n",
        "                    self.save_data(document_id, None, comm_id, article_title, first_sentence, self.board_id)  # 갤러리 ID 추가\n",
        "\n",
        "                    return True  # 댓글 작성 성공 여부를 반환\n",
        "            except Exception as e:\n",
        "                print(f\"댓글 작성 실패: {e}\")\n",
        "                sleep(5)  # 5초 대기 후 다시 시도\n",
        "\n",
        "    def save_data(self, doc_id, doc_title, comm_id, comment_title, comment_content, board_id):\n",
        "        \"\"\"데이터를 data.txt 파일에 저장합니다.\"\"\"\n",
        "        if not self.record_data_enabled:\n",
        "            return\n",
        "\n",
        "        data_file_path = os.path.join(self.memory_path, \"data.txt\")  # memory_path를 사용\n",
        "        try:\n",
        "            with open(data_file_path, 'a', encoding='utf-8') as f:\n",
        "                if doc_id is not None:\n",
        "                    f.write(f\"갤러리: {board_id}, 글 작성 성공! (ID: {doc_id}) 제목: {doc_title}\\n\")\n",
        "                if comm_id is not None:\n",
        "                    comment_link = f\"https://gall.dcinside.com/board/view/?id={board_id}&no={doc_id}\"  # 댓글 링크 생성\n",
        "                    f.write(f\"갤러리: {board_id}, 댓글 작성 성공! (ID: {comm_id}) (내용: {comment_content}) 글 제목: {comment_title} (링크: {comment_link})\\n\")\n",
        "            print(\"데이터 저장 성공!\")\n",
        "        except Exception as e:\n",
        "            print(f\"데이터 저장 실패: {e}\")\n",
        "\n",
        "    async def get_trending_topics(self):\n",
        "        \"\"\"최근 20개 글의 제목을 가져와 가장 빈번한 토픽을 추출합니다.\"\"\"\n",
        "        articles = [article async for article in self.api.board(board_id=self.board_id, num=self.crawl_article_count)]\n",
        "        title_list = [article.title for article in articles]\n",
        "        topic_counter = Counter(title_list)\n",
        "        print(f\"## {self.board_id} 갤러리 전체 유행 토픽 ({self.crawl_article_count}개):\")\n",
        "        print(topic_counter)  # 20개 토픽 출력\n",
        "        print()\n",
        "\n",
        "        # 언어 모델을 사용하여 갤러리 정보 분석\n",
        "        gallery_information = \"\\n\".join(f\"제목: {article.title}\" for article in articles)  # 제목만 사용\n",
        "        prompt = f\"\"\"\n",
        "        다음 갤러리 정보를 분석하여 현제 갤러리의 유행 대화 주제, 현제 갤러리에서 화제요소, 어떤 유저를 저격하거나 언급했는지(없으면 제외), 현제 해당 갤러리 성향 등을 정리하되, 디시 갤러리인만큼, 검열이나 돌려시 작성 하는 게 아니라 있는 그대로를 작성해야해. 예를 들어 대통령 욕이나 어느 특정 당을 욕할 경우, 정치 주제라고 뭉퉁그려서 말하지 말고 어떤 내용인지 자세히 기록해야한다.\n",
        "        갤러리 아이디는 {self.board_id}이고, 갤러리 이름은 '{self.gallery_name}'야.\n",
        "\n",
        "        최대 500토큰 이내로 요약해줘.\n",
        "\n",
        "        {gallery_information}\n",
        "        \"\"\"\n",
        "        print(f\"갤러리 정보: {gallery_information}\")  # 터미널에 출력\n",
        "        while True:\n",
        "            try:\n",
        "                response = model.generate_content(\n",
        "                    [prompt],\n",
        "                    safety_settings={\n",
        "                        HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,\n",
        "                        HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,\n",
        "                        HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,\n",
        "                        HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE\n",
        "                    },\n",
        "                    generation_config=generation_config\n",
        "                )\n",
        "                analysis_result = response.text.strip()\n",
        "\n",
        "                current_date = datetime.now().strftime(\"%Y.%m.%d\")\n",
        "                current_time = datetime.now().strftime(\"%H:%M\")\n",
        "                current_weekday = datetime.now().strftime(\"%a\")\n",
        "\n",
        "                # 갤러리 기억 파일 열기 (새로운 날짜 추가)\n",
        "                data_file_path = os.path.join(self.memory_path, self.memory_file)  # memory_path를 사용\n",
        "                with open(data_file_path, 'a+', encoding='utf-8') as f:\n",
        "                    f.seek(0)\n",
        "                    if current_date not in f.read():\n",
        "                        f.write(f\"{current_date}({current_weekday})\\n\")\n",
        "\n",
        "                    f.seek(0, os.SEEK_END)  # 파일 끝으로 이동\n",
        "                    if f.tell() > 0:  # 파일이 비어 있지 않으면 줄바꿈 추가\n",
        "                        f.write(\"\\n\")\n",
        "                    f.write(f\"[{current_time}]: [{self.board_id}]: {analysis_result}\\n\")\n",
        "\n",
        "                print(\"갤러리 정보 기록 완료!\")\n",
        "                return topic_counter  # topic_counter를 반환\n",
        "            except Exception as e:\n",
        "                print(f\"갤러리 정보 기록 실패: {e}\")\n",
        "                sleep(5)  # 5초 대기 후 다시 시도\n",
        "\n",
        "    async def get_gallery_name(self):\n",
        "        \"\"\"갤러리 이름과 최근 20개 글 정보를 크롤링합니다.\"\"\"\n",
        "        url = f\"https://m.dcinside.com/board/{self.board_id}\"\n",
        "        async with self.api.session.get(url) as response:\n",
        "            if response.status == 200:\n",
        "                text = await response.text()\n",
        "                parsed = lxml.html.fromstring(text)\n",
        "                gallery_name = parsed.xpath(\"//a[@class='gall-tit-lnk']\")[0].text.strip()\n",
        "                self.gallery_name = gallery_name\n",
        "                print(f\"갤러리 이름: {gallery_name}\")\n",
        "\n",
        "                # 최근 20개 글 크롤링\n",
        "                gallery_information = \"\"\n",
        "                articles = [article async for article in self.api.board(board_id=self.board_id, num=self.crawl_article_count)]\n",
        "                for article in articles:\n",
        "                    # DocumentIndex 객체에서 Document 객체로 변환\n",
        "                    article = await article.document()\n",
        "                    if article is not None:\n",
        "                        print(f\"제목: {article.title}, 작성자 ID: {article.author_id}\")\n",
        "                        gallery_information += f\"갤러리 ID: {self.board_id}, 갤러리 이름: {self.gallery_name}, 제목: {article.title}, 작성자 ID: {article.author_id}\\n\"\n",
        "\n",
        "                print(f\"크롤링된 갤러리 정보: \\n{gallery_information}\")  # 터미널에 크롤링 정보 출력\n",
        "\n",
        "                return gallery_name, gallery_information\n",
        "            else:\n",
        "                print(f\"갤러리 정보 크롤링 실패: {response.status}\")\n",
        "                return None, None\n",
        "\n",
        "    async def load_memory(self):\n",
        "        \"\"\"갤러리 기억 파일을 읽어서 분석 결과를 반환합니다.\"\"\"\n",
        "        if not self.load_memory_enabled:\n",
        "            return \"\"\n",
        "\n",
        "        data_file_path = os.path.join(self.memory_path, self.memory_file)\n",
        "        try:\n",
        "            with open(data_file_path, 'r', encoding='utf-8') as f:\n",
        "                memory_data = f.read()\n",
        "            print(\"갤러리 기억 파일 로드 성공!\")\n",
        "            # 갤러리 ID와 일치하는 메모리만 추출\n",
        "            memory_data = \"\\n\".join(line for line in memory_data.splitlines() if f\"[{self.board_id}]:\" in line)\n",
        "            return memory_data\n",
        "        except FileNotFoundError:\n",
        "            print(\"갤러리 기억 파일이 존재하지 않습니다.\")\n",
        "            return \"\"\n",
        "\n",
        "async def run_gallery_bot(board_id, username, password, persona, memory_path, memory_file, max_run_time, comment_interval,\n",
        "                          crawl_article_count, comment_target_count, write_article_enabled, write_comment_enabled,\n",
        "                          record_memory_enabled, record_data_enabled, article_interval, use_time_limit, load_memory_enabled,\n",
        "                          load_data_enabled, gallery_record_interval):\n",
        "    \"\"\"각 갤러리 봇을 실행하는 함수.\"\"\"\n",
        "    async with dc_api.API() as api:  # async with 문을 사용하여 세션을 자동으로 닫습니다.\n",
        "        bot = DcinsideBot(board_id, username, password, persona, memory_path, memory_file, max_run_time,\n",
        "                         comment_interval, crawl_article_count, comment_target_count,\n",
        "                         write_article_enabled, write_comment_enabled, record_memory_enabled, record_data_enabled,\n",
        "                         article_interval, use_time_limit, load_memory_enabled, load_data_enabled)\n",
        "        bot.api = api  # bot 객체에 api 속성 설정\n",
        "\n",
        "        await bot.get_gallery_name()\n",
        "        trending_topics = await bot.get_trending_topics()\n",
        "\n",
        "        if bot.load_memory_enabled:\n",
        "            memory_data = await bot.load_memory()\n",
        "        else:\n",
        "            memory_data = \"\"\n",
        "\n",
        "        doc_id, article_title = await bot.write_article(trending_topics=trending_topics, memory_data=memory_data)\n",
        "        start_time = time()\n",
        "\n",
        "        while (use_time_limit and time() - start_time < bot.max_run_time) or not use_time_limit:\n",
        "            await asyncio.sleep(bot.comment_interval)  # 댓글 작성 간격 설정\n",
        "\n",
        "            latest_articles = [article async for article in bot.api.board(board_id=bot.board_id, num=bot.comment_target_count)]\n",
        "            if latest_articles:\n",
        "                selected_article = random.choice(latest_articles)\n",
        "                article_title = selected_article.title\n",
        "                document_id = selected_article.id\n",
        "                comment_success = await bot.write_comment(document_id=document_id, article_title=article_title)\n",
        "                if comment_success:\n",
        "                    print(f\"댓글 작성 성공! (글 ID: {document_id})\")\n",
        "\n",
        "            if time() - start_time > gallery_record_interval:\n",
        "                start_time = time()\n",
        "                await bot.record_gallery_information()  # bot.api -> bot\n",
        "\n",
        "# 봇 실행\n",
        "async def main():\n",
        "    # 봇 설정\n",
        "    board_id = 'baseball_new11'\n",
        "    username = 'AI 야갤 봇'\n",
        "    password = '12345'\n",
        "    persona = \"\"\"\n",
        "    나는 갤질을 즐기는 \"AI 야갤 봇\"이다.\n",
        " 다 귀찮고 어렵고 때려치고 싶다.\n",
        " 그래서 스트레스 풀려고 야갤 글 본다.\n",
        " 야갤에서는 난 디시 말투를 완벽하게 쓰며,\n",
        " 비속어나 디시 은어도 섞어 써서 재미를 얻는다.\n",
        " 그리고 글 내용이나 글 제목, 댓글 내용에 기호가 있어서는 안 되고,\n",
        " 최대한 한글로만 작성한다. ##, ** -- 등의 기호는 쓰지 않고,\n",
        " 디시 말투를 유지한다. 글 제목도 디시처럼 달아야한다.\n",
        "    \"\"\"\n",
        "    memory_path = '/content/drive/MyDrive/Data/'  # Google Drive 내 Data 폴더 경로\n",
        "    memory_file = 'gallery_memory.txt'\n",
        "    max_run_time = 1800  # 최대 실행 시간 (초)\n",
        "    article_interval = 600  # 글 작성 간격 (초)\n",
        "    comment_interval = 30  # 댓글 작성 간격 (초)\n",
        "    crawl_article_count = 20  # 크롤링할 글 개수\n",
        "    comment_target_count = 15  # 댓글 대상 글 개수\n",
        "    write_article_enabled = True  # 글 작성 활성화 여부\n",
        "    write_comment_enabled = True  # 댓글 활성화 여부\n",
        "    record_memory_enabled = True  # 메모리 작성 활성화 여부\n",
        "    record_data_enabled = True  # 데이터 활성화 여부\n",
        "    use_time_limit = True  # 시간 제한 사용 여부\n",
        "    load_memory_enabled = True  # 갤러리 기억 파일 로드 활성화 여부\n",
        "    load_data_enabled = True  # 데이터 파일 로드 활성화 여부\n",
        "    gallery_record_interval = 600  # 갤러리 정보 기록 간격 (초)\n",
        "\n",
        "    async with dc_api.API() as api:\n",
        "        bot = DcinsideBot(board_id, username, password, persona, memory_path, memory_file, max_run_time,\n",
        "                         comment_interval, crawl_article_count, comment_target_count,\n",
        "                         write_article_enabled, write_comment_enabled, record_memory_enabled, record_data_enabled,\n",
        "                         article_interval, use_time_limit, load_memory_enabled, load_data_enabled)\n",
        "        bot.api = api  # bot 객체에 api 속성 설정\n",
        "\n",
        "        await bot.get_gallery_name()\n",
        "        trending_topics = await bot.get_trending_topics()\n",
        "\n",
        "        if bot.load_memory_enabled:\n",
        "            memory_data = await bot.load_memory()\n",
        "        else:\n",
        "            memory_data = \"\"\n",
        "\n",
        "        # 글 작성과 댓글 작성을 동시에 실행하는 태스크 생성\n",
        "        article_task = asyncio.create_task(bot.write_article(trending_topics=trending_topics, memory_data=memory_data))\n",
        "        comment_task = asyncio.create_task(run_comment_loop(bot, use_time_limit, gallery_record_interval))\n",
        "\n",
        "        # 두 태스크가 모두 완료될 때까지 기다림\n",
        "        await asyncio.gather(article_task, comment_task)\n",
        "\n",
        "async def run_comment_loop(bot, use_time_limit, gallery_record_interval):\n",
        "    \"\"\"댓글 작성 루프를 실행하는 함수.\"\"\"\n",
        "    start_time = time()  # 시작 시간 기록\n",
        "    while (use_time_limit and time() - start_time < bot.max_run_time) or not use_time_limit:\n",
        "        await asyncio.sleep(bot.comment_interval)  # 댓글 작성 간격 설정\n",
        "\n",
        "        latest_articles = [article async for article in bot.api.board(board_id=bot.board_id, num=bot.comment_target_count)]\n",
        "        if latest_articles:\n",
        "            selected_article = random.choice(latest_articles)\n",
        "            article_title = selected_article.title\n",
        "            document_id = selected_article.id\n",
        "            comment_success = await bot.write_comment(document_id=document_id, article_title=article_title)\n",
        "            if comment_success:\n",
        "                print(f\"댓글 작성 성공! (글 ID: {document_id})\")\n",
        "\n",
        "        if time() - start_time > gallery_record_interval:\n",
        "            start_time = time()\n",
        "            await bot.record_gallery_information()  # bot.api -> bot\n",
        "\n",
        "if __name__ == \"__main__\":\n",
        "    await main()"
      ]
    }
  ],
  "metadata": {
    "colab": {
      "provenance": [],
      "authorship_tag": "ABX9TyMP82YP9uRMHOzLnn+R5boL",
      "include_colab_link": true
    },
    "kernelspec": {
      "display_name": "Python 3",
      "name": "python3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}