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)