EdgeRank/main.py at main · Rogan003/EdgeRank · 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
# file with the main function, the app starts here

from parse_files import *
from load_graph import *
from functionalites import *

if __name__ == "__main__":
    print("Dobro dosli! Ucitava se aplikacija...")
    adjust_date_time("dataset/dataset/test_statuses.csv", "dataset/dataset/test_comments.csv",
                     "dataset/dataset/test_shares.csv", "dataset/dataset/test_reactions.csv")
    graph, users = unpickle_graph()
    statuses = load_statuses("dataset/dataset/original_statuses.csv")
    shares = load_shares("dataset/dataset/original_shares.csv")
    comments = load_comments("dataset/dataset/original_comments.csv")
    reactions = load_reactions("dataset/dataset/original_reactions.csv")
    user = None
    test_statuses = load_statuses("dataset/dataset/test_statuses.csv")
    test_shares = load_shares("dataset/dataset/test_shares.csv")
    test_comments = load_comments("dataset/dataset/test_comments.csv")
    test_reactions = load_reactions("dataset/dataset/test_reactions.csv")
    add_affinity(graph,users,test_statuses,test_comments,test_reactions,test_shares)

    for status in test_statuses:
        statuses.append(status)

    trie = load_trie(statuses)

    while True:
        print("Ulogujte se: ")
        username = input("Unesite korisnicko ime: ")

        if username in users:
            user = users[username]
            print("Uspesno ste se ulogovali!\n")
            break
        else:
            print("Netacno korisnicko ime!\n")

    while True:
        izbor = input("Izaberite opciju(1 - prikaz objava, 2 - pretraga, 3 - kraj): ")
        if izbor == "1":
            print("Prikaz objava: ")
            show_feed(user,users,graph,statuses)

        elif izbor == "2":
            text = input("Pretraga(ukoliko stavite ? na kraj dajemo autocomplete, pod \"\" tretiramo kao frazu): ")
            if text != "" and text[-1] == "?":
                text = autocomplete(text,trie)

            print("Rezultati za: " + text)
            search(user,graph,text,statuses,trie,users)

        elif izbor == "3":
            print("Kraj aplikacije!")
            break

        else:
            print("Nepostojeca opcija!")