Update app.py

app.py

@@ -158,13 +158,23 @@ if uploaded_files:
                         chunkdf = []
                         for i, chunk in enumerate(text_chunk):
                             inp = chunk
-                            try:
-                                df = pd.DataFrame(literal_eval(str(json.dumps(tablex_chain.run(inp)[0])).replace("\'", '\"')), index=[0]).fillna('')
-                            except:
-                                try:
-                                    df = pd.DataFrame(literal_eval(str(json.dumps(tablex_chain.run(inp)[0]) + ']').replace("\'", '\"')), index=[0]).fillna('')
-                                except SyntaxError:
-                                    df = pd.DataFrame(literal_eval('[' + str(json.dumps(tablex_chain.run(inp)[0]) + ']').replace("\'", '\"')), index=[0]).fillna('')
+                            # Assuming tablex_chain.run(inp)[0] returns a dictionary
+                            original_dict = tablex_chain.run(inp)[0]
+                            # Convert the dictionary to a JSON string
+                            json_str = json.dumps(original_dict)
+                            # Replace single quotes with double quotes in the JSON string
+                            json_str_fixed = json_str.replace("'", '"')
+                            # Use literal_eval to safely evaluate the JSON string as a Python dictionary
+                            fixed_dict = literal_eval(json_str_fixed)
+                            # Create a DataFrame from the fixed dictionary
+                            df = pd.DataFrame(fixed_dict, index=[0]).fillna('')
+                            # try:
+                            #     df = pd.DataFrame(literal_eval(str(json.dumps(tablex_chain.run(inp)[0])).replace("\'", '\"')), index=[0]).fillna('')
+                            # except:
+                            #     try:
+                            #         df = pd.DataFrame(literal_eval(str(json.dumps(tablex_chain.run(inp)[0]) + ']').replace("\'", '\"')), index=[0]).fillna('')
+                            #     except SyntaxError:
+                            #         df = pd.DataFrame(literal_eval('[' + str(json.dumps(tablex_chain.run(inp)[0]) + ']').replace("\'", '\"')), index=[0]).fillna('')
                             # df = pd.DataFrame(repair_json(tablex_chain.run(inp)[0]))
                             chunkdf.append(df)