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 ... content and the main answer.""" match = re.search(r"(.*?)", response_text, re.DOTALL) if match: thinking_part = match.group(1).strip() main_answer = re.sub(r".*?", "", 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)