Update app.py

app.py

@@ -13,7 +13,7 @@ import tempfile
 from inference import split_image_from_dataframe
 from datetime import datetime
 from predict_vit import extract_features, predict_similarity, compare_features, extract_features_cp
-from rag import generate_image, setup_client, setup_retriever
+
 from predict_copy import extract_features_with_augmentation, extract_features_with_augmentation_cp
 
 
@@ -79,7 +79,7 @@ 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()
+    # retriever = setup_retriever()
     st.success("Setting up BM25 Retriever:")
     for img_idx, item in enumerate(cropped_images):
         image = item["image"]
@@ -113,22 +113,22 @@ def perform_inference(cropped_images, species_feature_list, img_df):
                 # 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}")
+                    # species_info = retriever.invoke(f"Scientific name:{tree_species}")
                     
-                    ans = generate_image(species_info, client)
-                    emoji.append(ans)
-                    text_context = [doc.page_content for doc in species_info]
-                    text_context = ", ".join(text_context)
-                    species_context.append(text_context)
+                    # ans = generate_image(species_info, client)
+                    # emoji.append(ans)
+                    # text_context = [doc.page_content for doc in species_info]
+                    # text_context = ", ".join(text_context)
+                    # species_context.append(text_context)
                     # print(ans)
-                    species_result.append(tree_species)
+                    # 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 ""
-        img_df.at[img_idx, "retreived context"] = ", ".join(species_context) if species_context else ""
+        # img_df.at[img_idx, "emoji"] = ", ".join(emoji) if emoji else ""
+        # img_df.at[img_idx, "retreived context"] = ", ".join(species_context) if species_context else ""
         
         
     return cropped_images