Duplicate from yf12/gpt3_chatbot

Co-authored-by: yangfan li <yf12@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz 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

README.md

@@ -0,0 +1,13 @@
+---
+title: Gpt3 Chatbot
+emoji: 🔥
+colorFrom: indigo
+colorTo: indigo
+sdk: gradio
+sdk_version: 3.16.2
+app_file: app.py
+pinned: false
+duplicated_from: yf12/gpt3_chatbot
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,22 @@
+import openai
+import gradio as gr
+import uuid
+import os
+from gpt_model import *
+
+openai.api_key = os.environ.get('secret_key')
+
+gpt = GPT_Model(append_output_prefix_to_query=True)
+
+gpt.add_example(Example('hi','Hello, how are you?'))
+gpt.add_example(Example('What is Neo Ivy Capital','The Neo Ivy Capital Fund is a quantitative hedge fund that invests in liquid, publicly traded equity securities via artificial intelligence strategies.'))
+gpt.add_example(Example('Who is the CEO of Neo Ivy Capital','The CEO of Neo Ivy Capital is Renee Yao'))
+
+def chatbot(input, history=[]):
+    output = gpt.submit_request(input)
+    history.append((input, output))
+    return history, history
+
+gr.Interface(fn = chatbot,
+             inputs = ["text",'state'],
+             outputs = ["chatbot",'state']).launch()

gpt_model.py

@@ -0,0 +1,93 @@
+import openai
+import uuid
+
+class Example:
+    def __init__(self, inp, out):
+        self.input = inp
+        self.output = out
+        self.id = uuid.uuid4().hex
+
+    def get_input(self):
+        return self.input
+
+    def get_output(self):
+        return self.output
+
+    def get_id(self):
+        return self.id
+
+    def as_dict(self):
+        return {
+            'input': self.get_input(),
+            'ouput': self.get_output(),
+            'id': self.get_id()
+        }
+
+class GPT_Model:
+    def __init__(self,
+                 engine = 'text-davinci-003',
+                 temperature = 0.5,
+                 max_tokens = 1024,
+                 input_prefix = 'input: ',
+                 input_suffix = '\n',
+                 output_prefix = 'output: ',
+                 output_suffix = '\n\n',
+                 append_output_prefix_to_query=False):
+        self.examples = {}
+        self.engine = engine
+        self.temperature = temperature
+        self.max_tokens = max_tokens
+        self.input_prefix = input_prefix
+        self.input_suffix = input_suffix
+        self.output_prefix = output_prefix
+        self.output_suffix = output_suffix
+        self.append_output_prefix_to_query = append_output_prefix_to_query
+
+    def add_example(self,ex):
+        assert isinstance(ex,Example), "Please create an example object"
+        self.examples[ex.get_id()] = ex
+
+    def delete_example(self,id):
+        if id in self.examples:
+            del self.examples[id]
+
+    def get_exmaple(self,id):
+        return self.examples.get(id,'Input id unavailable')
+
+    def get_all_example(self):
+        return {k:v.as_dict() for k,v in self.examples.items()}
+
+    def get_prime_text(self):
+        return "".join([self.format_example(ex) for ex in self.examples.values()])
+
+    def get_engine(self):
+        return self.engine
+
+    def get_temperature(self):
+        return self.temperature
+
+    def get_max_tokens(self):
+        return self.max_tokens
+
+    def craft_query(self, prompt):
+        q = self.get_prime_text() + self.input_prefix + prompt + self.input_suffix
+        if self.append_output_prefix_to_query:
+            q = q + self.output_prefix
+        return q
+
+    def submit_request(self, prompt):
+        completions = openai.Completion.create(
+        engine=self.get_engine(),
+        prompt=self.craft_query(prompt),
+        max_tokens=self.get_max_tokens(),
+        n=1,
+        temperature=self.get_temperature(),
+        )
+
+        message = completions.choices[0].text
+        return message.strip()
+
+    def format_example(self,ex):
+        return self.input_prefix+ex.get_input()+self.input_suffix+self.output_prefix+ex.get_output()+self.output_suffix
+
+    

requirements.txt

@@ -0,0 +1,3 @@
+openai
+gradio
+uuid