Update app.py

app.py

@@ -24,26 +24,51 @@ def generate_text(prompt, max_length=100, min_length=20, temperature=1.0):
     return generated_text
 
 with gr.Blocks() as demo:
-    # Left Sidebar
-    gr.Text("TinyLlama Text Generator")
-    prompt_txt = gr.Textbox(label="User:", lines=2)
-    max_len_slider = gr.Slider(0, 2048, 100, label="Max Length")
-    min_len_slider = gr.Slider(0, 2048, 20, label="Min Length")
-    temp_slider = gr.Slider(0.1, 2.0, 1.0, label="Temperature")
-
-    submit_btn = gr.Button(value="Submit")
-    
-    # Right Conversation Panel
-    chat_history = []
-
-    def respond(message, chat_history):
-        bot_message = generate_text(message, max_length=max_len_slider.value, min_length=min_len_slider.value, temperature=temp_slider.value)
-        chat_history.append((message, bot_message))
-        return "", chat_history
-
-    submit_btn.click(respond, [prompt_txt, chat_history], [prompt_txt, chat_history])
-
-    gr.Conversation([prompt_txt, max_len_slider, min_len_slider, temp_slider, submit_btn], [chat_history])
+
+    chat_history = gr.Chatbot(
+        [],
+        elem_id="chatbot",
+        bubble_full_width=False,
+        avatar_images=(None, None),
+    )
+
+    with gr.Row():
+        gr.Container(
+            [
+                gr.Textbox(
+                    label="Prompt",
+                    scale=3,
+                    show_label=False,
+                    placeholder="Enter your text prompt here...",
+                ),
+                gr.Slider(0, 2048, 100, label="Max Length"),
+                gr.Slider(0, 2048, 20, label="Min Length"),
+                gr.Slider(0.1, 2.0, 1.0, label="Temperature"),
+                gr.Button("Generate Text"),
+            ],
+            style="min-width:300px",
+        )
+
+    demo.textbox_result = demo.container_result = gr.Textbox(
+        value="", label="Generated Text", interactive=False
+    )
+
+    def generate_and_update(prompt, max_len, min_len, temp):
+        generated_text = generate_text(prompt, max_len, min_len, temp)
+        demo.textbox_result.value = generated_text
+        return demo.textbox_result
+
+    def add_prompt_and_generate(chat_history, prompt):
+        chat_history.append(((prompt,), None))
+        return generate_and_update(prompt, 100, 20, 1.0)
+
+    chat_history = chat_history.textbox(
+        add_prompt_and_generate,
+        demo.textbox_result,
+        scale=3,
+        show_label=False,
+        placeholder="Type a prompt and press enter...",
+    )
 
 if __name__ == "__main__":
     demo.launch()