update layout and fixed chatgpt inference code

app.py

@@ -33,7 +33,7 @@ def predict_chatgpt(inputs, top_p_chatgpt, temperature_chatgpt, openai_api_key,
     #Handling the different roles for ChatGPT
     if chat_counter_chatgpt != 0 :
         messages=[]
-        for data in chatbot:
+        for data in chatbot_chatgpt:
           temp1 = {}
           temp1["role"] = "user" 
           temp1["content"] = data[0] 
@@ -75,7 +75,7 @@ def predict_chatgpt(inputs, top_p_chatgpt, temperature_chatgpt, openai_api_key,
         if chunk.decode() :
           chunk = chunk.decode()
           # decode each line as response data is in bytes
-          if len(chunk) > 12 and "delta" in json.loads(chunk[6:])['choices'][0]:
+          if len(chunk) > 12 and "content" in json.loads(chunk[6:])['choices'][0]["delta"]:
               partial_words = partial_words + json.loads(chunk[6:])['choices'][0]["delta"]["content"]
               if token_counter == 0:
                 history.append(" " + partial_words)
@@ -163,7 +163,8 @@ def reset_chat(chatbot, state):
     return None, []
 
 
-title = """<h1 align="center">🚀ChatGPT & 🔥OpenChatKit Comparison Gradio Demo</h1>"""
+title = """<h1 align="center">🔥🔥Comparison: ChatGPT & OpenChatKit </h1><br>
+          <h3 align="center">🚀A Gradio Streaming Demo</h3>"""
 description = """Language models can be conditioned to act like dialogue agents through a conversational prompt that typically takes the form:
 ```
 User: <utterance>
@@ -186,10 +187,10 @@ with gr.Blocks(css="""#col_container {width: 1000px; margin-left: auto; margin-r
               with gr.Row():
                   with gr.Column(scale=13):
                     openai_api_key = gr.Textbox(type='password', label="Enter your OpenAI API key here for ChatGPT")
-                    inputs = gr.Textbox(placeholder="Hi there!", label="Type an input and press Enter" )
+                    inputs = gr.Textbox(placeholder="Hi there!", label="Type an input and press Enter ⤵️ " )
                   with gr.Column(scale=1):
-                      #b1 = gr.Button(elem_id = 'run')
-                      b2 = gr.Button('Clear up Chatbots!', elem_id = 'clear').style(full_width=True)
+                      b1 = gr.Button(elem_id = '🏃Run')
+                      b2 = gr.Button('🔄Clear up Chatbots!', elem_id = 'clear') #.style(full_width=True)
                   state_chatgpt = gr.State([])
                   state_together = gr.State([])
 
@@ -200,7 +201,7 @@ with gr.Blocks(css="""#col_container {width: 1000px; margin-left: auto; margin-r
           
       with gr.Column(scale=2, elem_id='parameters'):
           with gr.Box():
-              gr.HTML("Parameters for #OpenChatKit")
+              gr.HTML("Parameters for #OpenCHAtKit")
               top_p = gr.Slider(minimum=-0, maximum=1.0,value=0.95, step=0.05,interactive=True, label="Top-p",)
               temperature = gr.Slider(minimum=-0, maximum=5.0, value=0.5, step=0.1, interactive=True, label="Temperature", )
               top_k = gr.Slider( minimum=1, maximum=50, value=4, step=1, interactive=True, label="Top-k",)
@@ -225,9 +226,15 @@ with gr.Blocks(css="""#col_container {width: 1000px; margin-left: auto; margin-r
     inputs.submit( predict_together,
                 [temp_textbox_together, inputs, top_p, temperature, top_k, repetition_penalty, watermark, chatbot_together, state_together, ],
                 [chatbot_together, state_together],)
+    b1.click( predict_chatgpt, 
+                [inputs, top_p_chatgpt, temperature_chatgpt, openai_api_key, chat_counter_chatgpt, chatbot_chatgpt, state_chatgpt], 
+                [chatbot_chatgpt, state_chatgpt, chat_counter_chatgpt],)
+    b1.click( predict_together,
+                [temp_textbox_together, inputs, top_p, temperature, top_k, repetition_penalty, watermark, chatbot_together, state_together, ],
+                [chatbot_together, state_together],)
 
     b2.click(reset_chat, [chatbot_chatgpt, state_chatgpt], [chatbot_chatgpt, state_chatgpt])
     b2.click(reset_chat, [chatbot_together, state_together], [chatbot_together, state_together])
     
     gr.Markdown(description)
-demo.queue(concurrency_count=16).launch(height= 2500, debug=True)
+    demo.queue(concurrency_count=16).launch(height= 2500, debug=True)