Update models.py

models.py

@@ -1,5 +1,6 @@
 import parser
 import requests
+import re
 
 def zephyr_score(sentence):
     prompt = f"""<|user|> 
@@ -89,7 +90,15 @@ def query_model_revise(sentence, api_key, model_id, prompt_fun):
     data = query({"inputs": f"{prompt}", "parameters": parameters, "options": options})
     revision = data[0]['generated_text']
     return revision
-
+    
+def extract_quotes(text):
+    pattern = r'"([^"]*)"'
+    matches = re.findall(pattern, text)
+    if matches:
+        return ' '.join(matches)
+    else:
+        return text
+        
 def revise_review(review, api_key, model_id, highlight_color):
     result = {
         "success": False,
@@ -124,6 +133,7 @@ def revise_review(review, api_key, model_id, highlight_color):
                     review_score = 1
                     revision_count +=1
                     revision = query_model_revise(sentence, api_key, model_id, revision_prompt)
+                    revision = extract_text_in_quotes(revision)
                     revision = revision.strip().strip('"')
                     review_revision += f" <div style='background-color: {highlight_color}; display: inline;'>{revision}</div>"
             else: