minor changes

app.py

@@ -4,35 +4,36 @@ from langchain.chat_models import ChatOpenAI
 from langchain.chains import ConversationChain
 from langchain.memory import ConversationBufferMemory
 
-
-def respond(openai_api_key, message, buffer_memory, chat_history):
+def respond(openai_api_key, openai_model, message, buffer_memory, chat_history):
     conversation = ConversationChain(
-        llm = ChatOpenAI(temperature=1.0, model='gpt-3.5-turbo'), 
-        memory = buffer_memory,
-        openai_api_key = openai_api_key
+        llm = ChatOpenAI(
+            temperature=0.0, 
+            model=openai_model, 
+            openai_api_key=openai_api_key
+        ), 
+        memory = buffer_memory
     )
     response = conversation.predict(input=message)
     chat_history.append([message, response])
     return "", buffer_memory, chat_history
 
 
-with gr.Blocks() as demo:
+with gr.Blocks(css="#component-0 { max-width: 900px; margin: auto; padding-top: 1.5rem; }") as demo:
     # with gr.Column():
-    with gr.Group(visible=True) as primary_settings:
-        with gr.Row():
-            openai_key = gr.Textbox(
-                label="OpenAI Key",
-                type="password",
-                placeholder="sk-a83jv6fn3x8ndm78b5W..."
-            )
-            model = gr.Dropdown(
-                ["gpt-4", "gpt-4-32k", 
-                 "gpt-3.5-turbo", "gpt-3.5-turbo-16k", "gpt-3.5-turbo-instruct", 
-                 "text-davinci-002", "text-davinci-003"], 
-                label="OpenAI Model",
-                value="gpt-3.5-turbo",
-                interactive=True
-            )
+    with gr.Row(visible=True) as primary_settings:
+        openai_key = gr.Textbox(
+            label="OpenAI Key",
+            type="password",
+            placeholder="sk-a83jv6fn3x8ndm78b5W..."
+        )
+        model = gr.Dropdown(
+            ["gpt-4", "gpt-4-32k", 
+                "gpt-3.5-turbo", "gpt-3.5-turbo-16k", "gpt-3.5-turbo-instruct", 
+                "text-davinci-002", "text-davinci-003"], 
+            label="OpenAI Model",
+            value="gpt-3.5-turbo",
+            interactive=True
+        )
     # with gr.Accordion("Advances Settings"):
     #     gr.Dropdown(
     #         [-1, 1, 5, 10, 25], label="Conversation Buffer (k)"
@@ -53,8 +54,8 @@ with gr.Blocks() as demo:
                         min_width=0)
 
     # Event Handling
-    query.submit(respond, [openai_key, query, memory, chatbot], [query, memory, chatbot])
-    submit.click(respond, [openai_key, query, memory, chatbot], [query, memory, chatbot])
+    query.submit(respond, [openai_key, model, query, memory, chatbot], [query, memory, chatbot])
+    submit.click(respond, [openai_key, model, query, memory, chatbot], [query, memory, chatbot])
     
 if __name__ == "__main__":
     demo.launch()