fix app and add parameters

app.py

@@ -2,11 +2,14 @@ import gradio as gr
 import openai
 
 
-def get_answer_chatgpt(chat_history, message, system_message=None):
+def get_answer_chatgpt(chat_history, message, system_message=None, model="gpt-3.5-turbo", max_length=1000):
     messages = []
-    if system_message:
+    if model is None:
+        model = "gpt-3.5-turbo"
+
+    if system_message or system_message != "":
         TEMPLATE_SYSTEM = {"role": "system", "content": system_message}
-        message.append(TEMPLATE_SYSTEM)
+        messages.append(TEMPLATE_SYSTEM)
 
     TEMPLATE_USER = {"role": "user"}
     TEMPLATE_BOT = {"role": "assistant"}
@@ -17,13 +20,13 @@ def get_answer_chatgpt(chat_history, message, system_message=None):
         TEMPLATE_BOT["content"] = pair[1]
         messages.append(TEMPLATE_USER)
         messages.append(TEMPLATE_BOT)
-
     # add latest input
     TEMPLATE_USER["content"] = message
     messages.append(TEMPLATE_USER)
     response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        messages=messages
+        model=model,
+        messages=messages,
+        max_tokens=max_length
     )
 
     return {
@@ -38,31 +41,37 @@ def reset_textbox():
 
 
 with gr.Blocks() as demo:
-    gr.Markdown() 
-    system_prompt = gr.Textbox(placeholder="You are helpful assistence", lines=5, label="System Message")
+    with gr.Row():
+        with gr.Column(scale=5):
+            clear = gr.Button("Clear all converstation")
+        with gr.Column(scale=5):
+            with gr.Accordion("Parameters", open=False):
+                model = gr.Dropdown(["gpt-3.5-turbo", "gpt-4"], label="Model", value="gpt-3.5-turbo")
+                max_length = gr.Slider(0, 10000, value=1000, step=1, label="Maximum length token", interactive=True)
+
+    system_message = gr.Textbox(placeholder="You are helpful assistence", lines=5, label="System Message")
     chatbot = gr.Chatbot()
     message = gr.Textbox(placeholder="Halo kak, aku mau bertanya", label="Chat Here")
     with gr.Row():
         with gr.Column(scale=5):
-            clear = gr.Button("Clear")
+            send = gr.Button("Send")
         with gr.Column(scale=5):
             retry = gr.Button("Retry")
-    send = gr.Button("Send")
 
-    def respond(chat_history, message):
-        response = get_answer_chatgpt(chat_history, message)
+    def respond(chat_history, message, system_message, model, max_length=1000):
+        response = get_answer_chatgpt(chat_history, message, system_message, model, max_length)
         response = response["completion"]
 
         return chat_history + [[message, response]]
 
-    message.submit(respond, [chatbot, message], chatbot)
+    message.submit(respond, [chatbot, message, system_message, model, max_length], chatbot)
 
     clear.click(lambda: None, None, chatbot, queue=False)
-    send.click(respond, [chatbot, message], chatbot)
+    send.click(respond, [chatbot, message, system_message, model, max_length], chatbot)
     send.click(reset_textbox, [], [message])
     message.submit(reset_textbox, [], [message])
 
-    retry.click(respond, [chatbot, message], chatbot)
+    retry.click(respond, [chatbot, message, system_message, model, max_length], chatbot)
     retry.click(reset_textbox, [], [message])
 
     demo.launch(debug=True)