Update app.py

app.py

@@ -314,6 +314,35 @@ def get_chart(inp):
         print(f'Error:: {e}')
         resp = e
     return resp
+
+def format_json(inp):
+
+    print("FORMATTING:::")
+    print(type(inp))
+    print("###########")
+    print(inp)
+    print("###########")
+    print("###########")
+    new_str=""
+    matches=["```","#","//"]
+    for i,line in enumerate(inp):
+        line = line.strip()
+        print(line)
+        #if not any(x in line for x in matches):
+        new_str+=line.strip("\n").strip("```").strip("#").strip("//")
+    print("###########")
+    print("###########")
+    #inp = inp.strip("<\s>")
+    new_str=new_str.strip("</s>")
+    out_json=eval(new_str)
+    print(out_json)
+    print("###########")
+    print("###########")
+    
+    return out_json
+
+
+    
 def summarize(inp,history,report_check,chart_check,data=None,files=None,directory=None,url=None,pdf_url=None,pdf_batch=None):
     json_box=[]
     error_box=""
@@ -405,9 +434,13 @@ def summarize(inp,history,report_check,chart_check,data=None,files=None,director
             rawp = compress_data_og(c,inp,out)
         else:
             rawp = out
+        try:
+            json_out=format_json(json_out)
+        except Exception as e:
+            print (e)
         if chart_check:
             print (f"making chart from ::: {rawp}")
-            error_box = get_chart(str(rawp))
+            error_box = get_chart(str(json_out))
             print(error_box)
     else:
         rawp = "Provide a valid data source"