correct structured

app.py

@@ -92,6 +92,8 @@ import html2text
 from interpreter.code_interpreter import CodeInterpreter
 from interpreter.code_block import CodeBlock
 
+import regex
+
 class CodeBlock:
     def __init__(self, code):
         self.code = code
@@ -276,6 +278,18 @@ class GPTRemote(LLM):
         if 'Action:' in output and 'Observation:' in output:
             output = output.split('Observation:')[0]
 
+        global Choice
+        if Choice == "Structured Zero Short Agent":
+            try:
+                # temp = output.split('{')[1].split('}')[0:-2]
+                pattern = r'\{((?:[^{}]|(?R))*)\}'
+                temp = regex.search(pattern, output)
+                rrr = temp.group()
+                output = output.replace(rrr, '```'+ '\n' + rrr + '\n'+'```')
+
+                print(output)
+            except Exception as e:
+                print("model internal error:", e)
         print("-" * 20)
         print("Treated output: \n", output)
         return output