Update

app.py

@@ -132,9 +132,20 @@ def extract_db_code(text):
     return [match.strip() for match in matches]
 
 def extract_from_code_block(text):
+    # First try to match text inside ```
     pattern = r'```(?:\w+)?\s?(.*?)```'
     match = re.search(pattern, text, re.DOTALL)
-    return match.group(1).strip() if match else ''
+    if match:
+        return match.group(1).strip()
+    
+    # If no match in ```, try to match SQL-like text
+    pattern = r'\((SELECT .*?)\)'
+    match = re.search(pattern, text, re.DOTALL)
+    if match:
+        return match.group(1).strip()
+
+    # Return None if no match found
+    return ""
 
 def generate_dummy_db(db_info, question):
     pre_prompt = """