Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
-import os
 import time
 
 # Load model and tokenizer
@@ -32,11 +31,14 @@ def add_text(history, text):
     history = history + [(text, None)]
     return history, gr.Textbox(value="", interactive=False)
 
-def bot(history, max_length, min_length, temperature):
-    user_input = history[-1][0]
-    generated_response = generate_text(user_input, max_length=max_length, min_length=min_length, temperature=temperature)
-    history[-1][1] = generated_response
-    time.sleep(0.5)
+def add_file(history, file):
+    history = history + [((file.name,), None)]
+    return history
+
+def bot(history, max_len, min_len, temp):
+    prompt = history[-1][0]
+    response = generate_text(prompt, max_length=max_len, min_length=min_len, temperature=temp)
+    history[-1] = (prompt, response)
     yield history
 
 with gr.Blocks() as demo:
@@ -44,24 +46,25 @@ with gr.Blocks() as demo:
         [],
         elem_id="chatbot",
         bubble_full_width=False,
+        avatar_images=(None, None),  # You can add an avatar image if needed
     )
 
     with gr.Row():
-        txt = gr.Textbox(
+        prompt_txt = gr.Textbox(
             scale=4,
             show_label=False,
-            placeholder="Enter text and press enter, or upload an image",
+            placeholder="Enter text and press enter",
             container=False,
         )
 
-    max_length_slider = gr.Slider(minimum=0, maximum=2048, value=10, label="Max Length")
-    min_length_slider = gr.Slider(minimum=0, maximum=2048, value=1, label="Min Length")
-    temperature_slider = gr.Slider(minimum=0.1, maximum=2.0, value=1.0, label="Temperature")
+        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")
 
-    txt_msg = txt.submit(add_text, [chatbot, txt], [chatbot, txt], queue=False).then(
-        bot, chatbot, chatbot, api_name="bot_response", max_length=max_length_slider, min_length=min_length_slider, temperature=temperature_slider
+    txt_msg = prompt_txt.submit(add_text, [chatbot, prompt_txt], [chatbot, prompt_txt], queue=False).then(
+        bot, chatbot, max_len_slider, min_len_slider, temp_slider, api_name="bot_response"
     )
-    txt_msg.then(lambda: gr.Textbox(interactive=True), None, [txt], queue=False)
+    txt_msg.then(lambda: gr.Textbox(interactive=True), None, [prompt_txt], queue=False)
 
     chatbot.like(print_like_dislike, None, None)