Spaces:

rorshi
/

autonomous-ai-agent-sim

Sleeping

App Files Files Community

rorshi commited on May 24

Commit

6c5b86d

1 Parent(s): 73db488

Complete step 3 and test wep play (deepseek ver)

Browse files

Files changed (7) hide show

app.py +2 -2
npc_social_network/models/deepseek_setup.py +9 -3
npc_social_network/routes/npc_route.py +2 -11
npc_social_network/static/js/npc_chat.js +1 -1
npc_social_network/templates/chat.html +19 -17
requirements.txt +66 -0
test.ipynb +44 -1

app.py CHANGED Viewed

@@ -8,8 +8,8 @@ def create_app():
     app = Flask(__name__) # static/template 경로를 기본값
     # 각 프로젝트는 Blueprint에서 자기 static/template 관리
-    app.register_blueprint(npc_bp, url_prefix="/npc")
-    # app.register_blueprint(stock_bp, url_prefix="/stock")
     @app.route("/")
     def index():

     app = Flask(__name__) # static/template 경로를 기본값
     # 각 프로젝트는 Blueprint에서 자기 static/template 관리
+    app.register_blueprint(npc_bp)
+    # app.register_blueprint(stock_bp)
     @app.route("/")
     def index():

npc_social_network/models/deepseek_setup.py CHANGED Viewed

@@ -1,13 +1,19 @@
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 def load_deepseek():
-    model_id = "deepseek-ai/deepseek-llm-7b-instruct"
-    tokenizer = AutoTokenizer.from_pretrained(model_id)
     model = AutoModelForCausalLM.from_pretrained(
         model_id,
         device_map="auto", # GPU 자동 할당
-        torch_dtype="auto"
     )
     pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)

 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+from dotenv import load_dotenv
+import os
 def load_deepseek():
+    load_dotenv() # .env 파일에서 환경 변수 로드
+    model_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"  # 모델 성능이 영 별로네(한글로 답변도 못 해주고 답변도 자연스럽지 못함)
+    access_token = os.getenv("deepseek_HF_token")
+    tokenizer = AutoTokenizer.from_pretrained(model_id, token=access_token)
     model = AutoModelForCausalLM.from_pretrained(
         model_id,
         device_map="auto", # GPU 자동 할당
+        torch_dtype="auto",
+        token=access_token
     )
     pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)

npc_social_network/routes/npc_route.py CHANGED Viewed

@@ -40,17 +40,8 @@ def chat():
     위의 정보를 바탕으로 {npc.name}으로서 자연스럽고 일관성 있는 답변을 해주세요.
     """
-    response = openai.ChatCompletion.create(
-        model="gpt-4o",
-        messages=[
-            {"role": "system", "content": "당신은 가상의 캐릭터로서 성격과 기억을 바탕으로 대화를 합니다." }
-            {"role": "user", "content": prompt}
-        ],
-        max_tokens = 150
-    )
-    npc_reply = response.choices[0].message.content.strip()
     # NPC 기억에 대화 추가 및 관계 점수 업데이트
     npc.remember_conversation(user_input, npc_reply)

     위의 정보를 바탕으로 {npc.name}으로서 자연스럽고 일관성 있는 답변을 해주세요.
     """
+    result = llm(prompt, max_new_tokens=150, do_sample=True)[0]["generated_text"]
+    npc_reply = result.strip().replace(prompt.strip(), "")
     # NPC 기억에 대화 추가 및 관계 점수 업데이트
     npc.remember_conversation(user_input, npc_reply)

npc_social_network/static/js/npc_chat.js CHANGED Viewed

@@ -4,7 +4,7 @@ async function sendMessage() {
     const userMessage = document.getElementById("message").value;
     const npc = document.getElementById("npc").value;
-    const response = await fetch('/chat', {
         method: 'POST',
         headers: { 'Content-Type': 'application/json' },
         body: JSON.stringify({ message: userMessage, npc: npc })

     const userMessage = document.getElementById("message").value;
     const npc = document.getElementById("npc").value;
+    const response = await fetch('/npc_social_network/chat', {
         method: 'POST',
         headers: { 'Content-Type': 'application/json' },
         body: JSON.stringify({ message: userMessage, npc: npc })

npc_social_network/templates/chat.html CHANGED Viewed

@@ -1,20 +1,22 @@
 <!-- portfolio/npc_social_network/templates/chat.html -->
 <!DOCTYPE html>
-<html lang="en">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>NPC Chat</title>
-</head>
-<body>
-    <h1>NPC 소셜 네트워크</h1>
-    <select id = "npc">
-        <option>Alice</option>
-        <option>Bob</option>
-        <option>Charlie</option>
-    </select>
-    <input type="text" id="message" placeholder="메세지를 입력하세요"/>
-    <button onclick="sendMessage()">보내기</button>
-    <div id="chatBox"></div>
-</body>
 </html>

 <!-- portfolio/npc_social_network/templates/chat.html -->
 <!DOCTYPE html>
+    <html lang="en">
+    <head>
+        <meta charset="UTF-8">
+        <meta name="viewport" content="width=device-width, initial-scale=1.0">
+        <title>NPC Chat</title>
+    </head>
+    <body>
+        <h1>NPC 소셜 네트워크</h1>
+        <select id = "npc">
+            <option>Alice</option>
+            <option>Bob</option>
+            <option>Charlie</option>
+        </select>
+        <input type="text" id="message" placeholder="메세지를 입력하세요"/>
+        <button onclick="sendMessage()">보내기</button>
+        <div id="chatBox"></div>
+        <script src="{{ url_for('npc_social.static', filename='js/npc_chat.js') }}"></script>
+    </body>
 </html>

requirements.txt ADDED Viewed

	@@ -0,0 +1,66 @@

+accelerate==1.7.0
+asttokens @ file:///home/conda/feedstock_root/build_artifacts/asttokens_1733250440834/work
+bitsandbytes==0.45.5
+blinker==1.9.0
+certifi==2025.4.26
+charset-normalizer==3.4.2
+click==8.2.0
+colorama==0.4.6
+comm @ file:///home/conda/feedstock_root/build_artifacts/comm_1733502965406/work
+debugpy @ file:///D:/bld/debugpy_1680755734210/work
+decorator @ file:///home/conda/feedstock_root/build_artifacts/decorator_1740384970518/work
+exceptiongroup @ file:///home/conda/feedstock_root/build_artifacts/exceptiongroup_1746947292760/work
+executing @ file:///home/conda/feedstock_root/build_artifacts/executing_1745502089858/work
+filelock==3.18.0
+Flask==3.1.1
+fsspec==2025.5.0
+huggingface-hub==0.32.0
+idna==3.10
+importlib_metadata @ file:///home/conda/feedstock_root/build_artifacts/bld/rattler-build_importlib-metadata_1747934053/work
+ipykernel @ file:///D:/bld/ipykernel_1719845595208/work
+ipython @ file:///D:/bld/bld/rattler-build_ipython_1745672185/work
+ipython_pygments_lexers @ file:///home/conda/feedstock_root/build_artifacts/ipython_pygments_lexers_1737123620466/work
+itsdangerous==2.2.0
+jedi @ file:///home/conda/feedstock_root/build_artifacts/jedi_1733300866624/work
+Jinja2==3.1.6
+jupyter_client @ file:///home/conda/feedstock_root/build_artifacts/jupyter_client_1733440914442/work
+jupyter_core @ file:///D:/bld/jupyter_core_1727163532151/work
+MarkupSafe==3.0.2
+matplotlib-inline @ file:///home/conda/feedstock_root/build_artifacts/matplotlib-inline_1733416936468/work
+mpmath==1.3.0
+nest_asyncio @ file:///home/conda/feedstock_root/build_artifacts/nest-asyncio_1733325553580/work
+networkx==3.4.2
+numpy==2.2.6
+packaging @ file:///home/conda/feedstock_root/build_artifacts/bld/rattler-build_packaging_1745345660/work
+parso @ file:///home/conda/feedstock_root/build_artifacts/parso_1733271261340/work
+pickleshare @ file:///home/conda/feedstock_root/build_artifacts/pickleshare_1733327343728/work
+pillow==11.0.0
+platformdirs @ file:///home/conda/feedstock_root/build_artifacts/bld/rattler-build_platformdirs_1746710438/work
+prompt_toolkit @ file:///home/conda/feedstock_root/build_artifacts/prompt-toolkit_1744724089886/work
+psutil @ file:///D:/bld/psutil_1681775154857/work
+pure_eval @ file:///home/conda/feedstock_root/build_artifacts/pure_eval_1733569405015/work
+Pygments @ file:///home/conda/feedstock_root/build_artifacts/pygments_1736243443484/work
+python-dateutil @ file:///home/conda/feedstock_root/build_artifacts/python-dateutil_1733215673016/work
+python-dotenv==1.1.0
+pywin32==304
+PyYAML==6.0.2
+pyzmq @ file:///D:/bld/pyzmq_1679317063994/work
+regex==2024.11.6
+requests==2.32.3
+safetensors==0.5.3
+six @ file:///home/conda/feedstock_root/build_artifacts/six_1733380938961/work
+stack_data @ file:///home/conda/feedstock_root/build_artifacts/stack_data_1733569443808/work
+sympy==1.14.0
+tokenizers==0.21.1
+torch==2.7.0+cu118
+torchaudio==2.7.0+cu118
+torchvision==0.22.0+cu118
+tornado @ file:///D:/bld/tornado_1681817639987/work
+tqdm==4.67.1
+traitlets @ file:///home/conda/feedstock_root/build_artifacts/traitlets_1733367359838/work
+transformers==4.52.3
+typing_extensions @ file:///home/conda/feedstock_root/build_artifacts/bld/rattler-build_typing_extensions_1744302253/work
+urllib3==2.4.0
+wcwidth @ file:///home/conda/feedstock_root/build_artifacts/wcwidth_1733231326287/work
+Werkzeug==3.1.3
+zipp @ file:///home/conda/feedstock_root/build_artifacts/zipp_1732827521216/work

test.ipynb CHANGED Viewed

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": null,
    "id": "ecf4c973",
    "metadata": {},
    "outputs": [],
@@ -33,6 +33,41 @@
     "# 1. ai 모델 학습을 위해 어떤 내용이 필요한지\n",
     "# 2. 학습한 모델을 저장, 활용할 수 있는지"
    ]
   }
  ],
  "metadata": {
@@ -42,7 +77,15 @@
    "name": "python3"
   },
   "language_info": {
    "name": "python",
    "version": "3.11.11"
   }
  },

  "cells": [
   {
    "cell_type": "code",
+   "execution_count": 2,
    "id": "ecf4c973",
    "metadata": {},
    "outputs": [],
     "# 1. ai 모델 학습을 위해 어떤 내용이 필요한지\n",
     "# 2. 학습한 모델을 저장, 활용할 수 있는지"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "7469ad02",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "True\n",
+      "1\n",
+      "NVIDIA GeForce GTX 1070 Ti\n"
+     ]
+    }
+   ],
+   "source": [
+    "# 1. GPU 사용 여부 확인\n",
+    "import torch\n",
+    "print(torch.cuda.is_available())  # 👉 True면 GPU 사용 가능\n",
+    "print(torch.cuda.device_count())  # 👉 연결된 GPU 수\n",
+    "print(torch.cuda.get_device_name(0))  # 👉 GPU 이름"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "b055896e",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# GPU 지원 버전으로 다시 설치\n",
+    "# !pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118"
+   ]
   }
  ],
  "metadata": {
    "name": "python3"
   },
   "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
    "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
    "version": "3.11.11"
   }
  },