Update app.py

app.py

@@ -5,8 +5,6 @@ import os
 import markdown
 from markdown.extensions.codehilite import CodeHiliteExtension
 import markdown.extensions.fenced_code
-import bleach
-from html import escape
 
 API_URL = "https://host.palple.polrambora.com/pmsq"
 
@@ -119,15 +117,13 @@ def respond(message, api_key, max_tokens, top_p, temperature):
 
 
 
-
 def render_message(history):
     messages_html = """
     <div id="chatbox-container" class="chatbox" style="height: 400px; overflow: auto; border: 1px solid #262626; padding: 10px; background-color: #171717; display: flex; flex-direction: column-reverse;">
         <div id="messages" style="display: block; margin-bottom: 10px;">"""
 
-    seen_messages = set()  # Track (role, message) pairs to avoid duplicates
+    seen_messages = set() 
 
-    # Tags and attributes explicitly allowed
     allowed_tags = [
         'p', 'strong', 'em', 'ul', 'ol', 'li', 'a', 'code', 'pre', 'br', 'blockquote', 'hr', 
         'h1', 'h2', 'h3', 'h4', 'h5', 'h6', 'img'
@@ -138,13 +134,11 @@ def render_message(history):
         'img': ['src', 'alt', 'title', 'width', 'height']
     }
 
-    # Iterate through history and render each message
     for user_message, assistant_message, user_profile, assistant_profile, user_pic, assistant_pic in history:
         if user_message and ("user", user_message) not in seen_messages:
             seen_messages.add(("user", user_message))
-            # Render markdown -> sanitize output
             user_message_html = markdown.markdown(
-                user_message,  # Process markdown
+                user_message,  
                 extensions=["fenced_code", "codehilite"]
             )
             user_message_html = bleach.clean(
@@ -158,9 +152,8 @@ def render_message(history):
 
         if assistant_message and ("assistant", assistant_message) not in seen_messages:
             seen_messages.add(("assistant", assistant_message))
-            # Render markdown -> sanitize output
             assistant_message_html = markdown.markdown(
-                assistant_message,  # Process markdown
+                assistant_message,  
                 extensions=["fenced_code", "codehilite"]
             )
             assistant_message_html = bleach.clean(
@@ -181,7 +174,7 @@ def escape_html(unsafe_text):
     escaped_text = ''.join(f"&#{ord(char)};" if char not in ('\n', '\r') else '<br>' for char in unsafe_text)
     return escaped_text
 
-    
+
 css="""
 .chatbox {height: 400px; overflow: auto; border: 1px solid #262626; padding: 10px; background-color: #171717; display: flex;  flex-direction: column-reverse;}
 """
@@ -317,6 +310,6 @@ with gr.Blocks(css=css) as demo:
 
     auth_button.click(authorize_and_proceed, inputs=[api_user_input, api_key_input], outputs=[auth_view, chat_view, chatbot_output, history_state])
     save_instructions_btn.click(save_custom_instructions, inputs=[api_key_input, system_instructions_input], outputs=auth_status)
-    demo.launch(show_api=False)
+    demo.launch(show_api=True)
 if __name__ == "__main__":
     demo.queue = False