Update gradio_app.py

gradio_app.py

@@ -60,7 +60,6 @@ def toggle_feedback(request_data, response_data, like_clicked, dislike_clicked):
     else:
         return "Feedback sent successfully!"
 
-
 def preprocess_and_flatten(json_results, mode, meta_fields=None):
     # Ensure 'meta_fields' is a list or set default fields
     if meta_fields is None:
@@ -106,9 +105,14 @@ def preprocess_and_flatten(json_results, mode, meta_fields=None):
                     'sentence': item.get('sentence', ''),
                     'doi': item.get('doi', '')
                 })
-    
+
     # Convert to DataFrame
-    return pd.DataFrame(flattened_data)
+    dataframe_results = pd.DataFrame(flattened_data)
+
+    # Capitalize column names
+    dataframe_results.columns = [col.capitalize() for col in dataframe_results.columns]
+
+    return dataframe_results
     
 # Define the functions to handle the inputs and outputs
 def news_analysis(text):