Update app.py

app.py

@@ -1,8 +1,6 @@
 import time
 import logging
 import gradio as gr
-import os
-os.environ.getattribute("OPENAI_API_KEY")
 
 from src.llm_boilers import llm_boiler
 
@@ -112,11 +110,30 @@ def call_inf_server(prompt, openai_key):
         logging.warning(f"Result of text generation: {response}")
         return response
 
-with gr.Blocks(theme=gr.themes.Glass(primary_hue=gr.themes.colors.blue, secondary_hue=gr.themes.colors.lime)) as demo:
-        conversation = Chat()
-    
-     chatbot = gr.Chatbot().style(height=200, background-color=blue)
-    with gr.Column():
+
+with gr.Blocks(
+    theme=gr.themes.Soft(),
+    css=".disclaimer {font-variant-caps: all-small-caps;}",
+) as demo:
+    gr.Markdown(
+        """<h1><center>Chat with gpt-3.5-turbo</center></h1>
+        This is a lightweight demo of gpt-3.5-turbo conversation completion. It was designed as a template for in-context learning applications to be built on top of.
+"""
+    )
+    conversation = Chat()
+    with gr.Row():
+        with gr.Column():
+            # to do: change to openaikey input for public release
+            openai_key = gr.Textbox(
+                label="OpenAI Key",
+                value="",
+                type="password",
+                placeholder="sk..",
+                info="You have to provide your own OpenAI API key.",
+            )
+    chatbot = gr.Chatbot().style(height=400)
+    with gr.Row():
+        with gr.Column():
             msg = gr.Textbox(
                 label="Chat Message Box",
                 placeholder="Chat Message Box",