Update app.py

app.py

@@ -76,11 +76,13 @@ Context Information: {context}
 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..."
+
 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: [Your detailed reasoning about the company's industry classification]
+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]
 
 POSSIBILITY_1: [Industry name] - NAICS Code [6-digit code] - [XX]% confidence
 POSSIBILITY_2: [Industry name] - NAICS Code [6-digit code] - [XX]% confidence  
@@ -97,11 +99,13 @@ Company Name: {company_name}
 Context Information: {context}
 
 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..."
+
 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: [Your detailed reasoning about the company's industry classification]
+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]
 
 POSSIBILITY_1: [Industry name] - NAICS Code [6-digit code] - [XX]% confidence
 POSSIBILITY_2: [Industry name] - NAICS Code [6-digit code] - [XX]% confidence  
@@ -182,9 +186,8 @@ def find_naics_code(api_key, company_name, company_description):
         else:
             result = get_naics_classification(model, company_name, company_description, naics_candidates)
         
-        # Format the output
-        output = f"## NAICS Code for {company_name}\n\n"
-        output += f"**NAICS Code:** {result['naics_code']}\n\n"
+        # Format the output with NAICS code at the end
+        output = f"## Analysis for {company_name}\n\n"
         output += f"**Reasoning:**\n{result['reasoning']}\n\n"
         
         # Add possibilities section
@@ -193,13 +196,17 @@ 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"
         
-        if naics_candidates:
-            output += f"**Candidate NAICS Codes Found from Google:**\n{', '.join(naics_candidates)}\n\n"
-            output += f"*Search query used: \"{search_query}\"*"
+        # Add final NAICS code at the very end
+        output += f"**FINAL NAICS CODE: {result['naics_code']}**"
         
         return output