This repository contains the code for the DBS Bank project, part of the Management of AI in Organizations course. The project focuses on analyzing and visualizing sentiment and excitement from news headlines using different AI models.
The Python script included in this repository performs several key functions:
-
CSV Processing: The script starts by truncating a CSV file containing news headlines to include only those up to a specified date, using the
truncate_csv_by_date
function from thecsv_processing
custom module. -
Data Querying: It then queries sentiment analysis models from different sources:
- Initially, the script used the
query_gpt
function from theopenai_client
custom module to fetch AI-generated sentiment data. - Currently, it uses the
query_gemini
function from thegemini_client
custom module to fetch the data.
- Initially, the script used the
-
Data Manipulation: After fetching the data, the script counts the number of lines and adds a header to the data for better understanding and structure.
-
Data Visualization: Finally, it visualizes the sentiment and excitement scores over time using the
plot_data
function from theplotting
custom module.