chandini2595 commited on
Commit
5f25ca2
·
1 Parent(s): 210b4bb
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -57,7 +57,12 @@ def load_model():
57
  return processor, model
58
 
59
  def extract_json_from_llm_output(llm_result):
60
- match = re.search(r'\{.*\}', llm_result, re.DOTALL)
 
 
 
 
 
61
  if match:
62
  return match.group(0)
63
  return None
 
57
  return processor, model
58
 
59
  def extract_json_from_llm_output(llm_result):
60
+ # Try to extract JSON from a code block first (```json ... ``` or ``` ... ```)
61
+ code_block_match = re.search(r"```(?:json)?\s*({[\s\S]*?})\s*```", llm_result, re.IGNORECASE)
62
+ if code_block_match:
63
+ return code_block_match.group(1)
64
+ # Fallback: extract first {...} block
65
+ match = re.search(r'\{[\s\S]*\}', llm_result)
66
  if match:
67
  return match.group(0)
68
  return None