Update app/services/chat_service.py

app/services/chat_service.py

@@ -33,44 +33,32 @@ class ChatService:
             context = self.prepare_context(code_chunks)
             
             prompt = f"""You are an expert code assistant analyzing the {repository_name} repository.
-
 User Question: {query}
-
 Code Context:
 {context}
-
 RESPONSE FORMAT:
 Use clean, professional markdown formatting similar to GitHub README files:
-
 ## Main Answer
 [Direct answer to the user's question]
-
-## Implementation Overview  
+## Implementation Overview
 [High-level explanation of how the feature/system works]
-
 ### Key Components
 - **Component Name**: Brief description (`filename.py`, lines X-Y)
 - **Component Name**: Brief description (`filename.py`, lines X-Y)
-
 ### Technical Details
 [Detailed explanation with code references]
-
 When referencing code:
 - Use **bold** for important file names and concepts
 - Use `backticks` for functions, variables, classes, and code snippets
 - Reference specific files and line numbers: `filename.py` (lines X-Y)
 - Use > blockquotes for important insights or warnings
-
 ### How It Works
 1. **Step 1**: Description of first step
-2. **Step 2**: Description of second step  
+2. **Step 2**: Description of second step
 3. **Step 3**: Description of third step
-
 > **Key Insight**: Important observations about the implementation
-
 ## Additional Notes
 [Any limitations, missing information, or recommendations]
-
 REQUIREMENTS:
 - NO emojis - use clean text only
 - Be comprehensive and detailed
@@ -79,11 +67,25 @@ REQUIREMENTS:
 - Use proper markdown hierarchy (##, ###, -, >, etc.)
 - Focus on explanation rather than just code listing
 - Professional documentation style
-
 Your detailed markdown analysis:"""
             
             response = self.model.generate_content(prompt)
             
+            # --- START: BEST FIX ---
+            # Clean the response text to remove markdown code block wrappers
+            response_text = response.text
+            
+            if response_text.startswith("```markdown"):
+                response_text = response_text[len("```markdown"):]
+            elif response_text.startswith("```"):
+                response_text = response_text[len("```"):]
+                
+            if response_text.endswith("```"):
+                response_text = response_text[:-len("```")]
+                
+            response_text = response_text.strip() # Remove any leading/trailing whitespace
+            # --- END: BEST FIX ---
+
             sources = []
             for chunk in code_chunks:
                 sources.append({
@@ -95,7 +97,7 @@ Your detailed markdown analysis:"""
                 })
             
             return {
-                'response': response.text,
+                'response': response_text,  # Use the cleaned text
                 'sources': sources,
                 'context_chunks_used': len(code_chunks),
                 'repository_name': repository_name,