Update app.py

app.py

@@ -130,80 +130,7 @@ CONCLUSION: I am [XX]% confident this company is [industry description] which is
             possibilities.append(poss3_match.group(1).strip())
         
         # Extract conclusion
-        conclusion_match = re.search(r'CONCLUSION:(.*?)
-    except Exception as e:
-        print(f"Error getting NAICS classification: {str(e)}")
-        return {
-            "naics_code": "000000",
-            "reasoning": f"Error analyzing company: {str(e)}"
-        }
-
-def find_naics_code(api_key, company_name, company_description):
-    """Main function to find NAICS code that will be called by Gradio"""
-    if not api_key or not company_name:
-        return "Please provide both API key and company name."
-    
-    try:
-        # Initialize Gemini API
-        model = initialize_gemini(api_key)
-        
-        # Search for NAICS candidates
-        naics_candidates = google_search_naics(company_name)
-        
-        # Get classification
-        if not naics_candidates:
-            result = get_naics_classification(model, 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"
-        output += f"**Reasoning:**\n{result['reasoning']}\n\n"
-        
-        # Add possibilities section
-        if 'possibilities' in result and result['possibilities']:
-            output += f"**Possible Classifications:**\n\n"
-            for i, possibility in enumerate(result['possibilities'], 1):
-                output += f"{i}. {possibility}\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)}"
-        
-        return output
-    
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-# Create Gradio Interface
-with gr.Blocks(title="NAICS Code Finder") as app:
-    gr.Markdown("# NAICS Code Finder")
-    gr.Markdown("This app helps you find the appropriate NAICS code for a company based on its name and description.")
-    
-    with gr.Row():
-        with gr.Column():
-            api_key = gr.Textbox(label="Google Gemini API Key", placeholder="Enter your Gemini API key here", type="password")
-            company_name = gr.Textbox(label="Company Name", placeholder="Enter the company name")
-            company_description = gr.Textbox(label="Company Description", placeholder="Enter a brief description of the company", lines=5)
-            
-            submit_btn = gr.Button("Find NAICS Code")
-        
-        with gr.Column():
-            output = gr.Markdown(label="Result")
-    
-    submit_btn.click(
-        fn=find_naics_code,
-        inputs=[api_key, company_name, company_description],
-        outputs=output
-    )
-
-if __name__ == "__main__":
-    app.launch()
-, response_text, re.DOTALL | re.IGNORECASE)
+        conclusion_match = re.search(r'CONCLUSION:(.*?)$', response_text, re.DOTALL | re.IGNORECASE)
         conclusion = conclusion_match.group(1).strip() if conclusion_match else "No conclusion provided."
         
         # Extract final NAICS code from conclusion
@@ -225,7 +152,9 @@ if __name__ == "__main__":
         print(f"Error getting NAICS classification: {str(e)}")
         return {
             "naics_code": "000000",
-            "reasoning": f"Error analyzing company: {str(e)}"
+            "reasoning": f"Error analyzing company: {str(e)}",
+            "possibilities": [],
+            "conclusion": "Error in analysis"
         }
 
 def find_naics_code(api_key, company_name, company_description):
@@ -251,8 +180,18 @@ def find_naics_code(api_key, company_name, company_description):
         output += f"**NAICS Code:** {result['naics_code']}\n\n"
         output += f"**Reasoning:**\n{result['reasoning']}\n\n"
         
+        # Add possibilities section
+        if 'possibilities' in result and result['possibilities']:
+            output += f"**Possible Classifications:**\n\n"
+            for i, possibility in enumerate(result['possibilities'], 1):
+                output += f"{i}. {possibility}\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:**\n{', '.join(naics_candidates)}"
+            output += f"**Candidate NAICS Codes Found from Google:**\n{', '.join(naics_candidates)}"
         
         return output