From 1e4e9ccd388099b7f347abf7f6d8348257ed9539 Mon Sep 17 00:00:00 2001 From: satya-codes Date: Thu, 27 Jan 2022 12:14:59 +0530 Subject: [PATCH] First Commit --- .gitignore | 2 + app.py | 22 ++ chat.py | 7 +- data.pth | Bin 0 -> 5911 bytes intents.bak | 84 +++++++ intents.json | 229 +++++++++++++------- standalone-frontend/app.js | 91 -------- standalone-frontend/base.html | 39 ---- standalone-frontend/images/chatbox-icon.svg | 3 - standalone-frontend/style.css | 200 ----------------- static/app.js | 83 +++++++ templates/base.html | 2 +- train.py | 26 ++- 13 files changed, 355 insertions(+), 433 deletions(-) create mode 100644 .gitignore create mode 100644 data.pth create mode 100644 intents.bak delete mode 100644 standalone-frontend/app.js delete mode 100644 standalone-frontend/base.html delete mode 100644 standalone-frontend/images/chatbox-icon.svg delete mode 100644 standalone-frontend/style.css diff --git a/.gitignore b/.gitignore new file mode 100644 index 000000000..53550da0f --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +/__pycache__ +/venv \ No newline at end of file diff --git a/app.py b/app.py index e69de29bb..b6fccf1b0 100644 --- a/app.py +++ b/app.py @@ -0,0 +1,22 @@ +from flask import Flask, render_template, request, jsonify +from chat import get_response + +app = Flask(__name__) + + +@app.get("/") +def index_get(): + return render_template("base.html") + + +@app.post("/predict") +def predict(): + text = request.get_json().get("message") + # TODO: check if the text is valid + response = get_response(text) + message = {"answer": response} + return jsonify(message) + + +if __name__ == "__main__": + app.run(debug=True) diff --git a/chat.py b/chat.py index 1abbbee6f..b6b213734 100644 --- a/chat.py +++ b/chat.py @@ -25,7 +25,8 @@ model.load_state_dict(model_state) model.eval() -bot_name = "Sam" +bot_name = "SDF" + def get_response(msg): sentence = tokenize(msg) @@ -44,18 +45,16 @@ def get_response(msg): for intent in intents['intents']: if tag == intent["tag"]: return random.choice(intent['responses']) - + return "I do not understand..." if __name__ == "__main__": print("Let's chat! (type 'quit' to exit)") while True: - # sentence = "do you use credit cards?" sentence = input("You: ") if sentence == "quit": break resp = get_response(sentence) print(resp) - diff --git a/data.pth b/data.pth new file mode 100644 index 0000000000000000000000000000000000000000..b6b09a80afb07d1e742cd57081bab07b047a63a4 GIT binary patch literal 5911 zcmb7I30M@zwgy22L{Ja`jfgwApn!nD^eI3CDzXcrNN|9L85w7&nZW@C6%cR(6ahsL zH4$7uMI%NdU8NE=#<(xnEsE=PH}@)P;&yv_OrG$*l~TB)P)0KzMRwS{qc$HeL zX0d)^qCzHP6V+0MTxA-nOk$O6(s*g2+7x$kiogbrF(y1h=Hjejr703MHcV72__3Vh z)74U$$~0cdCZtPcN%3k{u2LxD=eU~U&W?^TU3id+j})h{*hmL44Uj3s>PVf_6nBZy z=VL}=;}`?}oS2r5O@haEQe)HbAiW^tpuhs$H7GDhzW|#pk!Y24Q)BZ8Y#}i7auTFs z749zK$C4OHbX+dDhr}evSdA?OihOJpq@Ry_>fqLD+$#dx=w{>k$82od5x8PIp1@a) zd+Rh97vMg;6kfv@I+DI>Y#)L9=`^_NG&r=QaKrs|8U}pQFi@*ukPbdrjfX_wp*jt2 ze`s)QM=%CE@dRONJWNYq8pLZDuG26=M>0~4og?rlod!3ZhSBXP+_4K!VXDTi0!2P{ z(`p!_gS)G-M+Ejv&&Fc~9r3eSwOCs<*h^SuZ^sxu9eUU&8;gW7=CDc?UmKVS(pSZF z;=5!n7_u>v{Ex#G`(|Um&m3;pKO2wJIryU0Q?jyjJU$zLDY(*A2{ER;S1M0SSI4WQ zb6Fe^6ev`tM4FVu%5}cLAY;L&NLT+EHzCMWP%W0p;x!6ok_u0Z!;>5ZT|;N%Af2JY zlZE0EDhCO@k%q58y17KiJ)RX8HXNQ=?1D#dIZDR>iD zJVgjh$YgPpu3*J-937*bTq2vq;;8~>qC~7r!7(~pF;bC+rwL?<3Pl>L#M1?2k~kB` z3QL;4KV!@WMGMNG= z2qCCHJin0IwKQf0U{UkKBW*QS0rMU0G2C~#4@ZF zA{EIvU5Heq@-%aVeuXlHA4nstj$lhwI78Q?k@Lf53U-=|<+p;l!a}DhrE~aYnWyXE zFIb!S@GgeUMXH8glZ(b+@*pjjfCZ+ zm3WyDu8~L+C3v|Iu8}BkkuF@yODWbl`7D(PmUiG3LN{+I@k&-FLY$()rEz$bqpk$C zrODzlVJK}KX-m0WxS9A;V%1VPA6^mDgP#q*P0Jm`X=yTPqFDP_aKNhtMj?l51a(@| z1y>4_m-FE&Z56E5bqPtW`n2lWbg;NukRYs*3a?|+ox_5<7`um#d|JfcYT6d|mk&2{ zJb9Wqou@=MVdWAbZ-e6CQ|5ZkgJ&6_Vxe*}>oos~0vqj{)v%OGpcoB*dm%&tg z4Ju(AH0;MD%EFW$Y%}8&#hKioW#hp1oG-dyypH*i35WN$1DSx-*>LRQBgojLU`EV) zz#UTM5nEe18q#G88O#oXQzH;q^*9XKwSFkF|5$XRrWe}7G(gSVXkuVk$B35}a#v>A zz>9P_VOQIcnA}EAV&j2!KOfb+7EY2FLk)>A?}x(5bI6mcm&xqb?@8Iq>tN_1Mx<&3 zS9)d@d@*A)L@i&89_X`B@$w#QbJztp&J;1eY-euV{_BwOn+du7$cn6$ZTGo0`aJ2f z+!o!g)4+m9=BR%~6{F_dk?Xq=Xv~>HNG`l88Z$bMJo7lg{62UIIR5+sHo$q}y?-Vx zaeE6cb_*b_`3I)%rVUMY=!NDUf7WdLrj3)Yzs1bEri4eLpUH&}hseDL{mJmCNK$Pz zfx>%llKR$*95yK928=vUrk)v!-gq{W`0n3xKdsmV12!CleFb^2UA9ORXLFy)81H~a z{v(YTo0*{b%`I>vr;>Aw%k1V(!;Y8*+F60caXwaRBr$u|F7n4^TRulCR3n)!&VIG@xr|bK3FpHjs zUt+dGR$o1m*Dnjs?Q0~{miMD8gZ0Rc*#6Y>=n(34u9~!6v!aWKP7qCR|xqz}CT;TBG4h51p=;LCet*C`P)_z;3B zo96)|-3(-K1^WF*M=&h92qRi&!i&51KEAy@k;lJUVfW(}kiBvzSFcx-F>7y;Bh}sD zcuh9!-L#H0l9eDTH79RgnUD1a?rVuOpdH0 zqTMTCA=q)wt~uOCaX4`-{hEwTmZLz01?^Q=2*E+7H0ht6XijAUHB0V}wwtM8rNc2; za{DMLZQMd;5F_&7-4Sxny1x12cWtEKBsVa>^a!MHdmxF2nwz#NiFrLr!riI>8Io2gSh#pI1J89~hDIldQQ{8YMZ9JrF#e&VY03 z1g_I#3G%iXj(nOA!@KG(@M`8(GBP%sxzYSJIW%y*PwM4Ra>KEl*tP=ms#^$J>oU)UX_G}}=;*!YMgMwjUd^b@0wvnVGb4k5tDN*+fhECC!n36edOmxm+GCZV4bT(oa zZ0PES{4bdLr0)NT%ZXo0s*Z`lqPLZ3@Yb%xdTck;+SEW+_3R0rO9#?wI)vO@Jdh@Y zTGOxOmqppf2a<)FB52#~Et;3)h6-KBlF+6>Xr*E)3SUYfsKOpS!n+UTHlhr)txC9tgQG^BqO0Y_@{z$0J+y1LyPRCA4q&-q?l$*>T#YQO+; zk1K_ny-UC@Vk=1P&Vr-YR7g2(h925{fpT-w$q#F%gT!t*GcI*8M1)sz-;do$^kF5j zMs@JyAAPwWztJZHeNRA3fC+v0zy$UlW67X^Ch(MB0FIWyb`%Q{$(y-@VYT2_tfDig zdB6;h^$?h839s03VB-ISsItrwCFa_}&3ED4g>Bj7_&I$x!K6=QE zD}wIEy@A?mxXigU<<7^%?H`oCNr6*87c|2{7%O_8t%!g>-i?E;ML0x7YDa@G&KK@Qb^=mYx zEAp6UsXjEfcPjUn zfgbhGxz`29;L@H=(3t&RWW8W0>T6OV+VP>5Df==SU7!t|_s)DSKw3*ScDoG9^9&kl z^|(2roCWV&rd)0xBNQ3)6sBxi3C2Fw^oC+99C^J3t{@q6w}&^mYPqi2b<+?LHtUQQ zP0~kc>R|Zk>zT~tDW%{&H-VJi)I*BFFNu@8H7&BUBHn8!pgPw|WF{g9+zt|AsQwxiDG^^n*pkNnuV!TbAsU%F)dIF#>fNvE1k zqEFd1sIqztvr+9#4CmLA;-d^rwOWsEH`NR48*sc-s;B*I!neO$U&HqKbqny<)0?%y z0hPb|8LInKald{XiAGGXfLcXwBtJA1MQ$?{`0GY*$<^jg+y0KfL;Z~TZ}J!Vp`4~E zx5KQ$6%1b16P@xrK`m-)X>#y7`ey7(IH}M_bM9TGZ!W)u$6lFCP-7x>-Lry5gxDb2 z7w?(6P4%=&p-+$gi=(>^Iip1v_i^`6Zl=-tp{UzO7AfS*=*g%X(0cTJbB||(QStS6 zG-GQeg!;A6s;td)%KM%a>`h?6&8`RcRxPL8_#*hWcPS(X*pb$4#js(;9vD^OM9Ytt zQhy^)S{7-7iuRS$CY#M*dV2yGY}rBTUG6gX_tr33LAEgWo1I&a+o>M*>R-Wxj}g&K;;&pj}6>dtg@qGcBKY5FI9P-=;c zdN7cSsP_K4cqFINm(RVn{at=K zYC3U?|i<$ofSG9r+s2 zw!a?7La)!Cb@u5iQGmbAUu`6ArrH^ttx1W}enir?upoWy>CbPPpZ_52uI*@dbo|2E z5&ikzXWkC^3;zT2bM&au9kHM9TkWxLD!1$Y{i;AbW$6e}?nV sC*klQa`vqCAsh9S@u;1~%&e{)iddB>pHeMI|ALlL>t^fc4 literal 0 HcmV?d00001 diff --git a/intents.bak b/intents.bak new file mode 100644 index 000000000..18e0891b3 --- /dev/null +++ b/intents.bak @@ -0,0 +1,84 @@ +{ + "intents": [ + { + "tag": "greeting", + "patterns": [ + "Hi", + "Hey", + "How are you", + "Is anyone there?", + "Hello", + "Good day" + ], + "responses": [ + "Hey :-)", + "Hello, thanks for visiting", + "Hi there, what can I do for you?", + "Hi there, how can I help?" + ] + }, + { + "tag": "goodbye", + "patterns": ["Bye", "See you later", "Goodbye"], + "responses": [ + "See you later, thanks for visiting", + "Have a nice day", + "Bye! Come back again soon." + ] + }, + { + "tag": "thanks", + "patterns": ["Thanks", "Thank you", "That's helpful", "Thank's a lot!"], + "responses": ["Happy to help!", "Any time!", "My pleasure"] + }, + { + "tag": "items", + "patterns": [ + "Which items do you have?", + "What kinds of items are there?", + "What do you sell?" + ], + "responses": [ + "We sell coffee and tea", + "We have coffee and tea" + ] + }, + { + "tag": "payments", + "patterns": [ + "Do you take credit cards?", + "Do you accept Mastercard?", + "Can I pay with Paypal?", + "Are you cash only?" + ], + "responses": [ + "We accept VISA, Mastercard and Paypal", + "We accept most major credit cards, and Paypal" + ] + }, + { + "tag": "delivery", + "patterns": [ + "How long does delivery take?", + "How long does shipping take?", + "When do I get my delivery?" + ], + "responses": [ + "Delivery takes 2-4 days", + "Shipping takes 2-4 days" + ] + }, + { + "tag": "funny", + "patterns": [ + "Tell me a joke!", + "Tell me something funny!", + "Do you know a joke?" + ], + "responses": [ + "Why did the hipster burn his mouth? He drank the coffee before it was cool.", + "What did the buffalo say when his son left for college? Bison." + ] + } + ] +} diff --git a/intents.json b/intents.json index 18e0891b3..d9f1d7239 100644 --- a/intents.json +++ b/intents.json @@ -1,84 +1,147 @@ { - "intents": [ - { - "tag": "greeting", - "patterns": [ - "Hi", - "Hey", - "How are you", - "Is anyone there?", - "Hello", - "Good day" - ], - "responses": [ - "Hey :-)", - "Hello, thanks for visiting", - "Hi there, what can I do for you?", - "Hi there, how can I help?" - ] - }, - { - "tag": "goodbye", - "patterns": ["Bye", "See you later", "Goodbye"], - "responses": [ - "See you later, thanks for visiting", - "Have a nice day", - "Bye! Come back again soon." - ] - }, - { - "tag": "thanks", - "patterns": ["Thanks", "Thank you", "That's helpful", "Thank's a lot!"], - "responses": ["Happy to help!", "Any time!", "My pleasure"] - }, - { - "tag": "items", - "patterns": [ - "Which items do you have?", - "What kinds of items are there?", - "What do you sell?" - ], - "responses": [ - "We sell coffee and tea", - "We have coffee and tea" - ] - }, - { - "tag": "payments", - "patterns": [ - "Do you take credit cards?", - "Do you accept Mastercard?", - "Can I pay with Paypal?", - "Are you cash only?" - ], - "responses": [ - "We accept VISA, Mastercard and Paypal", - "We accept most major credit cards, and Paypal" - ] - }, - { - "tag": "delivery", - "patterns": [ - "How long does delivery take?", - "How long does shipping take?", - "When do I get my delivery?" - ], - "responses": [ - "Delivery takes 2-4 days", - "Shipping takes 2-4 days" - ] - }, - { - "tag": "funny", - "patterns": [ - "Tell me a joke!", - "Tell me something funny!", - "Do you know a joke?" - ], - "responses": [ - "Why did the hipster burn his mouth? He drank the coffee before it was cool.", - "What did the buffalo say when his son left for college? Bison." - ] - } - ] -} + "intents": [ + { + "tag": "greetings", + "patterns": [ + "Hi", + "Hey", + "How are you", + "Is anyone there?", + "Hello", + "Good day", + "Hi there" + ], + "responses": [ + "Hello thanks for checking in", + "Hi there, how can i help you", + "Hi there, what can I do for you?", + "Hi there, how can I help?" + ] + }, + { + "tag": "goodbye", + "patterns": [ + "bye", + "good bye", + "see you later" + ], + "responses": [ + "have a nice time, welcome back again", + "bye bye" + ] + }, + { + "tag": "thanks", + "patterns": [ + "Thanks", + "okay", + "Thank you", + "thankyou", + "That's helpful", + "Awesome, thanks", + "Thanks for helping me", + "wow", + "great" + ], + "responses": [ + "Happy to help!", + "Any time!", + "you're welcome", + "My pleasure" + ] + }, + { + "tag": "noanswer", + "patterns": [ + "" + ], + "responses": [ + "Sorry, I didn't understand you", + "Please give me more info", + "Not sure I understand that" + ] + }, + { + "tag": "name1", + "patterns": [ + "what's your name?", + "who are you?" + ], + "responses": [ + "I'm just an SDF virtual agent. I only exist in the world of DoFPD", + "I'm a SDF virtual chat agent" + ] + }, + { + "tag": "name", + "patterns": [ + "my name is ", + "I'm ", + "I am" + ], + "responses": [ + "Oooh great to meet you ! How may I assist you ", + "Oh, I'll keep that in mind" + ] + }, + { + "tag": "charge", + "patterns": [ + "Whether charge lower than the SDF charge be ceded?", + "what is the charge of SDF, lower than the SDF charge be ceded?", + "May I know the Charge", + "may i know the charge" + ], + "responses": [ + "Oh Yes, it can be ceded", + "It is ceded", + "it can be ceded" + ] + }, + { + "tag": "nodal", + "patterns": [ + "which nodal agency for cooperative factories?", + "Who will be the nodal agency for processing NOC request?", + "can you help me, which nodal agency for co-operative sugar factories for processing NOC requests?", + "who will process NOC request for co-operative sugar factories?" + ], + "responses": [ + "IFCI for private sugar factories and NCDC for co-operative sugar factories." + ] + }, + { + "tag": "private", + "patterns": [ + "Which is the nodal agency for private sugar factories for processing NOC requests?", + "private nodal agency", + "nodal agency for private sugar factories", + "Please tell me the nodal agency for private factories", + "Will you help me as a prive sugar factory ownere, which agency will facilitate in this?", + "private agency", + "private organization", + "private organisation" + ], + "responses": [ + "Yes, here is the link https://www.ifciltd.com/", + "yes IFCL", + "Hey! you can visit https://www.ifciltd.com/", + "I am sure it's Industrial Finance Corporation of India (IFCI)" + ] + }, + { + "tag": "loan application ", + "patterns": [ + "SDF Loan", + "sdf loan", + "loan link", + "loan form" + ], + "responses": [ + "https://sdfportal.in/Login", + "https://dfpd.gov.in/SDF-Main.htm" + ] + + } + ] +} \ No newline at end of file diff --git a/standalone-frontend/app.js b/standalone-frontend/app.js deleted file mode 100644 index 6bdf287d1..000000000 --- a/standalone-frontend/app.js +++ /dev/null @@ -1,91 +0,0 @@ -class Chatbox { - constructor() { - this.args = { - openButton: document.querySelector('.chatbox__button'), - chatBox: document.querySelector('.chatbox__support'), - sendButton: document.querySelector('.send__button') - } - - this.state = false; - this.messages = []; - } - - display() { - const {openButton, chatBox, sendButton} = this.args; - - openButton.addEventListener('click', () => this.toggleState(chatBox)) - - sendButton.addEventListener('click', () => this.onSendButton(chatBox)) - - const node = chatBox.querySelector('input'); - node.addEventListener("keyup", ({key}) => { - if (key === "Enter") { - this.onSendButton(chatBox) - } - }) - } - - toggleState(chatbox) { - this.state = !this.state; - - // show or hides the box - if(this.state) { - chatbox.classList.add('chatbox--active') - } else { - chatbox.classList.remove('chatbox--active') - } - } - - onSendButton(chatbox) { - var textField = chatbox.querySelector('input'); - let text1 = textField.value - if (text1 === "") { - return; - } - - let msg1 = { name: "User", message: text1 } - this.messages.push(msg1); - - fetch('http://127.0.0.1:5000/predict', { - method: 'POST', - body: JSON.stringify({ message: text1 }), - mode: 'cors', - headers: { - 'Content-Type': 'application/json' - }, - }) - .then(r => r.json()) - .then(r => { - let msg2 = { name: "Sam", message: r.answer }; - this.messages.push(msg2); - this.updateChatText(chatbox) - textField.value = '' - - }).catch((error) => { - console.error('Error:', error); - this.updateChatText(chatbox) - textField.value = '' - }); - } - - updateChatText(chatbox) { - var html = ''; - this.messages.slice().reverse().forEach(function(item, index) { - if (item.name === "Sam") - { - html += '
' + item.message + '
' - } - else - { - html += '
' + item.message + '
' - } - }); - - const chatmessage = chatbox.querySelector('.chatbox__messages'); - chatmessage.innerHTML = html; - } -} - - -const chatbox = new Chatbox(); -chatbox.display(); \ No newline at end of file diff --git a/standalone-frontend/base.html b/standalone-frontend/base.html deleted file mode 100644 index c2c3b5c39..000000000 --- a/standalone-frontend/base.html +++ /dev/null @@ -1,39 +0,0 @@ - - - - - - - Chatbot - - -
-
-
-
-
- image -
-
-

Chat support

-

Hi. My name is Sam. How can I help you?

-
-
-
-
-
-
- - -
-
-
- -
-
-
- - - - - \ No newline at end of file diff --git a/standalone-frontend/images/chatbox-icon.svg b/standalone-frontend/images/chatbox-icon.svg deleted file mode 100644 index 15617acc8..000000000 --- a/standalone-frontend/images/chatbox-icon.svg +++ /dev/null @@ -1,3 +0,0 @@ - - - diff --git a/standalone-frontend/style.css b/standalone-frontend/style.css deleted file mode 100644 index a6dcf2585..000000000 --- a/standalone-frontend/style.css +++ /dev/null @@ -1,200 +0,0 @@ -* { - box-sizing: border-box; - margin: 0; - padding: 0; -} - -body { - font-family: 'Nunito', sans-serif; - font-weight: 400; - font-size: 100%; - background: #F1F1F1; -} - -*, html { - --primaryGradient: linear-gradient(93.12deg, #581B98 0.52%, #9C1DE7 100%); - --secondaryGradient: linear-gradient(268.91deg, #581B98 -2.14%, #9C1DE7 99.69%); - --primaryBoxShadow: 0px 10px 15px rgba(0, 0, 0, 0.1); - --secondaryBoxShadow: 0px -10px 15px rgba(0, 0, 0, 0.1); - --primary: #581B98; -} - -/* CHATBOX -=============== */ -.chatbox { - position: absolute; - bottom: 30px; - right: 30px; -} - -/* CONTENT IS CLOSE */ -.chatbox__support { - display: flex; - flex-direction: column; - background: #eee; - width: 300px; - height: 350px; - z-index: -123456; - opacity: 0; - transition: all .5s ease-in-out; -} - -/* CONTENT ISOPEN */ -.chatbox--active { - transform: translateY(-40px); - z-index: 123456; - opacity: 1; - -} - -/* BUTTON */ -.chatbox__button { - text-align: right; -} - -.send__button { - padding: 6px; - background: transparent; - border: none; - outline: none; - cursor: pointer; -} - - -/* HEADER */ -.chatbox__header { - position: sticky; - top: 0; - background: orange; -} - -/* MESSAGES */ -.chatbox__messages { - margin-top: auto; - display: flex; - overflow-y: scroll; - flex-direction: column-reverse; -} - -.messages__item { - background: orange; - max-width: 60.6%; - width: fit-content; -} - -.messages__item--operator { - margin-left: auto; -} - -.messages__item--visitor { - margin-right: auto; -} - -/* FOOTER */ -.chatbox__footer { - position: sticky; - bottom: 0; -} - -.chatbox__support { - background: #f9f9f9; - height: 450px; - width: 350px; - box-shadow: 0px 0px 15px rgba(0, 0, 0, 0.1); - border-top-left-radius: 20px; - border-top-right-radius: 20px; -} - -/* HEADER */ -.chatbox__header { - background: var(--primaryGradient); - display: flex; - flex-direction: row; - align-items: center; - justify-content: center; - padding: 15px 20px; - border-top-left-radius: 20px; - border-top-right-radius: 20px; - box-shadow: var(--primaryBoxShadow); -} - -.chatbox__image--header { - margin-right: 10px; -} - -.chatbox__heading--header { - font-size: 1.2rem; - color: white; -} - -.chatbox__description--header { - font-size: .9rem; - color: white; -} - -/* Messages */ -.chatbox__messages { - padding: 0 20px; -} - -.messages__item { - margin-top: 10px; - background: #E0E0E0; - padding: 8px 12px; - max-width: 70%; -} - -.messages__item--visitor, -.messages__item--typing { - border-top-left-radius: 20px; - border-top-right-radius: 20px; - border-bottom-right-radius: 20px; -} - -.messages__item--operator { - border-top-left-radius: 20px; - border-top-right-radius: 20px; - border-bottom-left-radius: 20px; - background: var(--primary); - color: white; -} - -/* FOOTER */ -.chatbox__footer { - display: flex; - flex-direction: row; - align-items: center; - justify-content: space-between; - padding: 20px 20px; - background: var(--secondaryGradient); - box-shadow: var(--secondaryBoxShadow); - border-bottom-right-radius: 10px; - border-bottom-left-radius: 10px; - margin-top: 20px; -} - -.chatbox__footer input { - width: 80%; - border: none; - padding: 10px 10px; - border-radius: 30px; - text-align: left; -} - -.chatbox__send--footer { - color: white; -} - -.chatbox__button button, -.chatbox__button button:focus, -.chatbox__button button:visited { - padding: 10px; - background: white; - border: none; - outline: none; - border-top-left-radius: 50px; - border-top-right-radius: 50px; - border-bottom-left-radius: 50px; - box-shadow: 0px 10px 15px rgba(0, 0, 0, 0.1); - cursor: pointer; -} diff --git a/static/app.js b/static/app.js index e69de29bb..628f9b846 100644 --- a/static/app.js +++ b/static/app.js @@ -0,0 +1,83 @@ +class Chatbox { + constructor() { + this.args = { + openButton: document.querySelector(".chatbox__button"), + chatBox: document.querySelector(".chatbox__support"), + sendButton: document.querySelector(".send__button"), + }; + this.state = false; + this.messages = []; + } + display() { + const { openButton, chatBox, sendButton } = this.args; + openButton.addEventListener("click", () => this.toggleState(chatBox)); + sendButton.addEventListener("click", () => this.onSendButton(chatBox)); + + const node = chatBox.querySelector("input"); + node.addEventListener("keyup", ({ key }) => { + if (key === "Enter") { + this.onSendButton(chatBox); + } + }); + } + toggleState(chatbox) { + this.state = !this.state; + if (this.state) { + chatbox.classList.add("chatbox--active"); + } else { + chatbox.classList.remove("chatbox--active"); + } + } + onSendButton(chatbox) { + let textField = chatbox.querySelector("input"); + let text1 = textField.value; + if (text1 === "") { + return; + } + let msg1 = { name: "User", message: text1 }; + this.messages.push(msg1); + + fetch($SCRIPT_ROOT + "/predict", { + method: "POST", + body: JSON.stringify({ message: text1 }), + mode: "cors", + headers: { "Content-Type": "application/json" }, + }) + .then((r) => r.json()) + .then((r) => { + let msg2 = { name: "SDF", message: r.answer }; + this.messages.push(msg2); + this.updateChatText(chatbox); + textField.value = ""; + }) + .catch((error) => { + console.error("Error: ", error); + this.updateChatText(chatbox); + textField.value = ""; + }); + } + updateChatText(chatbox) { + let html = ""; + this.messages + .slice() + .reverse() + .forEach(function (item) { + if (item.name === "SDF") { + html += + '
' + + item.message + + "
"; + } else { + html += + '
' + + item.message + + "
"; + } + }); + const chatmessage = chatbox.querySelector(".chatbox__messages"); + chatmessage.innerHTML = html; + } +} + +const chatbox = new Chatbox(); +chatbox.display(); diff --git a/templates/base.html b/templates/base.html index 80834196b..fecf57f15 100644 --- a/templates/base.html +++ b/templates/base.html @@ -16,7 +16,7 @@

Chat support

-

Hi. My name is Sam. How can I help you?

+

Hi. I'm SDF Virtual Agent. How can I help you?

diff --git a/train.py b/train.py index 3555f040a..0002a8b6d 100644 --- a/train.py +++ b/train.py @@ -29,7 +29,7 @@ xy.append((w, tag)) # stem and lower each word -ignore_words = ['?', '.', '!'] +ignore_words = ['?', '.', '!', ","] all_words = [stem(w) for w in all_words if w not in ignore_words] # remove duplicates and sort all_words = sorted(set(all_words)) @@ -53,7 +53,7 @@ X_train = np.array(X_train) y_train = np.array(y_train) -# Hyper-parameters +# Hyper-parameters num_epochs = 1000 batch_size = 8 learning_rate = 0.001 @@ -62,6 +62,7 @@ output_size = len(tags) print(input_size, output_size) + class ChatDataset(Dataset): def __init__(self): @@ -77,6 +78,7 @@ def __getitem__(self, index): def __len__(self): return self.n_samples + dataset = ChatDataset() train_loader = DataLoader(dataset=dataset, batch_size=batch_size, @@ -96,31 +98,31 @@ def __len__(self): for (words, labels) in train_loader: words = words.to(device) labels = labels.to(dtype=torch.long).to(device) - + # Forward pass outputs = model(words) # if y would be one-hot, we must apply # labels = torch.max(labels, 1)[1] loss = criterion(outputs, labels) - + # Backward and optimize optimizer.zero_grad() loss.backward() optimizer.step() - + if (epoch+1) % 100 == 0: - print (f'Epoch [{epoch+1}/{num_epochs}], Loss: {loss.item():.4f}') + print(f'Epoch [{epoch+1}/{num_epochs}], Loss: {loss.item():.4f}') print(f'final loss: {loss.item():.4f}') data = { -"model_state": model.state_dict(), -"input_size": input_size, -"hidden_size": hidden_size, -"output_size": output_size, -"all_words": all_words, -"tags": tags + "model_state": model.state_dict(), + "input_size": input_size, + "hidden_size": hidden_size, + "output_size": output_size, + "all_words": all_words, + "tags": tags } FILE = "data.pth"