Update app.py

app.py

@@ -27,25 +27,53 @@ def string_rep(pq):
 
     return work
 
-def add_elem(date_str, item_name):
+
+def inputter(date_str, item_name, add):
 
     cur_date = datetime.strptime(date_str, date_format)
 
-    pq.put((cur_date, item_name))
+    if add == "Add":
+
+        pq.put((cur_date, item_name))
 
-    return string_rep(pq)
 
-def pop():
+    else:
 
-    pq.get()
+        remove((cur_date,item_name))
 
     return string_rep(pq)
 
 
+def remove(element):
+
+    temp = PriorityQueue()
+
+    while not pq.empty():
+
+        elem = pq.get()
+        if elem == element:
+
+            break
+
+        else:
+
+            temp.put(elem)
+
+    while not pq.empty():
+
+        temp.put(pq.get())
+
+    pq = temp
+
+
+
 datebox = gr.Textbox(label="Enter your date here:", placeholder="MM-DD-YYYY at HH:MM")
 
 hwbox = gr.Textbox(label="Enter your assignment name here:", placeholder="Homework")
 
+addbox = gr.Textbox(label="Add or Drop?:", placeholder="Add or Drop")
+
+
 ouputbox = gr.Textbox(label="List of Assignments:", placeholder="Homework is due on MM-DD-YYYY at HH:MM", lines  = 10)
 
 def tester(str1, str2):
@@ -55,8 +83,8 @@ def tester(str1, str2):
     return str(cur_date.month)
 
 iface = gr.Interface(
-    fn=add_elem,  
-    inputs=[datebox, hwbox],  
+    fn=inputter,  
+    inputs=[datebox, hwbox, addbox],  
     outputs=ouputbox,  
     title="Homework Prioritzer",
     description="Add Work",