Add files via upload

Time-based-Activity-Recommender-System · Apr 11, 2017 · 552dada · 552dada
1 parent c36131f
commit 552dada
Showing 1 changed file with 23 additions and 19 deletions.
diff --git a/entertime_UI_run.py b/entertime_UI_run.py
@@ -6,6 +6,7 @@
 import pandas as pd 
 import numpy as np
 
+
 class Entertime(QtGui.QMainWindow):
     def __init__(self, parent=None):
         QtGui.QWidget.__init__(self,parent)
@@ -38,7 +39,7 @@ def disp_pop(self):
 
         while(ind1 == ind2):
             ind2 = np.random.randint(0, l)
-        while(ind1 == ind3 and ind2 == ind3):
+        while(ind1 == ind3 or ind2 == ind3):
             ind3 = np.random.randint(0, l)
 
         self.ui.tableWidget.setItem(0, 0, QtGui.QTableWidgetItem(pop_movies['movie_title'][ind1]))
@@ -56,33 +57,31 @@ def disp_pop(self):
 
         while(ind1 == ind2):
             ind2 = np.random.randint(0, l)
-        while(ind1 == ind3 and ind2 == ind3):
+        while(ind1 == ind3 or ind2 == ind3):
             ind3 = np.random.randint(0, l)
 
         self.ui.tableWidget.setItem(0, 2, QtGui.QTableWidgetItem(pop_songs['song_title'][ind1]))
         self.ui.tableWidget.setItem(1, 2, QtGui.QTableWidgetItem(pop_songs['song_title'][ind2]))
         self.ui.tableWidget.setItem(2, 2, QtGui.QTableWidgetItem(pop_songs['song_title'][ind3]))
-
-	#display popular books
-	pop_cols = ['book_id', 'book_title']
-	pop_books = pd.read_csv('popular_books.data', sep=';', names=pop_cols, encoding='latin-1')
-	l = len(pop_movies) 
-	#print "Length = ", l
-	ind1 = np.random.randint(0, l)
-	ind2 = np.random.randint(0, l)
-	ind3 = np.random.randint(0, l)
-
 
-	while(ind1 == ind2):
-	    ind2 = np.random.randint(0, l)
-	while(ind1 == ind3 and ind2 == ind3):
-	    ind3 = np.random.randint(0, l)
+        pop_cols = ['book_id', 'book_title']
+        pop_books = pd.read_csv('popular_books.data', sep=';', names=pop_cols, encoding='latin-1')
+        l = len(pop_books) 
+        #print "Length = ", l
+        ind1 = np.random.randint(0, l)
+        ind2 = np.random.randint(0, l)
+        ind3 = np.random.randint(0, l)
 
-	self.ui.tableWidget.setItem(0, 2, QtGui.QTableWidgetItem(pop_books['book_title'][ind1]))
-	self.ui.tableWidget.setItem(1, 2, QtGui.QTableWidgetItem(pop_books['book_title'][ind2]))
-	self.ui.tableWidget.setItem(2, 2, QtGui.QTableWidgetItem(pop_books['book_title'][ind3]))
+        while(ind1 == ind2):
+            ind2 = np.random.randint(0, l)
+        while(ind1 == ind3 or ind2 == ind3):
+            ind3 = np.random.randint(0, l)
 
+        self.ui.tableWidget.setItem(0, 1, QtGui.QTableWidgetItem(pop_books['book_title'][ind1]))
+        self.ui.tableWidget.setItem(1, 1, QtGui.QTableWidgetItem(pop_books['book_title'][ind2]))
+        self.ui.tableWidget.setItem(2, 1, QtGui.QTableWidgetItem(pop_books['book_title'][ind3]))
 
+
         #print pop_movies['movie_title'][ind1]
 	    #print pop_movies['movie_title'][ind2]
 	    #print pop_movies['movie_title'][ind3]
@@ -103,6 +102,11 @@ def disp_pop(self):
     def input_time(self):
         self.hr = self.ui.hrs_spinBox.value()
         self.min = self.ui.mins_spinBox_2.value()
+
+        d = {'hour':[self.hr],'minute':[self.min]}
+        df = pd.DataFrame(d)
+        df.to_csv('time.txt',sep='\t',index=False, header=False)
+
         if(self.hr == 0 and self.min < 15): #song(1 or 2)
             self.hide()
             os.system('python finalPage1_UI_run.py')