Update gradio_app.py

gradio_app.py

@@ -73,9 +73,16 @@ def news_analysis(text):
             headers=HEADERS
         )
         response.raise_for_status()
+
         # Prepare results for JSON output
         json_results = response.json()
-        # Normalize the JSON data
+
+        # Preprocess to convert the claim_objects from a dict to a list
+        for result in json_results:
+            if isinstance(result.get('claim_objects'), dict):
+                result['claim_objects'] = list(result['claim_objects'].values())
+        
+        # Flatten the JSON
         dataframe_results = pd.json_normalize(
             json_results,
             record_path=['claim_objects'],                # Nested array to flatten
@@ -100,14 +107,30 @@ def claim_verification(text):
             headers=HEADERS
         )
         response.raise_for_status()
+
         # Prepare results for JSON output
         json_results = response.json()
-        # Normalize the JSON data
+
+        # Preprocess to convert nested fields from dict to list
+        for result in json_results:
+            # Convert 'support' to a list if it's a dictionary
+            if isinstance(result.get('support'), dict):
+                result['support'] = list(result['support'].values())
+            
+            # Convert 'refute' to a list if it's a dictionary
+            if isinstance(result.get('refute'), dict):
+                result['refute'] = list(result['refute'].values())
+            
+            # Convert 'no_info' to a list if it's a dictionary
+            if isinstance(result.get('no_info'), dict):
+                result['no_info'] = list(result['no_info'].values())
+        
+        # Flatten the JSON
         dataframe_results = pd.json_normalize(
             json_results,
             record_path=['support', 'refute', 'no_info'],   # All possible nested lists
-            meta=['doc_id', 'details'],                     # Top-level fields
-            sep='_',                                        # Custom separator for nested keys
+            meta=['doc_id', 'details'],                     # Top-level fields to retain
+            sep='_',                                        # Separator for nested fields
             errors='ignore'                                 # To handle cases with missing keys
         )
         return json_results, dataframe_results