Update app.py

app.py

@@ -14,7 +14,7 @@ MAX_GPU_MEMORY = "60GiB"
 # --- Model Loading ---
 @GPU(memory=60)
 def load_model():
-    """Load the Qwen3-4B model without quantization for full precision."""
+    """Load the Q-year-old model without quantization for full precision."""
     print(f"Attempting to load model: {MODEL_ID} without quantization")
     try:
         pipe = pipeline(
@@ -142,7 +142,7 @@ def parse_llm_output(full_output, input_prompt_list):
     cleaned_text = re.sub(r'^<\/?s?>', '', cleaned_text).strip()
     cleaned_text = re.sub(r'^(assistant|ASSISTANT|System|SYSTEM)[:\s]*', '', cleaned_text).strip()
     if not cleaned_text and generated_text:
-        print("Wireturning original generation.")
+        print("Warning: Parsing resulted in empty string, returning original generation.")
         return generated_text
     if last_input_content and last_occurrence_index == -1:
         print("Warning: Could not find last input prompt in LLM output. Returning cleaned full output.")
@@ -266,7 +266,8 @@ class PersonaChat:
             self.messages = []
             self.enhanced_profile = ""
             status = f"Searching for information about {name}..."
-            yield status, "", "", [{"role": "system", "content": "Initializing persona creation..."}]
+            # Yield empty history for intermediate steps to avoid invalid messages
+            yield status, "", "", []
             search_results = search_person(name, context)
             if isinstance(search_results, str) and search_results.startswith("Error"):
                 error_msg = f"Failed to set persona: {search_results}"
@@ -276,14 +277,14 @@ class PersonaChat:
             if bio_text.startswith("Could not extract text"):
                 yield f"Warning: {bio_text}", "", "", [{"role": "system", "content": bio_text}]
             status = f"Creating enhanced profile for {name}..."
-            yield status, "", bio_text, [{"role": "system", "content": status}]
+            yield status, "", bio_text, []
             self.enhanced_profile = generate_enhanced_persona(name, bio_text, context)
             profile_for_prompt = self.enhanced_profile
             if self.enhanced_profile.startswith("Error enhancing profile"):
                 yield f"Warning: Could not enhance profile. Using basic info.", "", self.enhanced_profile, [{"role": "system", "content": self.enhanced_profile}]
                 profile_for_prompt = bio_text
             status = f"Generating optimal system prompt for {name}..."
-            yield status, self.enhanced_profile, self.enhanced_profile, [{"role": "system", "content": status}]
+            yield status, self.enhanced_profile, self.enhanced_profile, []
             self.system_prompt = generate_system_prompt_with_llm(name, profile_for_prompt, context)
             self.messages = [{"role": "system", "content": self.system_prompt}]
             yield f"Persona set to '{name}'. Ready to chat!", self.system_prompt, self.enhanced_profile, self.messages
@@ -326,7 +327,7 @@ def create_interface():
     .message-input { margin-top: 10px; }
     .send-button { background-color: #2c3e50 !important; color: white !important; }
     .persona-button { background-color: #4ca1af !important; color: white !important; }
-    .system-prompt-display { background-color: #f5f5f5; border-radius: 8px; padding: 15px; margin-top: 15px; border: 1px solid #e0e0e0; font-family: monospace; white-space: pre-wrap; word-wrap: break-word; }
+    .system-prompt-display { background-color: #f5f5f5; border-radius: 8px; padding: 15px; margin-top: 15px; border: 1px solid F#e0e0e0; font-family: monospace; white-space: pre-wrap; word-wrap: break-word; }
     .footer { text-align: center; margin-top: 20px; font-size: 0.9em; color: #666; }
     .typing-indicator { color: #aaa; font-style: italic; }
     """
@@ -372,13 +373,13 @@ def create_interface():
             initial_profile = "Enhanced profile will appear here..."
             initial_history = []
             yield initial_status, initial_prompt, initial_profile, initial_character_display, initial_history
-            final_status, final_prompt, final_profile = "Error", "", ""
-            final_history = initial_history
+            final_status, final_prompt, final_profile, final_character_display, final_history = "Error", "", "", f"### Error creating {name}", []
             try:
                 for status_update, prompt_update, profile_update, history_update in persona_chat.set_persona(name, context):
-                    final_status, final_prompt, final_profile = status_update, prompt_update, profile_update
-                    if isinstance(history_update, list):
-                        final_history = history_update
+                    final_status = status_update
+                    final_prompt = prompt_update
+                    final_profile = profile_update
+                    final_history = history_update if history_update else []
                     character_display = f"### Preparing chat with {name}..."
                     if "Ready to chat" in status_update:
                         character_display = f"### Chatting with {name}"