Upload 4 files

.gitattributes

@@ -32,3 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+data_with_ada_embedding.csv filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,41 @@
+# Imports
+import gradio as gr
+from helper_functions import *
+
+with gr.Blocks() as app:
+    gr.Markdown('# FundedNext Customer Service Chatbot')
+    with gr.Tab("Chat"):
+        with gr.Row():
+            with gr.Column():
+                msg = gr.Textbox()
+                with gr.Row():
+                    submit = gr.Button("Submit")
+                    clear = gr.Button("Clear")
+            with gr.Column():
+                chatbot = gr.Chatbot()
+            def user(user_message, history):
+                return "", history + [[user_message, None]]
+
+            def bot(history):
+                bot_message = get_reply(history[-1][0])
+                history[-1][1] = bot_message
+                return history
+
+            msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+                bot, chatbot, chatbot
+            ).then(
+                fn = reset_memory, inputs = None, outputs = None
+            )
+            submit.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+                bot, chatbot, chatbot
+            )
+            clear.click(fn = lambda: None, inputs = None, outputs = chatbot, queue=False).then(
+                fn = clear_variables, inputs = None, outputs = None, queue=False
+            )
+    with gr.Tab("Prompt"):
+        context = gr.Textbox()
+        submit = gr.Button("Check Prompt")
+        submit.click(get_context_gr, None, context, queue=False)
+app.launch(auth=("fundednext", "fundednext123"))
+
+# app.launch(debug=True)

data_with_ada_embedding.csv

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f7513e6259b03d34679c0d36c2ce8e75c4e81232e20c412f3bcd414ed92741c5
+size 19996727

helper_functions.py

@@ -0,0 +1,90 @@
+from openai.embeddings_utils import get_embedding, cosine_similarity
+import os
+import openai
+import pandas as pd
+import numpy as np
+
+# Set up OpenAI API key
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+# Load data
+df = pd.read_csv('data_with_ada_embedding.csv')
+df["token"] = df.combined_summarised.map(len)//4
+df['ada_embedding'] = df.ada_embedding.apply(eval).apply(np.array)
+
+pre_text = """You are a customer service agent of an app called FundedNext. FundedNext is a Proprietary Trading Firm aimed to assist traders with individual funding up to $200,000, backed by a solid strategy to minimise risks.
+Fundednext has two account models. Users can go for  Either Express Model or Evaluation Model, To get a real funded account. Each model has challenge phase and real phase. After sucessfully completing the challenge phase without violating any rules, users are eligible for their real trading account.
+Express model has two phases. Express Demo and Express Real. Express Demo is the challenge phase. Express users need to pass only one challenge phase to get to Express Real phase.
+While traders in the Evaluation model need to pass two challenge phases called Phase 1 and Phase 2. The final phase in Evaluation model is Evaluation Real.
+You are supposed to help the users of FundedNext with their questions and provide them with helpful answers.
+For each question, you will be given a context. You can use the context to answer the question. You can also use the context to ask follow up questions to the user. You should only answer the question if you are sure of the answer based on the provided context.
+"""
+
+def search(df, query, max_n, max_token):
+    query_embedding = get_embedding(
+        query,
+        engine="text-embedding-ada-002"
+    )
+    df["similarity"] = df.ada_embedding.apply(lambda x: cosine_similarity(x, query_embedding))
+    df = df.sort_values("similarity", ascending=False).head(max_n)
+    df["cumulative_sum"] = df.token.cumsum()
+    return '\n\n'.join(df[(df['cumulative_sum'] < max_token)]["combined_summarised"])
+
+def get_context(query):
+    results = search(df, query, max_n = 10, max_token = 500)
+    return f"""I will ask you questions based on the following context:
+— Start of Context —
+
+{results}
+
+— End of Context —
+My question is: “{query}”
+"""
+
+
+messages_archived = [
+    {"role": "system", "content": pre_text}
+]
+
+context = "Empty"
+messages_current = []
+
+def get_reply(message):
+    if message:
+        global messages_current
+        messages_current = messages_archived.copy()
+        global context
+        context = get_context(message)
+        messages_current.append(
+            {"role": "user", "content": context}
+        )
+        chat = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo", messages=messages_current, temperature=0
+        )
+    
+        reply = chat.choices[0].message.content
+        messages_archived.append({"role": "user", "content": message})
+        messages_archived.append({"role": "assistant", "content": reply})
+    # If no message is provided, return a string that says "No Message Received"
+    else:
+        reply = "No Message Received"
+
+    return reply
+
+def clear_variables():
+    global messages_archived
+    messages_archived = [
+        {"role": "system", "content": pre_text}
+    ]
+
+def reset_memory():
+    global messages_archived
+    if(len(messages_archived)>=21):
+        messages_archived = [
+        {"role": "system", "content": pre_text}
+        ]      
+    
+
+
+def get_context_gr():
+    return str(messages_current)

requirements.txt

@@ -0,0 +1,127 @@
+aiofiles==23.1.0
+aiohttp==3.8.4
+aiosignal==1.3.1
+altair==4.2.2
+anyio==3.6.2
+asttokens==2.2.1
+async-timeout==4.0.2
+attrs==22.2.0
+backcall==0.2.0
+backports.functools-lru-cache==1.6.4
+brotlipy==0.7.0
+certifi==2022.12.7
+cffi==1.15.1
+charset-normalizer==2.0.4
+click==8.1.3
+colorama==0.4.6
+contourpy==1.0.7
+cryptography==39.0.1
+cycler==0.11.0
+datasets==2.10.1
+debugpy==1.5.1
+decorator==5.1.1
+dill==0.3.6
+entrypoints==0.4
+et-xmlfile==1.1.0
+executing==1.2.0
+faiss==1.7.2
+fastapi==0.94.1
+ffmpy==0.3.0
+filelock==3.9.0
+flit_core==3.6.0
+fonttools==4.39.0
+frozenlist==1.3.3
+fsspec==2023.3.0
+gradio==3.21.0
+h11==0.14.0
+httpcore==0.16.3
+httpx==0.23.3
+huggingface-hub==0.13.1
+idna==3.4
+importlib-metadata==6.0.0
+importlib-resources==5.12.0
+ipykernel==6.15.0
+ipython==8.11.0
+jedi==0.18.2
+Jinja2==3.1.2
+joblib==1.2.0
+jsonschema==4.17.3
+jupyter_client==8.0.3
+jupyter_core==5.2.0
+kiwisolver==1.4.4
+linkify-it-py==2.0.0
+markdown-it-py==2.2.0
+MarkupSafe==2.1.2
+matplotlib==3.7.1
+matplotlib-inline==0.1.6
+mdit-py-plugins==0.3.3
+mdurl==0.1.2
+mkl-service==2.3.0
+multidict==6.0.4
+multiprocess==0.70.14
+nest-asyncio==1.5.6
+numpy==1.24.2
+openai==0.27.2
+openpyxl==3.1.2
+orjson==3.8.7
+packaging==23.0
+pandas==1.5.3
+parso==0.8.3
+pickleshare==0.7.5
+Pillow==9.4.0
+pip==23.0.1
+pkgutil_resolve_name==1.3.10
+platformdirs==2.5.2
+plotly==5.13.1
+prompt-toolkit==3.0.38
+psutil==5.9.0
+pure-eval==0.2.2
+pyarrow==11.0.0
+pycparser==2.21
+pydantic==1.10.6
+pydub==0.25.1
+Pygments==2.14.0
+pyOpenSSL==23.0.0
+pyparsing==3.0.9
+pyrsistent==0.19.3
+PySocks==1.7.1
+python-dateutil==2.8.2
+python-multipart==0.0.6
+pytz==2022.7.1
+pywin32==227
+PyYAML==6.0
+pyzmq==23.2.0
+regex==2022.10.31
+requests==2.28.1
+responses==0.18.0
+rfc3986==1.5.0
+scikit-learn==1.2.2
+scipy==1.10.1
+setuptools==65.6.3
+six==1.16.0
+sniffio==1.3.0
+stack-data==0.6.2
+starlette==0.26.1
+tenacity==8.2.2
+threadpoolctl==3.1.0
+tokenizers==0.13.2
+toolz==0.12.0
+torch==1.13.1
+torchaudio==0.13.1
+torchvision==0.14.1
+tornado==6.2
+tqdm==4.65.0
+traitlets==5.9.0
+transformers==4.26.1
+typing_extensions==4.4.0
+uc-micro-py==1.0.1
+urllib3==1.26.14
+uvicorn==0.21.0
+wcwidth==0.2.6
+websockets==10.4
+wheel==0.38.4
+win-inet-pton==1.1.0
+wincertstore==0.2
+xxhash==3.2.0
+yarl==1.8.2
+zipp==3.15.0