Update app.py

app.py

@@ -31,9 +31,14 @@ def initialize_gemini(api_key: str):
     )
     return model
 
-def google_search_naics(company_name: str) -> List[str]:
-    """Find potential NAICS codes for a company using Google search"""
-    query = f"NAICS code 2022 for {company_name}"
+def google_search_naics(company_name: str, company_description: str = "") -> List[str]:
+    """Find potential NAICS codes for a company using Google search with enhanced context"""
+    # Create a more effective search query with company description if available
+    if company_description:
+        query = f"2022 NAICS code for \"{company_name}\" {company_description} industry classification"
+    else:
+        query = f"2022 NAICS code for \"{company_name}\" company industry classification"
+    
     naics_codes = set()
     
     try:
@@ -166,8 +171,10 @@ def find_naics_code(api_key, company_name, company_description):
         # Initialize Gemini API
         model = initialize_gemini(api_key)
         
-        # Search for NAICS candidates
-        naics_candidates = google_search_naics(company_name)
+        # Search for NAICS candidates with company description for better context
+        naics_candidates = google_search_naics(company_name, company_description)
+        # Store search query for reporting
+        search_query = f"2022 NAICS code for \"{company_name}\" {company_description} industry classification" if company_description else f"2022 NAICS code for \"{company_name}\" company industry classification"
         
         # Get classification
         if not naics_candidates:
@@ -191,7 +198,8 @@ def find_naics_code(api_key, company_name, company_description):
             output += f"**Conclusion:**\n{result['conclusion']}\n\n"
         
         if naics_candidates:
-            output += f"**Candidate NAICS Codes Found from Google:**\n{', '.join(naics_candidates)}"
+            output += f"**Candidate NAICS Codes Found from Google:**\n{', '.join(naics_candidates)}\n\n"
+            output += f"*Search query used: \"{search_query}\"*"
         
         return output