sentdex_dash_tutorials/dash_sa_app.py at main · FCCMac/sentdex_dash_tutorials · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
import sqlite3
import pandas as pd
import dash
from dash.dependencies import Input, Output
import dash_core_components as dcc
import dash_html_components as html
import plotly
import random
import plotly.graph_objs as go
import plotly.express as px
from collections import deque
from datetime import datetime
import dash_bootstrap_components as dbc
import logging

logging.basicConfig(filename='dash_app.log', level=logging.DEBUG, format='%(levelname)-%(asctime)s-%(message)s')

app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
app.layout = html.Div(children=[
    html.Div(children=[
        html.Div(children=[
            dbc.Alert(
                'No results for the search term',
                id='no-results-alert',
                color='danger',
                is_open=False
            ),

            html.H2('Live Twitter Sentiment'),

            html.Div(children=[
                html.Small('enter a search term below...')
            ])
        ], className='mt-3'),

        dcc.Input(
            id='search-term',
            value='',
            type='text',
            className='form-control'
        ),

        dcc.Graph(
            id='live-graph',
            animate=False
        ),

        dcc.Interval(
            id='graph-update',
            interval=1000
        )
    ], className='container')

])


@app.callback(
    [Output(component_id='live-graph', component_property='figure'),
     Output(component_id='no-results-alert', component_property='is_open')],
    [Input(component_id='graph-update', component_property='n_intervals'),
     Input(component_id='search-term', component_property='value')]
)
def update_graph(n, search_term):
    try:
        conn = sqlite3.connect('twitter.db')
        c = conn.cursor()

        df = pd.read_sql("SELECT * FROM sentiment WHERE tweet LIKE ? ORDER BY unix DESC LIMIT 1000", conn, params=('%' + search_term + '%',))
        df.sort_values('unix', inplace=True)

        if df.size == 0:
            return {}, True
            logging.warning('No values for search term {}'.format(search_term))

        df['sentiment_ra'] = df['sentiment'].rolling(window=int(len(df) / 5)).mean()
        df['timestamp'] = pd.to_datetime(df['unix'], unit='ms').dt.tz_localize('UTC').dt.tz_convert('US/Mountain')
        df.set_index('timestamp', inplace=True)

        if df.size > 1000:
            df = df.resample('1s').mean()


        df.dropna(inplace=True)

        X = df.index[-100:]
        Y = df['sentiment_ra'].values[-100:]

        data = go.Scatter(
            x=list(X),
            y=list(Y),
            name='Scatter',
            mode='lines+markers'
        )

        return {
            'data': [data],
            'layout': go.Layout(
                xaxis=dict(range=[min(X), max(X)]),
                yaxis=dict(range=[min(Y) - 0.1, max(Y) + 0.1])
            )
        }, False

    except Exception as e:
        logging.error(str(e))


if __name__ == "__main__":
    app.run_server(debug=True)