Function create_bargraph

scraper/utils.py:285–333

Creates a word cloud visualization based on a list of countries. Args: countries (list[str]): A list of countries to be used to generate the word cloud. Returns: A JSON string containing the Plotly Express figure of the word cloud.

(countries: list[str])

    return fig.to_json()

def create_bargraph(countries: list[str]) -> object:
    """
    Creates a word cloud visualization based on a list of countries.

    Args:
        countries (list[str]): A list of countries to be used to generate the word cloud.

    Returns:
        A JSON string containing the Plotly Express figure of the word cloud.
    """

    # Count the occurrences of each country
    country_counts = Counter(countries)
    
    # Get the names and counts of the countries
    country_names = list(country_counts.keys())
    country_values = list(country_counts.values())
    
    # Create a bar graph with the country names on the x-axis and counts on the y-axis
    fig = go.Figure(
        go.Bar(
            x=country_names,
            y=country_values,
            hoverinfo='text',
            hovertext=[f"Country: {country}<br>Citations: {count}" for country, count in zip(country_names, country_values)],
            marker=dict(
                color=country_values,
                colorscale='RdYlGn_r',
                showscale=True,
                colorbar=dict(
                    title='Citations'
                )
            )
        )
    )
    
    fig.update_layout(
        xaxis_title="Country of Origin",
        yaxis_title="Citations",
        title={
            'text': "Frequently Cited Countries",
            'xanchor': 'center',
            'yanchor': 'top',
            'y': 0.9,
            'x': 0.5},
        plot_bgcolor='rgba(0,0,0,0)'
    )

    return fig.to_json()