edited pred df columns

run_domain2go_app.py

@@ -106,8 +106,9 @@ def generate_function_predictions(domains_df, mapping_path):
         return [result_text]
     
     else:
-        merged_df = merged_df[['accession', 'name', 'locations', 'GO', 's']]
-        merged_df.columns = ['domain_accession', 'domain_name', 'domain_locations', 'GO_id', 'probability']
+        merged_df['protein_name'] = domains_df['protein_name'].iloc[0]
+        merged_df = merged_df[['protein_name', 'GO', 'locations', 's', 'accession', 'name',]]
+        merged_df.columns = ['protein_name', 'GO_id', 'domain_locations', 'probability', 'domain_accession', 'domain_name',]
 
         # save protein function predictions
         protein_name = domains_df['protein_name'].iloc[0]