Update app.py

app.py

@@ -117,10 +117,11 @@ def unavailable(input_file, input_checks):
     return gr.update(value=output, label="Oops!", visible=True)
 
 def file(input_file, input_checks):
-    output = "output.txt"
-    f = open(output, 'w')
-    f.write("The predictions come here.")
-    f.close()
+    #output = "output.txt"
+    #f = open(output, 'w')
+    #f.write("The predictions come here.")
+    #f.close()
+    output = "showcase/data.txt"
     if "emotion frequencies" in input_checks:
         return gr.update(value=output, visible=True), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)  # next_button_freq becomes available
     elif "emotion distribution over time" in input_checks:
@@ -133,9 +134,25 @@ def file(input_file, input_checks):
         return gr.update(value=output, visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)  # no next_button becomes available
 
 def freq(output_file, input_checks):
+    #simple = pd.DataFrame({
+    #'Emotion category': ['neutral', 'anger', 'fear', 'joy', 'love', 'sadness'],
+    #'Frequency': [10, 8, 2, 15, 3, 4]})
+    
+    f = open("showcase/data.txt", 'r')
+    data = f.read().split("\n")
+    f.close()
+    data = [line.split("\t") for line in data[1:-1]]
+    
+    freq_dict = {}
+    for line in data:
+    	if line[1] not in freq_dict.keys():
+    		freq_dict[line[1]] = 1
+    	else:
+    		freq_dict[line[1]] += 1
+    
     simple = pd.DataFrame({
-    'Emotion category': ['neutral', 'anger', 'fear', 'joy', 'love', 'sadness'],
-    'Frequency': [10, 8, 2, 15, 3, 4]})
+    	'Emotion category': ['neutral', 'anger', 'fear', 'joy', 'love', 'sadness'],
+    	'Frequency': [freq_dict['neutral'], freq_dict['anger'], freq_dict['fear'], freq_dict['joy'], freq_dict['love'], freq_dict['sadness']]})
 
     domain = ['neutral', 'anger', 'fear', 'joy', 'love', 'sadness']
     range_ = ['#999999', '#b22222', '#663399', '#ffcc00', '#db7093', '#6495ed']