Update app.py

app.py

@@ -69,11 +69,9 @@ h3 {
 }
 .message-wrap {
     overflow-x: auto;
-    white-space: pre-wrap !important;
 }
 .message-wrap p {
     margin-bottom: 1em;
-    white-space: pre-wrap !important;
 }
 .message-wrap pre {
     background-color: #f6f8fa;
@@ -120,7 +118,7 @@ def initialize_model():
     return model, tokenizer
 
 def format_text(text):
-    """Format text with proper spacing and tag highlighting"""
+    """Format text with proper spacing and tag highlighting (but keep tags visible)"""
     tag_patterns = [
         (r'<Thinking>', '\n<Thinking>\n'),
         (r'</Thinking>', '\n</Thinking>\n'),
@@ -141,13 +139,28 @@ def format_text(text):
     return formatted
 
 def format_chat_history(history):
-    """Format chat history for display in text area"""
+    """Format chat history for display, keeping tags visible"""
     formatted = []
     for user_msg, assistant_msg in history:
         formatted.append(f"User: {user_msg}")
         if assistant_msg:
             formatted.append(f"Assistant: {assistant_msg}")
     return "\n\n".join(formatted)
+    
+def create_examples():
+    """Create example queries for the UI"""
+    return [
+        "Explain the concept of artificial intelligence.",
+        "How does photosynthesis work?",
+        "What are the main causes of climate change?",
+        "Describe the process of protein synthesis.",
+        "What are the key features of a democratic government?",
+        "Explain the theory of relativity.",
+        "How do vaccines work to prevent diseases?",
+        "What are the major events of World War II?",
+        "Describe the structure of a human cell.",
+        "What is the role of DNA in genetics?"
+    ]
 
 @spaces.GPU()
 def chat_response(
@@ -161,7 +174,7 @@ def chat_response(
     top_k: int = 20,
     penalty: float = 1.2,
 ):
-    """Generate chat responses with proper tag handling"""
+    """Generate chat responses, keeping tags visible in the output"""
     conversation = [
         {"role": "system", "content": system_prompt}
     ]
@@ -199,7 +212,6 @@ def chat_response(
     )
     
     buffer = ""
-    current_line = ""
     
     with torch.no_grad():
         thread = Thread(target=model.generate, kwargs=generate_kwargs)
@@ -209,19 +221,11 @@ def chat_response(
         
         for new_text in streamer:
             buffer += new_text
-            current_line += new_text
+            formatted_buffer = format_text(buffer)
+            history[-1][1] = formatted_buffer
+            chat_display = format_chat_history(history)
             
-            if '\n' in current_line:
-                lines = current_line.split('\n')
-                current_line = lines[-1]
-                formatted_buffer = format_text(buffer)
-                history[-1][1] = formatted_buffer
-                chat_display = format_chat_history(history)
-                yield history, chat_display
-            else:
-                history[-1][1] = buffer
-                chat_display = format_chat_history(history)
-                yield history, chat_display
+            yield history, chat_display
 
 def process_example(example: str) -> tuple:
     """Process example query and return empty history and updated display"""