Update app.py

app.py

@@ -59,7 +59,7 @@ def google_search_naics(company_name: str, company_description: str = "") -> Lis
         print(f"Error performing Google search: {str(e)}")
         return []
 
-def get_naics_classification(model, company_name: str, context: str, candidates: List[str]) -> dict:
+def get_naics_classification(model, company_name: str, context: str, candidates: List[str], search_query: str) -> dict:
     """
     Use Gemini AI to determine the most appropriate NAICS code from candidates
     First provides reasoning, then multiple possibilities with confidence levels
@@ -72,17 +72,20 @@ You are a NAICS code classification expert. Based on the company information pro
 
 Company Name: {company_name}
 Context Information: {context}
-
+Google Search Query Used: {search_query}
 NAICS Code Candidates from Google Search: {candidates}
 
-First, explain your reasoning for which industry this company belongs to.
-Start with "Based on the Google search results for this company and the provided context, I can suggest that..."
+First, start with a section titled "GOOGLE_FINDINGS:" where you describe what the Google search results suggest about this company based on the NAICS codes found.
+
+Then, in a section titled "REASONING:", explain your reasoning for which industry this company belongs to.
 
 Then list 3 potential NAICS classifications with confidence percentages (must add up to 100%).
 Finally, provide your final conclusion.
 
 Your response should be in this format:
-REASONING: Based on the Google search results for this company and the provided context, I can suggest that... [Your detailed reasoning about the company's industry classification]
+GOOGLE_FINDINGS: [Describe what the Google search results suggest about this company based on the NAICS codes found]
+
+REASONING: [Your detailed reasoning about the company's industry classification]
 
 POSSIBILITY_1: [Industry name] - NAICS Code [6-digit code] - [XX]% confidence
 POSSIBILITY_2: [Industry name] - NAICS Code [6-digit code] - [XX]% confidence  
@@ -97,15 +100,19 @@ You are a NAICS code classification expert. Based on the company information pro
 
 Company Name: {company_name}
 Context Information: {context}
+Google Search Query Used: {search_query}
 
-First, explain your reasoning for which industry this company belongs to.
-Start with "Based on the limited information available and without specific Google search results, I can suggest that..."
+First, start with a section titled "GOOGLE_FINDINGS:" where you acknowledge that the Google search did not return any specific NAICS codes for this company.
+
+Then, in a section titled "REASONING:", explain your reasoning for which industry this company belongs to based on the limited information available.
 
 Then list 3 potential NAICS classifications with confidence percentages (must add up to 100%).
 Finally, provide your final conclusion.
 
 Your response should be in this format:
-REASONING: Based on the limited information available and without specific Google search results, I can suggest that... [Your detailed reasoning about the company's industry classification]
+GOOGLE_FINDINGS: No specific NAICS codes were found in the Google search results using the query "{search_query}".
+
+REASONING: [Your detailed reasoning about the company's industry classification based on the limited information available]
 
 POSSIBILITY_1: [Industry name] - NAICS Code [6-digit code] - [XX]% confidence
 POSSIBILITY_2: [Industry name] - NAICS Code [6-digit code] - [XX]% confidence  
@@ -116,6 +123,10 @@ CONCLUSION: I am [XX]% confident this company is [industry description] which is
         response = model.generate_content(prompt)
         response_text = response.text.strip()
         
+        # Extract Google findings
+        google_findings_match = re.search(r'GOOGLE_FINDINGS:(.*?)REASONING:', response_text, re.DOTALL | re.IGNORECASE)
+        google_findings = google_findings_match.group(1).strip() if google_findings_match else "No Google findings provided."
+        
         # Extract reasoning
         reasoning_match = re.search(r'REASONING:(.*?)POSSIBILITY_1:', response_text, re.DOTALL | re.IGNORECASE)
         reasoning = reasoning_match.group(1).strip() if reasoning_match else "No reasoning provided."
@@ -153,6 +164,7 @@ CONCLUSION: I am [XX]% confident this company is [industry description] which is
         
         return {
             "naics_code": naics_code,
+            "google_findings": google_findings,
             "reasoning": reasoning,
             "possibilities": possibilities,
             "conclusion": conclusion
@@ -161,6 +173,7 @@ CONCLUSION: I am [XX]% confident this company is [industry description] which is
         print(f"Error getting NAICS classification: {str(e)}")
         return {
             "naics_code": "000000",
+            "google_findings": "Error occurred during Google search.",
             "reasoning": f"Error analyzing company: {str(e)}",
             "possibilities": [],
             "conclusion": "Error in analysis"
@@ -182,12 +195,21 @@ def find_naics_code(api_key, company_name, company_description):
         
         # Get classification
         if not naics_candidates:
-            result = get_naics_classification(model, company_name, company_description, [])
+            result = get_naics_classification(model, company_name, company_description, [], search_query)
         else:
-            result = get_naics_classification(model, company_name, company_description, naics_candidates)
+            result = get_naics_classification(model, company_name, company_description, naics_candidates, search_query)
         
         # Format the output with NAICS code at the end
         output = f"## Analysis for {company_name}\n\n"
+        
+        # Display search query prominently at the top
+        output += f"**Google Search Query Used:**\n`{search_query}`\n\n"
+        
+        # Add Google findings first
+        if 'google_findings' in result and result['google_findings']:
+            output += f"**Google Search Findings:**\n{result['google_findings']}\n\n"
+        
+        # Then reasoning
         output += f"**Reasoning:**\n{result['reasoning']}\n\n"
         
         # Add possibilities section
@@ -196,11 +218,6 @@ def find_naics_code(api_key, company_name, company_description):
             for i, possibility in enumerate(result['possibilities'], 1):
                 output += f"{i}. {possibility}\n\n"
         
-        # Add Google search information
-        if naics_candidates:
-            output += f"**Candidate NAICS Codes Found from Google:**\n{', '.join(naics_candidates)}\n\n"
-            output += f"*Search query used: \"{search_query}\"*\n\n"
-        
         # Add conclusion
         if 'conclusion' in result and result['conclusion']:
             output += f"**Conclusion:**\n{result['conclusion']}\n\n"