Update app.py

app.py

@@ -24,29 +24,26 @@ def generate_text(prompt, max_length=100, min_length=20, temperature=1.0):
     return generated_text
 
 with gr.Blocks() as demo:
-
-    prompt_txt = gr.Textbox(label="Prompt", 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")
-
-    output_txt = gr.Textbox(value="", label="Generated Text")
-
-    submit_btn = gr.Button(value="Generate Text")
-    submit_btn.click(
-        generate_text,
-        inputs=[prompt_txt, max_len_slider, min_len_slider, temp_slider],
-        outputs=[output_txt]
-    )
-
-    gr.Markdown("## Text Generation Example")
-    gr.Examples(
-        examples=[["Write a creative story", 100, 20, 1.0]],
-        inputs=[prompt_txt, max_len_slider, min_len_slider, temp_slider],
-        outputs=output_txt,
-        fn=generate_text,
-        cache_examples=True,
-    )
+    # Left Sidebar
+    gr.Text("TinyLlama Text Generator", h=1, font_size=20)
+    prompt_txt = gr.Textbox(label="User:", lines=2, h=4)
+    max_len_slider = gr.Slider(0, 2048, 100, label="Max Length", h=2)
+    min_len_slider = gr.Slider(0, 2048, 20, label="Min Length", h=2)
+    temp_slider = gr.Slider(0.1, 2.0, 1.0, label="Temperature", h=2)
+
+    submit_btn = gr.Button(value="Submit", h=2)
+    
+    # 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], h=10)
 
 if __name__ == "__main__":
     demo.launch()