Improve UI further

app.py

@@ -24,7 +24,7 @@ Enter an OpenAI or Cohere key, choose your model and set the `max_length`.
 Built by [Bilge Yucel](https://twitter.com/bilgeycl) and [Stefano Fiorucci](https://github.com/anakin87), with [Haystack](https://github.com/deepset-ai/haystack).
 """
 
-with gr.Blocks() as demo:
+with gr.Blocks(theme="default") as demo:
     gr.Markdown(value=description)
     with gr.Row():
         api_key = gr.Textbox(label="Enter your api key", type="password")
@@ -38,21 +38,23 @@ with gr.Blocks() as demo:
             "Antibiotics are a type of medication used to treat bacterial infections. They work by either killing the bacteria or preventing them from reproducing, allowing the body’s immune system to fight off the infection. Antibiotics are usually taken orally in the form of pills, capsules, or liquid solutions, or sometimes administered intravenously. They are not effective against viral infections, and using them inappropriately can lead to antibiotic resistance. Explain the above in one sentence:",
             "Please give a sentiment for this context. Answer with positive, negative or neutral. Context: A flicker in the dark started of interesting and I was glued to the novel. It was just a little longer that I had anticipated to get to the bottom of the story. I felt sorry for the Chloe's mother but I had thought there was something odd about her brother. Well being a murderer was it any wonder he did not like this sister's boyfriend because I think he knew what happened. I love the cover of the book and the title is good. If only hardbacks had the cover printed onto them. Answer:",
         ],
+        examples_per_page=1,
         inputs=prompt,
-        label="Click on any example and press Submit!"
+        label="Click on any example"
         ) 
-    submit_btn = gr.Button("Submit")
+    submit_btn = gr.Button("✂️  Let's lemmatize and see!")
     with gr.Row():
         with gr.Column():
-            prompt_response = gr.TextArea(label="Output", show_copy_button=True)
             with gr.Row():
                 token_count_plain = gr.Number(label="Prompt Token Count")
                 token_count_plain_completion = gr.Number(label="Output Token Count")
+            with gr.Row():
+                prompt_response = gr.TextArea(label="Output", show_copy_button=True)
         with gr.Column():
-            lemmatized_prompt_response = gr.TextArea(label="Output (Lemmatized Prompt)", show_copy_button=True)
             with gr.Row():
                 token_count_lemmatized = gr.Number(label="Lemmatized Prompt Token Count")
                 token_count_lemmatized_completion = gr.Number(label="Output Token Count (Lemmatized Prompt)")
+            lemmatized_prompt_response = gr.TextArea(label="Output (Lemmatized Prompt)", show_copy_button=True)
             with gr.Accordion("See Lemmatized Prompt", open=False):
                 lemmatized_prompt = gr.TextArea(show_copy_button=True, show_label=False, container=False)