Add cleaning json string function and loading it to json from string

app.py

@@ -86,6 +86,14 @@ HEADER = """
 **Getting Started**: Provide a question and document or context given to your model in addition to the answer given by the model and then click submit. The output panel will indicate whether the reponse is a hallucination (Fail) or if it is faithful to the given document or context (Pass) through the score Pass or Fail and provide reasoning behind the score.
 """
 
+def clean_json_string(json_str):
+    # Replace single quotes with double quotes, but not apostrophes within words
+    json_str = re.sub(r"(?<!\\)'([^']*)'", r'"\1"', json_str)
+    # Add quotes around PASS or FAIL if they're not already quoted
+    json_str = re.sub(r'"SCORE":\s*(PASS|FAIL)', r'"SCORE": "\1"', json_str)
+    
+    return json_st
+
 # @spaces.GPU()
 # def model_call(question, document, answer, tokenizer, model):
 def model_call(question, document, answer):
@@ -97,7 +105,8 @@ def model_call(question, document, answer):
         prompt=NEW_FORMAT
     )
     print("RESPONSE FROM CLIENT:", response)
-    generated_text = json.loads(response.choices[0].text)
+    generated_text = clean_json_string(response.choices[0].text)
+    generated_text = json.loads(generated_text)
     print("GENERATED TEXT", generated_text)
     print("type of GENERATED TEXT", type(generated_text))
     reasoning = generated_text["REASONING"][0]