remove history_state, add gpt-3.5-turbo-16k, add temperature

app/app.py

@@ -118,10 +118,14 @@ def generate_image(prompt):
 
 
 # Define a function to handle the chat interaction with the AI model
-def chat(model, system_message, message, chatbot_messages, history_state):
-    # Initialize chatbot_messages and history_state if they are not provided
+def chat(model, temperature, system_message, message, chatbot_messages):
+    # Initialize chatbot_messages
     chatbot_messages = chatbot_messages or []
-    history_state = history_state or []
+
+    history_openai_format = []
+    for human, assistant in chatbot_messages:
+        history_openai_format.append({"role": "user", "content": human})
+        history_openai_format.append({"role": "assistant", "content": assistant})
 
     # Try to get the AI's reply using the get_ai_reply function
     try:
@@ -129,7 +133,8 @@ def chat(model, system_message, message, chatbot_messages, history_state):
             message,
             model=model,
             system_message=system_message,
-            message_history=history_state,
+            message_history=history_openai_format,
+            temperature=temperature,
         )
     except Exception as e:
         # If an error occurs, raise a Gradio error
@@ -138,12 +143,8 @@ def chat(model, system_message, message, chatbot_messages, history_state):
     # Append the user's message and the AI's reply to the chatbot_messages list
     chatbot_messages.append((message, ai_reply))
 
-    # Append the user's message and the AI's reply to the history_state list
-    history_state.append({"role": "user", "content": message})
-    history_state.append({"role": "assistant", "content": ai_reply})
-
-    # Return None (empty out the user's message textbox), the updated chatbot_messages, and the updated history_state
-    return None, chatbot_messages, history_state
+    # Return None (empty out the user's message textbox), the updated chatbot_messages
+    return None, chatbot_messages
 
 
 # Define a function to launch the chatbot interface using Gradio
@@ -183,16 +184,20 @@ def get_chatbot_app(additional_examples=[]):
                         max_lines=400,
                     )
                 with gr.Column():
-                    # Create a dropdown to select the AI model
-                    model_selector = gr.Dropdown(
-                        ["gpt-3.5-turbo"], label="Model", value="gpt-3.5-turbo"
-                    )
+                    with gr.Row():
+                        # Create a dropdown to select the AI model
+                        model_selector = gr.Dropdown(
+                            ["gpt-3.5-turbo", "gpt-3.5-turbo-16k"],
+                            label="Model",
+                            value="gpt-3.5-turbo",
+                        )
+                        temperature_slider = gr.Slider(
+                            minimum=0, maximum=2, step=0.1, value=0
+                        )
                     # Create a chatbot interface for the conversation
                     chatbot = gr.Chatbot(label="Conversation")
                     # Create a textbox for the user's message
                     message = gr.Textbox(label="Message")
-                    # Create a state object to store the conversation history
-                    history_state = gr.State()
                     # Create a button to send the user's message
                     btn = gr.Button(value="Send")
 
@@ -200,19 +205,19 @@ def get_chatbot_app(additional_examples=[]):
                 example_load_btn.click(
                     choose_example,
                     inputs=[example_dropdown],
-                    outputs=[system_message, message, chatbot, history_state],
+                    outputs=[system_message, message, chatbot],
                 )
                 # Connect the send button to the chat function
                 btn.click(
                     chat,
                     inputs=[
                         model_selector,
+                        temperature_slider,
                         system_message,
                         message,
                         chatbot,
-                        history_state,
                     ],
-                    outputs=[message, chatbot, history_state],
+                    outputs=[message, chatbot],
                 )
         with gr.Tab("Image Generation"):
             image_prompt = gr.Textbox(