DB and template added

Author: Faizgeeky

date_parser.py

@@ -0,0 +1,22 @@
+# date_parser.py
+from datetime import datetime
+
+def parse_date(date_str, output_format='%d/%m/%Y'):
+    date_formats = ['%d/%m/%Y', '%Y-%m-%d', '%d-%m-%Y', '%m/%d/%Y']
+    for fmt in date_formats:
+        try:
+            parsed_date = datetime.strptime(date_str, fmt)
+            return parsed_date.strftime(output_format)
+        except ValueError:
+            continue
+    return None
+
+def extract_day_month(date_str):
+    date_formats = ['%d/%m/%Y', '%Y-%m-%d', '%d-%m-%Y', '%m/%d/%Y']
+    for fmt in date_formats:
+        try:
+            parsed_date = datetime.strptime(date_str, fmt)
+            return parsed_date.strftime('%d/%m')  # Chỉ trả về ngày và tháng
+        except ValueError:
+            continue
+    return None

main.py

@@ -0,0 +1,149 @@
+import getpass
+import os
+
+from typing import *
+import json
+import sys
+import time
+import subprocess
+import traceback
+from tempfile import NamedTemporaryFile
+# from google.colab.patches import cv2_imshow
+
+import requests
+import openai
+# from dotenv import load_dotenv
+import xml.etree.ElementTree as ET
+from datetime import datetime
+import re
+from date_parser import extract_day_month
+
+
+# if not os.environ.get('OPENAI_API_KEY'):
+#     os.environ['OPENAI_API_KEY'] = getpass.getpass("Enter the OpenAI API Key(which starts with sk-): ")
+    
+# load_dotenv()  # This loads the variables from .env
+
+api_key = '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'
+assistant_id = 'asst_FECUwGnkMA50kVsyeGcbdxJI'
+
+client = openai.Client(api_key=api_key)
+
+def execute_python_code(s: str) -> str:
+    with NamedTemporaryFile(suffix='.py', delete=False) as temp_file:
+        temp_file_name = temp_file.name
+        temp_file.write(s.encode('utf-8'))
+        temp_file.flush()
+    try:
+        result = subprocess.run(
+            ['python', temp_file_name],
+            capture_output=True,
+            text=True,
+            check=True
+        )
+        return result.stdout
+    except subprocess.CalledProcessError as e:
+        return e.stderr
+    finally:
+        import os
+        os.remove(temp_file_name)
+
+def get_lottery_result(lottery_name, date):
+    # https://xskt.com.vn/rss-feed/mien-bac-xsmb.rss
+    # return "Toi khong biet!"
+    #  Chuyển đổi ngày từ chuỗi sang đối tượng datetime
+    converted_date = extract_day_month(date)
+    if converted_date:
+        print(f"Converted date: {converted_date}")
+    else:
+        print("Invalid date format")
+
+    date = datetime.strptime(converted_date, '%d/%m')
+    rss_url = 'https://xskt.com.vn/rss-feed/mien-bac-xsmb.rss'
+
+    # Tải dữ liệu RSS từ URL
+    response = requests.get(rss_url)
+    root = ET.fromstring(response.content)
+    
+    # Tìm kiếm kết quả trong các mục
+    for item in root.findall('.//item'):
+        title = item.find('title').text
+        # Kiểm tra xem tiêu đề có chứa chuỗi ngày/tháng hay không
+        if converted_date in title:
+            return item.find('description').text
+        
+    return "Không tìm thấy kết quả xổ số cho ngày này."
+
+def run_assistant(client, assistant_id, thread_id):
+    # Create a new run for the given thread and assistant
+    run = client.beta.threads.runs.create(
+        thread_id=thread_id,
+        assistant_id=assistant_id
+    )
+
+    # Loop until the run status is either "completed" or "requires_action"
+    while run.status == "in_progress" or run.status == "queued":
+        time.sleep(3)
+        run = client.beta.threads.runs.retrieve(
+            thread_id=thread_id,
+            run_id=run.id
+        )
+
+        # At this point, the status is either "completed" or "requires_action"
+        if run.status == "completed":
+            return client.beta.threads.messages.list(
+              thread_id=thread_id
+            )
+        if run.status == "requires_action":
+            tool_call = run.required_action.submit_tool_outputs.tool_calls[0]
+            tool_outputs = []
+            if tool_call.function.name == "execute_python_code":
+                generated_python_code = json.loads(tool_call.function.arguments)['code']
+                result = execute_python_code(generated_python_code)
+                tool_outputs.append(
+                    {
+                          "tool_call_id": tool_call.id,
+                          "output": result,
+                    },
+                )
+            # Mike add
+            elif tool_call.function.name == "get_lottery_result":
+                lottery_name = json.loads(tool_call.function.arguments)['lottery_name']
+                draw_date = json.loads(tool_call.function.arguments)['draw_date']
+                result = get_lottery_result(lottery_name, draw_date)
+                tool_outputs.append(
+                    {
+                          "tool_call_id": tool_call.id,
+                          "output": result,
+                    },
+                )
+
+            if tool_outputs:
+                run = client.beta.threads.runs.submit_tool_outputs(
+                    thread_id=thread_id,
+                    run_id=run.id,
+                    tool_outputs=tool_outputs
+                )
+
+if __name__ == "__main__":
+        script = input("python code to execute: \n")
+        
+        
+        # Create a new thread
+        thread = client.beta.threads.create()
+        thread_id = thread.id
+        # Create a new thread message with the provided task
+        thread_message = client.beta.threads.messages.create(
+            thread.id,
+            role="user",
+            content=script,
+        )
+
+        # assistant_id, thread_id = setup_assistant(client, script)
+        print(f"Debugging: Useful for checking the generated agent in the playground. https://platform.openai.com/playground?mode=assistant&assistant={assistant_id}")
+        print(f"Debugging: Useful for checking logs. https://platform.openai.com/playground?thread={thread_id}")
+
+        messages = run_assistant(client, assistant_id, thread_id)
+
+        # message_dict = json.loads(messages)
+        print(messages)

oapi/customBot/__init__.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

oapi/customBot/admin.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class CustombotConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "customBot"

oapi/customBot/apps.py

@@ -0,0 +1,30 @@
+# Generated by Django 4.2.9 on 2024-01-24 04:24
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = []
+
+    operations = [
+        migrations.CreateModel(
+            name="UserProfile",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                ("username", models.CharField(max_length=150)),
+                ("assitantId", models.CharField(max_length=200)),
+                ("threadID", models.CharField(max_length=200)),
+            ],
+        ),
+    ]

oapi/customBot/migrations/0001_initial.py

@@ -0,0 +1,14 @@
+from django.db import models
+
+# Create your models here.
+
+
+class UserProfile(models.Model):
+    username = models.CharField(max_length=150)
+    assitantId = models.CharField(max_length=200)
+    threadID = models.CharField(max_length=200)
+
+    # email = models.EmailField()
+    # password = models.CharField(max_length=100)
+    # user_type = models.CharField(max_length=20, choices=[('admin', 'Admin'),('employee', 'Employee'),('student', 'Student'), ('hostelOwner', 'Hostel Owner')])
+    

oapi/customBot/migrations/__init__.py

@@ -0,0 +1,119 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  <link rel="stylesheet" href="styles.css">
+  <style>
+    body {
+      margin: 0;
+      font-family: Arial, sans-serif;
+    }
+
+    .chat-container {
+      max-width: 600px;
+      margin: 0 auto;
+      padding: 20px;
+    }
+
+    .chat {
+      border: 1px solid #ccc;
+      padding: 10px;
+      border-radius: 10px;
+      overflow-y: scroll;
+      max-height: 400px;
+    }
+
+    .message {
+      margin: 10px;
+      padding: 10px;
+      border-radius: 10px;
+      max-width: 70%;
+    }
+
+    .received {
+      background-color: #f2f2f2;
+      align-self: flex-start;
+    }
+
+    .sent {
+      background-color: #dcf8c6;
+      align-self: flex-end;
+    }
+
+    .message-input {
+      display: flex;
+      margin-top: 20px;
+    }
+
+    .message-input input {
+      flex: 1;
+      padding: 10px;
+      border: none;
+      border-radius: 5px;
+    }
+
+    .message-input button {
+      padding: 10px 20px;
+      border: none;
+      background-color: #007bff;
+      color: #fff;
+      border-radius: 5px;
+      cursor: pointer;
+    }
+  </style>
+  <title>Chat with Assistant</title>
+</head>
+<body>
+  <div class="chat-container">
+    <div class="chat" id="chat-container">
+      <!-- Existing messages go here -->
+    </div>
+    <div class="message-input">
+      <input type="text" id="user-input" placeholder="Type your message...">
+      <button id="send-button">Send</button>
+    </div>
+  </div>
+
+  <script>
+    document.addEventListener('DOMContentLoaded', function () {
+      const chatContainer = document.getElementById('chat-container');
+      const userInput = document.getElementById('user-input');
+      const sendButton = document.getElementById('send-button');
+
+      sendButton.addEventListener('click', async function () {
+        const userMessage = userInput.value;
+        console.log("user message", userMessage);
+        try {
+          const response = await fetch('http://localhost:3000/answer', {
+            method: 'POST',
+            headers: {
+              'Content-Type': 'application/json',
+            },
+            body: JSON.stringify({
+              question: userMessage,
+              assitant_id: 'asst_8klujY8GWez2QcUITxD6AFlQ',  // Replace with your actual assistant ID
+            }),
+          });
+
+          const data = await response.json();
+
+          // Add the assistant's response to the chat
+          const assistantResponseDiv = document.createElement('div');
+          assistantResponseDiv.className = 'message received';
+          assistantResponseDiv.innerHTML = `<div class="message-content">${data.answer}</div>`;
+          chatContainer.appendChild(assistantResponseDiv);
+
+          // Clear the user input
+          userInput.value = '';
+        } catch (error) {
+          console.error('Error asking question:', error);
+        }
+      });
+    });
+  </script>
+</body>
+</html>
+
+
+

oapi/customBot/models.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

oapi/customBot/templates/index.html

@@ -0,0 +1,7 @@
+
+from django.urls import path
+from .views import *
+
+urlpatterns = [
+        path('', home, name='home'),
+]

oapi/customBot/tests.py

@@ -0,0 +1,6 @@
+from django.shortcuts import render
+
+# Create your views here.
+
+def home(request):
+    return render(request,'index.html')