Update app.py

app.py

@@ -13,7 +13,7 @@ def respond(message, history, system_message, max_tokens, temperature, top_p, se
         # Create an InferenceClient for the selected model
         client = InferenceClient(model_list.get(selected_model, "HuggingFaceH4/zephyr-7b-beta"))
         
-        # Build conversation messages for the client - without any emojis in the history
+        # Build conversation messages for the client
         messages = [{"role": "system", "content": system_message}]
         for user_msg, assistant_msg in history:
             if user_msg:  # Only add non-empty messages
@@ -147,35 +147,11 @@ with gr.Blocks(css=css) as demo:
             
         # Main area: Chat interface
         with gr.Column(scale=3):
-            # Using a custom chatbot renderer to add emojis
             chatbot = gr.Chatbot(
                 label="Conversation", 
                 show_label=True,
-                height=400,
-                render=False  # We'll render the messages manually
+                height=400
             )
-            
-            # Custom renderer for chat messages
-            def render_chat(history):
-                result = []
-                for user_msg, bot_msg in history:
-                    if user_msg is None:
-                        user_msg = ""
-                    if bot_msg is None:
-                        bot_msg = ""
-                    
-                    # Add emojis to non-empty messages
-                    if user_msg and not user_msg.startswith("👤"):
-                        user_msg = f"👤 {user_msg}"
-                    if bot_msg and not bot_msg.startswith("🛡️"):
-                        bot_msg = f"🛡️ {bot_msg}"
-                        
-                    result.append((user_msg, bot_msg))
-                return result
-            
-            # Custom chatbot with emoji rendering
-            chatbot.render = render_chat
-            
             with gr.Row():
                 user_input = gr.Textbox(
                     placeholder="Type your message here...", 
@@ -194,8 +170,9 @@ with gr.Blocks(css=css) as demo:
     
     # Define functions for chatbot interactions
     def user(user_message, history):
-        # Return the user's message and add it to history
-        return "", history + [[user_message, None]]
+        # Add emoji to user message
+        user_message_with_emoji = f"👤 {user_message}"
+        return "", history + [[user_message_with_emoji, None]]
     
     def bot(history, system_message, max_tokens, temperature, top_p, selected_model):
         # Ensure there's history
@@ -204,11 +181,23 @@ with gr.Blocks(css=css) as demo:
             
         # Get the last user message from history
         user_message = history[-1][0]
+        # Remove emoji for processing if present
+        if user_message.startswith("👤 "):
+            user_message = user_message[2:].strip()
+        
+        # Process previous history to clean emojis
+        clean_history = []
+        for h_user, h_bot in history[:-1]:
+            if h_user and h_user.startswith("👤 "):
+                h_user = h_user[2:].strip()
+            if h_bot and h_bot.startswith("🛡️ "):
+                h_bot = h_bot[2:].strip()
+            clean_history.append([h_user, h_bot])
         
         # Call respond function with the message
         response_generator = respond(
             user_message, 
-            history[:-1],  # Pass history without the current message
+            clean_history,  # Pass clean history
             system_message, 
             max_tokens, 
             temperature, 
@@ -216,12 +205,12 @@ with gr.Blocks(css=css) as demo:
             selected_model
         )
         
-        # Update history as responses come in
+        # Update history as responses come in, adding emoji
         for response in response_generator:
-            history[-1][1] = response
+            history[-1][1] = f"🛡️ {response}"
             yield history
     
-    # Wire up the event chain - submit and send button work the same way
+    # Wire up the event chain
     user_input.submit(
         user,
         [user_input, chatbot],