Svngoku commited on
Commit
6b0ebea
·
verified ·
1 Parent(s): ce67523

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -140,7 +140,7 @@ class OCRProcessor:
140
  base64_url = f"data:image/jpeg;base64,{encoded_image}"
141
  ocr_response = self._call_ocr_api({"type": "image_url", "image_url": base64_url})
142
  markdown = self._extract_markdown(ocr_response)
143
-
144
  chat_response = self._call_chat_complete(
145
  model="pixtral-12b-latest",
146
  messages=[{
@@ -156,12 +156,18 @@ class OCRProcessor:
156
  response_format={"type": "json_object"},
157
  temperature=0
158
  )
159
-
160
- content = json.loads(chat_response.choices[0].message.content if chat_response.choices else "{}")
 
 
 
 
 
161
  return self._format_structured_response(temp_path, content)
162
  except Exception as e:
163
  return self._handle_error("structured OCR", e)
164
 
 
165
  @staticmethod
166
  def _extract_markdown(response: OCRResponse) -> str:
167
  return response.pages[0].markdown if response.pages else "No text extracted"
 
140
  base64_url = f"data:image/jpeg;base64,{encoded_image}"
141
  ocr_response = self._call_ocr_api({"type": "image_url", "image_url": base64_url})
142
  markdown = self._extract_markdown(ocr_response)
143
+
144
  chat_response = self._call_chat_complete(
145
  model="pixtral-12b-latest",
146
  messages=[{
 
156
  response_format={"type": "json_object"},
157
  temperature=0
158
  )
159
+
160
+ # Ensure the response is a dictionary
161
+ response_content = chat_response.choices[0].message.content
162
+ if isinstance(response_content, list):
163
+ response_content = response_content[0] if response_content else "{}"
164
+
165
+ content = json.loads(response_content)
166
  return self._format_structured_response(temp_path, content)
167
  except Exception as e:
168
  return self._handle_error("structured OCR", e)
169
 
170
+
171
  @staticmethod
172
  def _extract_markdown(response: OCRResponse) -> str:
173
  return response.pages[0].markdown if response.pages else "No text extracted"