Update app.py

app.py

@@ -8,6 +8,7 @@ import openai
 import datetime
 from datetime import datetime, date, time, timedelta
 import os
+import regex
 
 listofcategories=["Earnings Announcements", "Automotive", "Retail"]
 
@@ -49,7 +50,10 @@ def getstuff(openapikey,category_selector):
         )
         print(response.response)
         if 'dataframe' in querylist[i]:
-            df_tmp=pd.read_json(response.response)
+            pattern = regex.compile(r'\{(?:[^{}]|(?R))*\}')
+            jsonextract=pattern.findall(response.response)[0]
+            print("json extract\n",jsonextract)
+            df_tmp=pd.read_json(jsonextract)
             df=pd.DataFrame(df_tmp[df_tmp.columns[0]].tolist())
             answerlist.append(df)
         else: