Update app.py

app.py

@@ -111,27 +111,37 @@ def dataframe_info(data):
     value= data[:5]
     return str(value)
 
-def extract_python_code(text):
-    """
-    Extracts code blocks from a given text.
-    Supports triple-backtick blocks, indented code, and inline code.
-    """
-    code_snippets = []
+# def extract_python_code(text):
+#     """
+#     Extracts code blocks from a given text.
+#     Supports triple-backtick blocks, indented code, and inline code.
+#     """
+#     code_snippets = []
 
-    # Extract triple-backtick code blocks
-    triple_backtick_blocks = re.findall(r"```(?:[\w+\-]*)\n(.*?)```", text, re.DOTALL)
-    code_snippets.extend(triple_backtick_blocks)
+#     # Extract triple-backtick code blocks
+#     triple_backtick_blocks = re.findall(r"```(?:[\w+\-]*)\n(.*?)```", text, re.DOTALL)
+#     code_snippets.extend(triple_backtick_blocks)
 
-    # Extract indented code blocks (4 spaces or tab)
-    indented_blocks = re.findall(r"(?:^|\n)((?:    |\t).+(\n(?:    |\t).+)*)", text)
-    code_snippets.extend([block[0] for block in indented_blocks])
+#     # Extract indented code blocks (4 spaces or tab)
+#     indented_blocks = re.findall(r"(?:^|\n)((?:    |\t).+(\n(?:    |\t).+)*)", text)
+#     code_snippets.extend([block[0] for block in indented_blocks])
 
-    # Extract inline code snippets with single backticks
-    inline_code = re.findall(r"`([^`\n]+)`", text)
-    code_snippets.extend(inline_code)
+#     # Extract inline code snippets with single backticks
+#     inline_code = re.findall(r"`([^`\n]+)`", text)
+#     code_snippets.extend(inline_code)
 
-    return [snippet.strip() for snippet in code_snippets if snippet.strip()]
+#     return [snippet.strip() for snippet in code_snippets if snippet.strip()]
+def extract_python_code(text):
+    # Match triple backtick Python code blocks ```python ... ```
+    code_blocks = re.findall(r"```(?:python)?\n(.*?)```", text, re.DOTALL)
+    
+    # If no triple-backtick code found, try to match indented blocks (e.g. 4 spaces)
+    if not code_blocks:
+        code_blocks = re.findall(r"(?:^|\n)(    .+?)(?=\n\S|\Z)", text, re.DOTALL)
 
+    # Clean indentation if needed
+    code_blocks = [re.sub(r"^\s{4}", "", block, flags=re.MULTILINE) for block in code_blocks]
+    return code_blocks
 
 # @st.cache_resource
 def run_code_blocks(code_blocks,df):