Update app.py

app.py

@@ -4,6 +4,7 @@ from duckduckgo_search import DDGS
 import re
 
 client = InferenceClient("Pinkstack/Superthoughts-lite-v1")
+
 def format_search_results(query, results, result_type):
     formatted = f"{result_type} search results for '{query}':\n"
     for i, result in enumerate(results):
@@ -67,6 +68,9 @@ def respond(
         if token:
             response += token
 
+    # Clean response: normalize line breaks to prevent extra <p> tags
+    response = re.sub(r'\n\s*\n+', '\n', response.strip())
+
     # Process response to convert tags to HTML divs
     formatted_response = response
     formatted_response = formatted_response.replace("<think>", '<div class="thinking">').replace("</think>", "</div>")
@@ -84,10 +88,13 @@ css = """
 }
 .output {
     background-color: #f0f0f0;
-    color: black;
+    color: black !important;
     padding: 15px;
     border-radius: 8px;
 }
+.output * {
+    color: black !important;
+}
 .gr-chatbot {
     background-color: #f5f5f5;
     font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
@@ -107,6 +114,9 @@ css = """
     align-self: flex-start;
     box-shadow: 0 1px 2px rgba(0,0,0,0.1);
 }
+.gr-chatbot .message.assistant * {
+    color: inherit;
+}
 """
 
 system_message_default = """
@@ -119,7 +129,7 @@ Your thought process here...
 <output>
 Your final answer here...
 
-If search results are provided, cite relevant sources at the end of this section as a numbered list in the format:
+If search results are provided, you must cite relevant sources at the end of this section as a numbered list in the format:
 1. [Title](URL)
 2. [Title](URL)
 </output>
@@ -129,7 +139,7 @@ demo = gr.ChatInterface(
     respond,
     additional_inputs=[
         gr.Textbox(value=system_message_default, label="System message", lines=8),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
+        gr.Slider(minimum=1, maximum=4096, value=2048, step=1, label="Max new tokens"),
         gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p"),
         gr.Dropdown(
@@ -141,7 +151,8 @@ demo = gr.ChatInterface(
     ],
     css=css,
     title="Pinkchat - Superthoughts lite v1 (Just 1.7B parameters!)",
-    description="Chat with an AI that thinks step-by-step and cites sources when searching the web. NEW:Added deep research."
+    description="""Chat with an AI that thinks step-by-step and has web search.
+    BETA:Added deep research."""
 )
 
 if __name__ == "__main__":