Update question_handler.py

question_handler.py

@@ -26,3 +26,26 @@ def generate_detailed_prompt(question_metadata):
         f"\nPlease create a real-world interview question based on this information. "
         f"Include sections for problem description, code template, sample input, and expected output."
     )
+def extract_code_and_test_case(response):
+    import re
+    
+    code_template = ""
+    sample_test_case = ""
+    expected_output = ""
+
+    # Extract Python code block
+    code_match = re.search(r'```python(.*?)```', response, re.DOTALL)
+    if code_match:
+        code_template = code_match.group(1).strip()
+    
+    # Extract test case and expected output
+    test_case_match = re.search(r'Sample Input:\s*(.*?)\n', response, re.DOTALL)
+    expected_output_match = re.search(r'Expected Output:\s*(.*?)\n', response, re.DOTALL)
+
+    if test_case_match:
+        sample_test_case = test_case_match.group(1).strip()
+    
+    if expected_output_match:
+        expected_output = expected_output_match.group(1).strip()
+
+    return code_template, sample_test_case, expected_output