Update app.py

app.py

@@ -76,14 +76,10 @@ def extract_treespecies_features(folder_path):
 
 # print(species_feature_list[:2])
 def perform_inference(cropped_images, species_feature_list, img_df):
-    st.success("Setting up OPENAI Client:")
-    client = setup_client()
-    st.success("Setting up knowledge database & BM25 retriever:")
-    # retriever = setup_retriever()
-    st.success("Setting up BM25 Retriever:")
+
     for img_idx, item in enumerate(cropped_images):
         image = item["image"]
-            feature_cp = extract_features(image)
+            feature_cp = extract_features_cp(image)
         row_results = []
         species_result = []
         emoji = []
@@ -104,15 +100,14 @@ def perform_inference(cropped_images, species_feature_list, img_df):
                 item[f"result_{idx}"] = result
                 item[f"file_name_{idx}"] = species["file_name"]
                 row_results.append(species["file_name"])
-                # Regular expression to match the tree species name
-                species_pattern = r'identified_species\\([^\\]+) -'
+                # # Regular expression to match the tree species name
+                # species_pattern = r'identified_species\\([^\\]+) -'
 
-                # Search for the pattern in the file path
-                match = re.search(species_pattern, species["file_name"])
+                # # Search for the pattern in the file path
+                # match = re.search(species_pattern, species["file_name"])
 
                 # Extract and print the tree species name if found
-                if match:
-                    tree_species = match.group(1)
+              
                     # species_info = retriever.invoke(f"Scientific name:{tree_species}")
                     
                     # ans = generate_image(species_info, client)
@@ -122,9 +117,7 @@ def perform_inference(cropped_images, species_feature_list, img_df):
                     # species_context.append(text_context)
                     # print(ans)
                     # species_result.append(tree_species)
-                    
-                else:
-                    print("Tree species name not found.")
+    
         img_df.at[img_idx, "species_identified"] = ", ".join(species_result) if species_result else "No similar species found"            
         img_df.at[img_idx, "result_file_path"] = ", ".join(row_results) if row_results else ""
         # img_df.at[img_idx, "emoji"] = ", ".join(emoji) if emoji else ""