Update annotated_casl_app.py

annotated_casl_app.py

@@ -28,8 +28,15 @@ def clean_output_formatting(text):
     text = text.replace('**', '')                    # Remove any remaining **
     text = text.replace('*', '')                     # Remove any remaining *
     
-    # Remove hashtags from headers
-    text = re.sub(r'^#{1,6}\s*', '', text, flags=re.MULTILINE)  # Remove ### headers
+    # Remove hashtags from headers (at start of line)
+    text = re.sub(r'^#{1,6}\s+', '', text, flags=re.MULTILINE)  # Remove ### headers
+    
+    # Also remove hashtags that appear mid-sentence or in other positions
+    text = re.sub(r'\s#{1,6}\s+', ' ', text)  # Remove hashtags with spaces around them
+    text = re.sub(r'#{1,6}([A-Z])', r'\1', text)  # Remove # before capitalized words like #SECTION
+    text = re.sub(r'^#{1,6}$', '', text, flags=re.MULTILINE)  # Remove standalone hashtags on a line
+    text = re.sub(r'#([a-zA-Z])', r'\1', text)  # Remove # before any word
+    text = text.replace('#', '')  # Remove any remaining bare # symbols
     
     # Convert tables to lists - more comprehensive approach
     lines = text.split('\n')
@@ -168,7 +175,10 @@ def call_claude_api_quick_analysis(prompt):
         
         if response.status_code == 200:
             response_json = response.json()
-            return response_json['content'][0]['text']
+            raw_response = response_json['content'][0]['text']
+            # Clean formatting from response (removes **, #, tables)
+            cleaned_response = clean_output_formatting(raw_response)
+            return cleaned_response
         else:
             logger.error(f"Claude API error: {response.status_code} - {response.text}")
             return f"Error: Claude API Error: {response.status_code}"