Update app.py

app.py

@@ -8,7 +8,7 @@ from get_prompts import get_prompt_for_mode
 from dotenv import load_dotenv, set_key
 from milvus import initialize_milvus, DEFAULT_MILVUS_HOST, DEFAULT_MILVUS_PORT, DEFAULT_COLLECTION_NAME, DEFAULT_DIMENSION, DEFAULT_MAX_RETRIES, DEFAULT_RETRY_DELAY
 from pymilvus import connections, MilvusException, utility
-import markdown  # Import the markdown library
+import markdown
 
 # --- Configuration and Setup ---
 
@@ -206,28 +206,24 @@ def chat_ui(query, history, mode):
     if response is None or not response.strip():
         response = "An error occurred during processing. Please check the logs."
 
-    # Convert the response to Markdown format
-    md = markdown.Markdown(extensions=['fenced_code'])
-    formatted_response = ""  # Initialize formatted_response here
+    formatted_response = ""
 
     if mode == "developer":
         extracted_files = extract_files_from_response(response)
-
-        # Iterate through the extracted files and format them
         for filepath, content in extracted_files.items():
             formatted_response += f"## {filepath}\n`\n{content}\n`\n\n"
-
-        # Remove the last extra newline characters if they exist
-        if formatted_response.endswith("\n\n"):
-            formatted_response = formatted_response[:-2]
     else:
-        # Format the output for non-developer modes
         formatted_response = response
+    
+    # Use HTML directly for code blocks to preserve formatting
+    if mode == "developer":
+        formatted_response = formatted_response.replace("`\n", "<pre><code style=\"white-space: pre-wrap;\">").replace("\n`", "</code></pre>")
+    else:
+        # Convert the entire response to HTML at once
+        md = markdown.Markdown(extensions=['fenced_code'])
+        formatted_response = md.convert(formatted_response)
 
-    # Convert the entire response to HTML at once
-    formatted_response = md.convert(formatted_response)
     history.append((query, formatted_response))
-
     return history, history
 # ZIP Processing Interface
 zip_iface = gr.Interface(