.

cw2.py

@@ -123,29 +123,26 @@ def also_like(data, doc_uuid, visitor_uuid, ascending, req_5=True):
 
 # Req 6
 def also_like_graph(data, doc_uuid, visitor_uuid, ascending):
-    try:
-        top_docs = also_like(data, doc_uuid, visitor_uuid, ascending, req_5=False)
-        y = top_docs[1].value_counts()
-        y = pd.DataFrame(y)
-        y.loc[len(y.index)] = [visitor_uuid, doc_uuid]
-        print(y)
-        dot = graphviz.Digraph()
-        for index, row in top_docs.iterrows():
-            # if(index <= 7):
-            # print(row[0], row[1])
-            dot.node(str(row[0]), str(row[0])[-4:])
-            dot.node(str(row[1]), str(row[1])[-4:])
-            dot.edge(str(row[0]), str(row[1]))
-
-        # print(dot.source)
-
-        dot.render('output.dot').replace('\\', '/')
-
-        (graph,) = pydot.graph_from_dot_file('output.dot')
-        graph.write_png('output.png')
-        return 'output.png'
-    except:
-        return "output_test.jpg"
+    top_docs = also_like(data, doc_uuid, visitor_uuid, ascending, req_5=False)
+    y = top_docs[1].value_counts()
+    y = pd.DataFrame(y)
+    y.loc[len(y.index)] = [visitor_uuid, doc_uuid]
+    print(y)
+    dot = graphviz.Digraph()
+    for index, row in top_docs.iterrows():
+        # if(index <= 7):
+        # print(row[0], row[1])
+        dot.node(str(row[0]), str(row[0])[-4:])
+        dot.node(str(row[1]), str(row[1])[-4:])
+        dot.edge(str(row[0]), str(row[1]))
+
+    # print(dot.source)
+
+    dot.render('output.dot').replace('\\', '/')
+
+    (graph,) = pydot.graph_from_dot_file('output.dot')
+    graph.write_png('output.png')
+    return 'output.png'
 
 
 # print(also_like(data_tiny, "100713205147-2ee05a98f1794324952eea5ca678c026", "b5d13a36dad1147b", False).groupby(1).count().nlargest(10, [0]))