Update app.py

app.py

@@ -18,9 +18,9 @@ def format_prompt(history):
 
 def generate(prompt, history=[], temperature=0.2, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0):
     # Add the first user message to history if it's not already there
-    if not history:
-        history.append((first_user_message, ""))
-
+    if not history and prompt:
+        history = [(first_user_message, "")]
+    
     temperature = float(temperature)
     if temperature < 1e-2:
         temperature = 1e-2
@@ -43,26 +43,30 @@ def generate(prompt, history=[], temperature=0.2, max_new_tokens=256, top_p=0.95
     output = ""
     for response in stream:
         output += response.token.text
-
+    
     # Adding the latest prompts and responses to history
-    history.append((prompt, output))
+    if prompt: # to avoid adding empty strings to history
+        history.append((prompt, output.strip()))
 
     return output, history  # Pass the updated history back to the interface
 
 # Setting up the Chat Interface
-mychatbot = gr.Chatbot(
-    avatar_images=["./user.png", "./botm.png"], bubble_full_width=False, show_label=False, show_copy_button=True, likeable=True,
-)
+inputs=[gr.Textbox(label="User Input", lines=2, placeholder="Enter your message here..."), gr.State()]
+outputs=[gr.Textbox(label="AI Response", lines=2, placeholder="AI will respond here..."), gr.State()]
+
+def reset():
+    return "", [(first_user_message, "")]
 
 demo = gr.Interface(
-    fn=generate, 
-    inputs=[gr.Textbox(label="User Input", lines=2, placeholder="Enter your message here..."), gr.State()],
-    outputs=[gr.Textbox(label="AI Response", lines=2, placeholder="AI will respond here..."), gr.State()],
+    fn=generate,
+    inputs=inputs,
+    outputs=outputs,
     title="Mixtral 8x7b Chat",
     theme="default",
-    initial_state=[(first_user_message, "")],  # Initialize the history with the first message
     allow_screenshot=False,
-    allow_flagging="never"
+    allow_flagging="never",
+    live=True,
+    event_handlers=[("load", reset)]
 )
 
 demo.launch()