Update space

app.py

@@ -4,6 +4,47 @@ from huggingface_hub import InferenceClient
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
 """
+from new_chat import Conversation, ChatgptAPI
+
+chat_api = ChatgptAPI()
+
+def predict(system_input, password_input, user_in_file, user_input, conversation):
+    if password_input != '112233':
+       return [(None, "Wrong password!")], conversation, user_input
+
+    if conversation.is_initialized() == False:
+        conversation = Conversation(system_input, 5)
+    conversation = chat_api.get_single_round_completion(user_in_file, user_input, conversation)
+    return conversation, conversation, None
+    #_, conversation = chat_api.get_multi_round_completion(user_input, conversation)
+    #return conversation.get_history_messages(), conversation, None
+
+
+def clear_history(conversation):
+    conversation.clear()
+    return None, conversation
+
+
+with gr.Blocks(css="#chatbot{height:350px} .overflow-y-auto{height:600px}") as demo:
+    chatbot = gr.Chatbot(elem_id="chatbot")
+    conversation = gr.State(value=Conversation())
+
+    with gr.Row():
+        system_in_txt = gr.Textbox(lines=1, label="System role content:", placeholder="Enter system role content")
+        password_in_txt = gr.Textbox(lines=1, label="Password:", placeholder="Enter password")
+       
+    with gr.Row():
+        user_in_file = gr.File(label="Upload File")
+        user_in_txt = gr.Textbox(lines=3, label="User role content:", placeholder="Enter text...").style(container=False)
+    
+    with gr.Row():
+        submit_button = gr.Button("Submit")
+        reset_button = gr.Button("Reset")
+    
+    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)
+
+'''
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 
@@ -57,7 +98,7 @@ demo = gr.ChatInterface(
         ),
     ],
 )
-
+'''
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()

requirements.txt

@@ -1 +1,2 @@
-huggingface_hub==0.22.2
+openai
+huggingface_hub==0.22.2