Spaces:
Running
Running
File size: 2,026 Bytes
e422038 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
"""
Utility functions for the vibe-reader application
"""
import json
import re
from typing import Dict, Any, Optional
def parse_json_response(response: str) -> Optional[Dict[str, Any]]:
"""
Parse JSON from LLM response, handling various formats
Args:
response: Raw LLM response that may contain JSON
Returns:
Parsed JSON dict, or None if parsing fails
"""
# Remove markdown code blocks if present
cleaned = re.sub(r'```json\s*|\s*```', '', response, flags=re.IGNORECASE)
cleaned = cleaned.strip()
# Try to find JSON object in the response
json_match = re.search(r'\{.*\}', cleaned, re.DOTALL)
if json_match:
try:
return json.loads(json_match.group(0))
except json.JSONDecodeError as e:
print(f"JSON parsing error: {e}")
return None
return None
def extract_vibe_components(vibe_json: Dict[str, Any]) -> Dict[str, Any]:
"""
Extract and validate vibe components from parsed JSON
Args:
vibe_json: Parsed JSON from vibe extraction
Returns:
Dictionary with validated vibe components
"""
return {
"aesthetic_genre_keywords": vibe_json.get("aesthetic_genre_keywords", []),
"mood_atmosphere": vibe_json.get("mood_atmosphere", []),
"core_themes": vibe_json.get("core_themes", []),
"tropes": vibe_json.get("tropes", []),
"feels_like": vibe_json.get("feels_like", "")
}
def strip_thinking_tags(text: str) -> str:
"""
Remove <think>...</think> tags and any reasoning content from text
Qwen3 uses standard XML format: <think>...</think>
Args:
text: Text that may contain thinking tags
Returns:
Clean text without thinking tags
"""
# Remove <think>...</think> blocks
cleaned = re.sub(r'<think>.*?</think>', '', text, flags=re.DOTALL | re.IGNORECASE)
# Remove any leftover tags
cleaned = re.sub(r'</?think>', '', cleaned, flags=re.IGNORECASE)
return cleaned.strip()
|