Update app.py
Browse files
app.py
CHANGED
@@ -3,6 +3,64 @@ from aggregator import get_articles_sentiment
|
|
3 |
|
4 |
st.title("Real time financial news fast sentiment")
|
5 |
|
6 |
-
|
7 |
|
8 |
-
st.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
|
4 |
st.title("Real time financial news fast sentiment")
|
5 |
|
6 |
+
model_to_use = ""
|
7 |
|
8 |
+
model_selected = st.radio("Choose your model", ["fin_distilBert", "fin_tinyBert", "fin_miniLM"])
|
9 |
+
|
10 |
+
ticker = st.text_input("Insert the company ticker")
|
11 |
+
|
12 |
+
col1, col2 = st.columns(2)
|
13 |
+
if model_selected and ticker is not None:
|
14 |
+
|
15 |
+
if model_selected == "fin_distilBert":
|
16 |
+
model_to_use = "Andreagus/fin_distilbert_16"
|
17 |
+
if model_selected == "fin_tinyBert":
|
18 |
+
model_to_use = "Andreagus/fin_tinyBert_32"
|
19 |
+
if model_selected == "fin_miniLM12":
|
20 |
+
model_to_use = "Andreagus/fin_miniLM_16"
|
21 |
+
|
22 |
+
results = get_articles_sentiment(ticker, model_to_use)
|
23 |
+
|
24 |
+
with col1:
|
25 |
+
st.text("Bezinga news provider")
|
26 |
+
if results['bezinga']['bezinga_articles'] == 0:
|
27 |
+
st.text('Bezinga returned 0 articles')
|
28 |
+
else:
|
29 |
+
st.json(results['bezinga'], expanded=False)
|
30 |
+
|
31 |
+
st.text("finhub news provider")
|
32 |
+
if results['finhub']['finhub_articles'] == 0:
|
33 |
+
st.text('finhub returned 0 articles')
|
34 |
+
else:
|
35 |
+
st.json(results['finhub'], expanded=False)
|
36 |
+
|
37 |
+
st.text("marketaux news provider")
|
38 |
+
if results['marketaux']['marketaux_articles'] == 0:
|
39 |
+
st.text('marketaux returned 0 articles')
|
40 |
+
else:
|
41 |
+
st.json(results['marketaux'], expanded=False)
|
42 |
+
|
43 |
+
st.text("Newsapi news provider")
|
44 |
+
if results['newsapi']['newsapi_articles'] == 0:
|
45 |
+
st.text('newsapi returned 0 articles')
|
46 |
+
else:
|
47 |
+
st.json(results['newsapi'], expanded=False)
|
48 |
+
|
49 |
+
st.text('Newsdata news provider')
|
50 |
+
if results['newsdata']['newsdata_articles'] == 0:
|
51 |
+
st.text('newsdata returned 0 articles')
|
52 |
+
else:
|
53 |
+
st.json(results['newsdata'], expanded=False)
|
54 |
+
|
55 |
+
st.text("Vantage news provider")
|
56 |
+
if results['vantage']['vantage_articles'] == 0:
|
57 |
+
st.text('vantage returned 0 articles')
|
58 |
+
else:
|
59 |
+
st.json(results['vantage'], expanded=False)
|
60 |
+
|
61 |
+
with col2:
|
62 |
+
st.text("Summary results")
|
63 |
+
st.metric("Total articles", results['total_articles'])
|
64 |
+
st.metric("Total positive articles", results['total_positives'])
|
65 |
+
st.metric("Total negative articles", results['total_negatives'])
|
66 |
+
|