Spaces:
Runtime error
Runtime error
| 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) |