kmuthudurai commited on
Commit
a6e39e5
·
verified ·
1 Parent(s): bac35d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -128,15 +128,15 @@ This JSON format will be used to store and manage invoices in a structured and u
128
  model="gpt-4o-mini",
129
  messages=[
130
  {"role": "system", "content": system_prompt},
131
- {"role": "user", "content": f"Summarize the following text and provide the JSON format always: {text}"}
132
  ],
133
  temperature=0.5,
134
  max_tokens=16384
135
  )
136
  content = response.choices[0].message.content.strip()
137
  print("Before content:", content)
138
- cleaned_content = re.sub(r'^```json', '', content) # Remove '```json\n' at the beginning
139
- cleaned_content = re.sub(r'```$', '', cleaned_content) # Remove '\n```' at the end
140
 
141
  # Step 2: Parse the cleaned content as JSON
142
  #parsed_content = json.loads(cleaned_content)
 
128
  model="gpt-4o-mini",
129
  messages=[
130
  {"role": "system", "content": system_prompt},
131
+ {"role": "user", "content": f"{text}"}
132
  ],
133
  temperature=0.5,
134
  max_tokens=16384
135
  )
136
  content = response.choices[0].message.content.strip()
137
  print("Before content:", content)
138
+ cleaned_content = re.sub(r'^```json\n', '', content) # Remove '```json\n' at the beginning
139
+ cleaned_content = re.sub(r'\n```$', '', cleaned_content) # Remove '\n```' at the end
140
 
141
  # Step 2: Parse the cleaned content as JSON
142
  #parsed_content = json.loads(cleaned_content)