Omkar008 commited on
Commit
83d8c1d
1 Parent(s): b009d8f

Update services/utils.py

Browse files
Files changed (1) hide show
  1. services/utils.py +4 -4
services/utils.py CHANGED
@@ -69,18 +69,18 @@ def strcuture_document_data(raw_text:str)->dict:
69
  result = model(input.to_string())
70
 
71
 
72
- print(f"GPT Response {result}")
73
  # result = extract_json_from_string(result)
74
  # print(f"Formatted Response : {result}")
75
 
76
  class_object= parser.parse(result)
77
  dict_object=class_object.__dict__
78
- if not dict_object:
79
  print(dict_object)
80
  print("Got null for dict object")
81
 
82
- print("printing structured json")
83
- print(dict_object)
84
  return dict_object
85
  except Exception as e:
86
  print(f"Error occurred: {e}")
 
69
  result = model(input.to_string())
70
 
71
 
72
+ # print(f"GPT Response {result}")
73
  # result = extract_json_from_string(result)
74
  # print(f"Formatted Response : {result}")
75
 
76
  class_object= parser.parse(result)
77
  dict_object=class_object.__dict__
78
+ if all(value is None for value in dict_object.values()):
79
  print(dict_object)
80
  print("Got null for dict object")
81
 
82
+ # print("printing structured json")
83
+ # print(dict_object)
84
  return dict_object
85
  except Exception as e:
86
  print(f"Error occurred: {e}")