import pandas
import os
from __future__ import division

root='/projects/3572606d-aafc-4285-a345-c223c56d1457/ML/Main/WEKA-test-result/'
file_list=[]
for file in os.listdir(root):
    if file.endswith(".xlsx"):
        file_list.append(os.path.join(root, file))
#print(file_list)

data = {}
for path in file_list:
    base_name_without_ext = os.path.basename(path)
    base_name_without_ext = os.path.splitext(base_name_without_ext)[0]
    try:
        data[base_name_without_ext] = pandas.read_excel(open(path,'rb'), sheetname='test1').to_dict(orient='index')
    except Exception as e:
        print(e, base_name_without_ext)
        pass
data['RIVP-features-test1'][0]

os.path.expanduser('~')

import numpy as np
import plotly.offline as py
import plotly.graph_objs as go

py.init_notebook_mode()
def _draw2(data0, data1, data2, name1, name2):
    trace1 = go.Scatter(
    x = data0,
    y = data1, #value
    name = name1 ,
    line = dict(
        color = ('rgb(205, 12, 24)'),
        width = 4))
    trace2 = go.Scatter(
    x = data0,
    y = data2, #value
    name = name2,
    line = dict(
        color = ('rgb(33, 136, 220)'),
        width = 4))
    data = [trace1, trace2]
    # Edit the layout
    layout = dict(title = name1+' usporedba s ' + name2,
              xaxis = dict(title = 'Dani'),
              yaxis = dict(title = 'Vrijednost'),
              )
    fig = dict(data=data, layout=layout)
    py.iplot(fig, filename=name1)

def analiza(rezultat):
    actual = [] #stvarna
    error = [] #greška
    inst = [] #broj
    predicted = [] #predviđena
    
    for i in range(len(data[rezultat])):
        actual.append(data[rezultat][i][u'actual'])
        error.append(data[rezultat][i][u'error'])
        inst.append(data[rezultat][i][u'inst#'])
        predicted.append(data[rezultat][i][u'predicted'])
   
    _draw2(inst,actual,predicted,'Stvarna od '+rezultat,'Predviđena od '+rezultat)
    
    return 

analiza('RIVP-features-test1')
analiza('RIVP-test1')