Update app.py

app.py

@@ -24,51 +24,37 @@ def generate_text(prompt, max_length=100, min_length=20, temperature=1.0):
     return generated_text
 
 with gr.Blocks() as demo:
+    chat_history = []
 
-    chat_history = gr.Chatbot(
-        [],
+    chatbot = gr.Chatbot(
+        chat_history,
         elem_id="chatbot",
         bubble_full_width=False,
-        avatar_images=(None, None),
+        avatar_images=(None, None),  # You can add avatars if needed
     )
 
-    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
+    user_input = gr.Textbox(
+        scale=4,
+        show_label=False,
+        placeholder="Type your message and press enter...",
+        container=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)
+    def process_user_input(history, text):
+        history.append(("User", text))
+        return history, gr.Textbox(value="", interactive=False)
 
-    chat_history = chat_history.textbox(
-        add_prompt_and_generate,
-        demo.textbox_result,
-        scale=3,
-        show_label=False,
-        placeholder="Type a prompt and press enter...",
+    user_input_msg = user_input.submit(
+        process_user_input, [chat_history, user_input], [chatbot, user_input], queue=False
     )
 
+    def generate_response(history):
+        last_user_input = [msg for msg in history if msg[0] == "User"][-1][1]
+        generated_response = generate_text(last_user_input)
+        history.append(("Chatbot", generated_response))
+        return history
+
+    user_input_msg.then(generate_response, [chat_history], queue=False)
+
 if __name__ == "__main__":
     demo.launch()