pricefix

app.py

@@ -278,8 +278,6 @@ def autoPred(df):
     res = res[:-20]
 
     # Get the mean of the agencies
-    
-
     agencyToResult = {agency:result for agency, result in zip(topAgencies, agencyResults)}
     for agency, result in agencyToResult.items():
         print(agency, str(result))
@@ -287,25 +285,19 @@ def autoPred(df):
     # Get the top and bottom 3 agencies with the highest results
     sortedAgencies = sorted(agencyToResult.items(), key=lambda x: x[1])
     meanPrice = sum(agencyResults) / len(agencyResults)
-
-    top3 = sortedAgencies[-3:]
+    top3 = sortedAgencies[-5:]
     top3.reverse()
-    bottom3 = sortedAgencies[:3]
 
-    agencyString = parseAgencyResult(top3, bottom3, meanPrice)
+    agencyString = parseAgencyResult(top3, meanPrice)
 
     return res, agencyString
 
-def parseAgencyResult(top3, bottom3, meanPrice):
+def parseAgencyResult(top3, meanPrice):
     toReturn = '\nTo get the most money for your apartment, you should sell it with the help of one of these agencies:\n'
-    toReturn += 'Top 3:\n'
+    toReturn += 'Top 5:\n'
     for agency, result in top3:
         diff = result - meanPrice
         toReturn += f'{agency}: {parsePrice(result)} ({parsePrice(diff)} above mean)\n'
-    toReturn += 'And these are the worst agencies to use\n'
-    for agency, result in bottom3:
-        diff = result - meanPrice
-        toReturn += f'{agency}: {parsePrice(result)}\n'
 
     return toReturn