Duplicate from leifive/chatgpt

Co-authored-by: Yuren Wu <[email protected]>

.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,14 @@
+---
+title: AI ChatGPT
+emoji: 📚
+colorFrom: green
+colorTo: pink
+sdk: gradio
+sdk_version: 3.23.0
+app_file: app.py
+pinned: false
+license: mit
+duplicated_from: leifive/chatgpt
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,45 @@
+import os
+import argparse
+
+import gradio as gr
+
+from chat_completion import ChatCompletion
+
+api_key = os.environ.get("OPENAI_API_KEY")
+
+parser = argparse.ArgumentParser()
+parser.add_argument('--share', action='store_true', default=False)
+parser.add_argument('--welcome', type=str, default='Say something to ChatGPT here ...')
+parser.add_argument('--title', type=str, default='ChatGPT')
+parser.add_argument('--setting', type=str, default=None)
+args = parser.parse_args()
+
+bot = ChatCompletion(api_key=api_key)
+
+with gr.Blocks(title=args.title) as demo:
+    chatbot = gr.Chatbot(show_label=False)
+    msg = gr.TextArea(show_label=False, placeholder=args.welcome)
+    send_btn = gr.Button('Send')
+    retry_btn = gr.Button('Retry')
+    reset_btn = gr.Button('Reset')
+
+    def send(user_message, history):
+        if not user_message:
+            return '', history
+
+        response = bot(user_message, setting=args.setting) if user_message != 'retry' else bot.retry()
+        return '', history + [[user_message, response]]
+
+    def reset():
+        bot.reset()
+        return None, [[None, None]]
+
+    def retry(history):
+        return send('retry', history)
+
+    send_btn.click(send, inputs=[msg, chatbot], outputs=[msg, chatbot], show_progress=True)
+    reset_btn.click(reset, inputs=None, outputs=[msg, chatbot])
+    retry_btn.click(retry, inputs=chatbot, outputs=[msg, chatbot])
+
+
+demo.launch(share=args.share)

chat_completion.py

@@ -0,0 +1,62 @@
+import linecache
+import re
+from typing import Dict, List, Optional
+
+import openai
+
+
+class ChatCompletion:
+    def __init__(self, model: str = 'gpt-3.5-turbo',
+                 api_key: Optional[str] = None, api_key_path: str = './openai_api_key'):
+        if api_key is None:
+            openai.api_key = api_key
+            api_key = linecache.getline(api_key_path, 2).strip('\n')
+            if len(api_key) == 0:
+                raise EnvironmentError
+        openai.api_key = api_key
+
+        self.model = model
+        self.system_messages = []
+        self.user_messages = []
+
+    def chat(self, msg: str, setting: Optional[str] = None, model: Optional[str] = None) -> str:
+        if self._context_length() > 2048:
+            self.reset()
+        if setting is not None:
+            if setting not in self.system_messages:
+                self.system_messages.append(setting)
+        if not self.user_messages or msg != self.user_messages[-1]:
+            self.user_messages.append(msg)
+
+        return self._run(model)
+
+    def retry(self, model: Optional[str] = None) -> str:
+        return self._run(model)
+
+    def reset(self):
+        self.system_messages.clear()
+        self.user_messages.clear()
+
+    def _make_message(self) -> List[Dict]:
+        sys_messages = [{'role': 'system', 'content': msg} for msg in self.system_messages]
+        user_messages = [{'role': 'user', 'content': msg} for msg in self.user_messages]
+        return sys_messages + user_messages
+
+    def _context_length(self) -> int:
+        return len(''.join(self.system_messages)) + len(''.join(self.user_messages))
+
+    def _run(self, model: Optional[str] = None) -> str:
+        if model is None:
+            model = self.model
+        try:
+            response = openai.ChatCompletion.create(model=model, messages=self._make_message())
+            ans = response['choices'][0]['message']['content']
+            ans = re.sub(r'^\n+', '', ans)
+        except openai.error.OpenAIError as e:
+            ans = e
+        except Exception as e:
+            print(e)
+        return ans
+
+    def __call__(self, msg: str, setting: Optional[str] = None, model: Optional[str] = None) -> str:
+        return self.chat(msg, setting, model)

requirements.txt

@@ -0,0 +1 @@
+openai