utils/helpers.py

import os
import re
import time

def get_file_size(file):
    file.seek(0, os.SEEK_END)
    size = file.tell()
    file.seek(0)
    return size

def extract_thoughts(response_text):
    """Extracts <think>...</think> content and the main answer."""
    match = re.search(r"<think>(.*?)</think>", response_text, re.DOTALL)
    if match:
        thinking_part = match.group(1).strip()
        main_answer = re.sub(r"<think>.*?</think>", "", response_text, flags=re.DOTALL).strip()
    else:
        thinking_part = None
        main_answer = response_text.strip()
    
    return thinking_part, main_answer

# Streamed response emulator
def response_generator(response):
    for word in response.split():
        yield word + " "
        time.sleep(0.05)