Update code_extraction.py

code_extraction.py

@@ -0,0 +1,11 @@
+import re
+
+
+Class Extraction():
+
+    def extract_code(llm_output):
+        
+        code_match = re.search(r"```python\n(.*?)\n```", llm_output, re.DOTALL)
+        if code_match:
+            return code_match.group(1)
+        return None