Update app.py

app.py

@@ -46,40 +46,23 @@ def respond(
     except Exception as e:
         yield f"**Error:** {str(e)}"
 
-# Latest updates (this is just placeholder text)
-latest_updates = """
-**Chatbot - Latest Updates:**
 
-* Multiple Model Support
-* Improved Error Handling
-* Enhanced System Message Input
-"""
-
-def show_updates_and_respond(history, system_message, max_tokens, temperature, top_p, model_name):
+def update_chatbox(history, message, model_name, system_message, max_tokens, temperature, top_p):
     """
-    Shows the latest updates and generates a response from the model based on the updates.
+    Update the chat history and generate the next AI response.
     """
-    history.append(("User", "Show me the latest updates"))
-    yield from respond(
-        message="Show me the latest updates",
-        history=history,
-        system_message=system_message,
-        max_tokens=max_tokens,
-        temperature=temperature,
-        top_p=top_p,
-        model_name=model_name,
-    )
-    history[-1] = ("User", "Show me the latest updates")
-    history.append(("Assistant", latest_updates))
-    yield from respond(
-        message="What are the latest updates?",
+    history.append(("User", message))  # Add user message to history
+    ai_response = next(respond(
+        message=message,
         history=history,
         system_message=system_message,
         max_tokens=max_tokens,
         temperature=temperature,
         top_p=top_p,
-        model_name=model_name,
-    )
+        model_name=model_name
+    ))
+    history.append(("AI", ai_response))  # Add AI response to history
+    return history, ""  # Return updated history and clear the user input
 
 # Define the Gradio interface with the Blocks context
 with gr.Blocks(css=".gradio-container {border: none;}") as demo:
@@ -94,17 +77,17 @@ with gr.Blocks(css=".gradio-container {border: none;}") as demo:
     top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
     model_dropdown = gr.Dropdown(choices=list(available_models.keys()), value="Zephyr 7B Beta", label="Select Model")
     
-    # Define the chat interface
-    chatbot = gr.Chatbot()
+    chatbot = gr.Chatbot(label="Character-like AI Chat")
 
-    # Add the "Show Updates" button
-    updates_button = gr.Button("Show Latest Updates")
+    user_input = gr.Textbox(show_label=False, placeholder="Type your message here...")
+    send_button = gr.Button("Send")
 
-    # Button click event
-    updates_button.click(
-        fn=show_updates_and_respond,
-        inputs=[chat_history, system_message, max_tokens, temperature, top_p, model_dropdown],
-        outputs=chat_history
+    # When the send button is clicked, update chat history
+    send_button.click(
+        fn=update_chatbox,
+        inputs=[chat_history, user_input, model_dropdown, system_message, max_tokens, temperature, top_p],
+        outputs=[chatbot, user_input],  # Update chatbox and clear user input
+        queue=True  # Ensure responses are shown in order
     )
 
 # Launch the Gradio interface