Update app.py

app.py

@@ -124,10 +124,18 @@ def get_company_sectors(extracted_names, threshold=0.95):
                                             scorer=fuzz.token_set_ratio)
     similarity = name_match[1]/100
     if similarity >= threshold:
-      standard_names_tuples.append(name_match[:2]) 
+      standard_names_tuples.append(name_match[:2])
   
+  for extracted_name in extracted_names:
+    name_match = process.extractOne(extracted_name,
+                                            DF_SP500.Symbol, 
+                                            scorer=fuzz.token_set_ratio)
+    similarity = name_match[1]/100
+    if similarity >= threshold:
+      standard_names_tuples.append(name_match[:2]) 
+
   for std_comp_name, _ in standard_names_tuples:
-    sectors = list(DF_SP500[['Name','Sector']].where(DF_SP500.Name == std_comp_name).dropna().itertuples(index=False, name=None))
+    sectors = list(DF_SP500[['Name','Sector']].where( (DF_SP500.Name == std_comp_name) | (DF_SP500.Symbol == std_comp_name)).dropna().itertuples(index=False, name=None))
     output += sectors
   return output