Update button

app.py

@@ -38,7 +38,7 @@ class ChatgptAPI:
         )
         message=completion.choices[0].message.content
         conversation.append_answer(message)
-        return conversation
+        return message, conversation
         
     
     def get_multi_round_completion(self, prompt, conversation, model='gpt-3.5-turbo'):
@@ -58,7 +58,7 @@ class ChatgptAPI:
         return message, conversation
 
 class Conversation:
-    def __init__(self, system_prompt='', num_of_round = 5):
+    def __init__(self, system_prompt='iii', num_of_round = 5):
         self.num_of_round = num_of_round
         self.history = []
         self.initialized = False
@@ -103,7 +103,7 @@ def predict(system_input, password_input, user_in_file, user_input, conversation
 
     if conversation.is_initialized() == False:
         conversation = Conversation(system_input, 5)
-    conversation = chat_api.get_single_round_completion(user_in_file, user_input, conversation)
+    _, conversation = chat_api.get_single_round_completion(user_in_file, user_input, conversation)
     return conversation.get_history_messages(), conversation, None
     #_, conversation = chat_api.get_multi_round_completion(user_input, conversation)
     #return conversation.get_history_messages(), conversation, None
@@ -127,8 +127,8 @@ with gr.Blocks(css="#chatbot{height:350px} .overflow-y-auto{height:600px}") as d
         user_in_txt = gr.Textbox(lines=3, label="User role content:", placeholder="Enter text...", container=False)
     
     with gr.Row():
-        submit_button = gr.Button("Submit")
-        reset_button = gr.Button("Reset")
+        submit_button = gr.Button("Reset")
+        reset_button = gr.Button("Submit")
     
     submit_button.click(predict, [system_in_txt, password_in_txt, user_in_file, user_in_txt, conversation], [chatbot, conversation, user_in_txt])
     reset_button.click(clear_history, [conversation], [chatbot, conversation], queue=False)

new_chat.py

@@ -0,0 +1,82 @@
+from pathlib import Path
+from openai import OpenAI
+
+class ChatgptAPI:
+    def __init__(self, ):
+        self.client = OpenAI(
+            api_key = "sk-u8YI0ArRHFRRdMEdboouRAXVc3PpR6EhZOfxO4tST5Ua9147",
+            base_url = "https://api.moonshot.cn/v1",
+        )
+    def get_single_round_completion(self, file_path, prompt, conversation):
+        file_object = client.files.create(file=Path(file_path), purpose="file-extract")
+        file_content = client.files.content(file_id=file_object.id).text
+        messages = [
+            {
+                "role": "system",
+                "content": "���� Kimi���� Moonshot AI �ṩ���˹��������֣�����ó����ĺ�Ӣ�ĵĶԻ������Ϊ�û��ṩ��ȫ���а�����׼ȷ�Ļش�ͬʱ�����ܾ�һ���漰�ֲ����壬�������ӣ���ɫ����������Ļش�Moonshot AI Ϊר�����ʣ����ɷ�����������ԡ�",
+            },
+            {
+                "role": "system",
+                "content": file_content,
+            },
+            {"role": "user", "content": prompt},
+        ]
+        completion = self.client.chat.completions.create(
+            model="moonshot-v1-32k",
+            messages=messages,
+            temperature=0.3,
+        )
+        return completion.choices[0].message
+        
+    
+    def get_multi_round_completion(self, prompt, conversation, model='gpt-3.5-turbo'):
+        conversation.append_question(prompt)
+        prompts = conversation.get_prompts()
+
+        response = openai.ChatCompletion.create(
+            model=model,
+            messages=prompts,
+            temperature=0,
+            max_tokens=2048,
+            top_p=1,
+        )
+        message = response.choices[0].message['content']
+        conversation.append_answer(message)
+
+        return message, conversation
+
+class Conversation:
+    def __init__(self, system_prompt='', num_of_round = 5):
+        self.num_of_round = num_of_round
+        self.history = []
+        self.initialized = False
+        self.history.append({"role": "system", "content": system_prompt})
+
+        if len(system_prompt) > 0:
+            logger.info(f'Conversation initialized with system prompt: {system_prompt}')
+            self.initialized = True
+
+    def is_initialized(self):
+        return self.initialized
+    
+    def append_question(self, question):
+        self.history.append({"role": "user", "content": question})
+
+    def append_answer(self, answer):
+        self.history.append({"role": "assistant", "content": answer})
+
+        if len(self.history) > self.num_of_round * 2:
+            del self.history[1:3]
+
+    def clear(self):
+        self.history.clear()
+        self.initialized = False
+
+    def get_prompts(self):
+        return self.history
+    
+    def round_size(self):
+        return 0 if len(self.history) < 2 else len(self.hitory) - 1
+    
+    def get_history_messages(self):
+        return [(u['content'], b['content']) for u,b in zip(self.history[1::2], self.history[2::2])]