from main import load_or_create_the_index ,ORIGIN_DATA ,InvertedIndex
from query import Query

load_or_create_the_index()

def print_origin_new(res):
     for i,j in enumerate(res):
            
            print(ORIGIN_DATA[j]['title'] +" -- "+str(i+1))
            print("doc_id =",j,"                        ")
            print()
            print()
        

import matplotlib.pyplot as plt
import math
With_stops_INDEX = './data/without_stops_index'

InvertedIndex.load(Without_stops_INDEX)

x= InvertedIndex.get_counts()
len_key = len(list(x.keys()))
values = list(x.values())
values.sort(key=lambda t:t,reverse=True)


max_number = values[0]

l = []
l2 = []
l3 = []
for i in values:
    l3.append(math.log(i, 10))

for i in range(len_key):
    l.append(math.log(i+1, 10))
    l2.append(math.log(max_number/(i+1), 10))

plt.plot(l, l2)
plt.plot(l, l3)
plt.show()

import matplotlib.pyplot as plt
import math
NORMAL_INDEX = './data/index'

InvertedIndex.load(NORMAL_INDEX)

x= InvertedIndex.get_counts()
len_key = len(list(x.keys()))
values = list(x.values())
values.sort(key=lambda t:t,reverse=True)


max_number = values[0]

l = []
l2 = []
l3 = []
for i in values:
    l3.append(math.log(i, 10))

for i in range(len_key):
    l.append(math.log(i+1, 10))
    l2.append(math.log(max_number/(i+1), 10))

plt.plot(l, l2)
plt.plot(l, l3)
plt.show()

res =Query('بین‌الملل').best_search()
print_origin_new(res)

res =Query('دانشگاه امیرکبیر').best_search()
print_origin_new(res)

res =Query('دانشگاه امیرکبیر').full_search()
print_origin_new(res)

res =Query('دانشگاه صنعتی امیرکبیر').best_search()
print_origin_new(res)

res =Query('دانشگاه صنعتی امیرکبیر').full_search()
print_origin_new(res)

res =Query('سازمان ملل متحد').best_search()
print_origin_new(res)

res =Query('جمهوری اسلامی ایران').best_search()
print_origin_new(res)

res =Query('ژیمناستیک').best_search()
print_origin_new(res)

res =Query('واکسن آسترازنکا').best_search()
print_origin_new(res)

res =Query('واکسن آسترازنکا').full_search()
print_origin_new(res)