Update app.py

app.py

@@ -1,3 +1,37 @@
+import os
+import re
+import json
+import requests
+from typing import List, Dict, Optional, Tuple
+import gradio as gr
+from googlesearch import search
+import google.generativeai as genai
+from google.generativeai.types import HarmCategory, HarmBlockThreshold
+from bs4 import BeautifulSoup
+
+def initialize_gemini(api_key: str):
+    """Initialize the Google Gemini API with appropriate configurations"""
+    genai.configure(api_key=api_key)
+    generation_config = {
+        "temperature": 0.2,
+        "top_p": 0.8,
+        "top_k": 40,
+        "max_output_tokens": 1024,
+    }
+    safety_settings = {
+        HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
+        HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
+        HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
+        HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
+    }
+    
+    model = genai.GenerativeModel(
+        model_name="gemini-1.5-flash",
+        generation_config=generation_config,
+        safety_settings=safety_settings
+    )
+    return model
+
 def google_search_company_info(company_name: str) -> str:
     """
     Search for basic company information to help with NAICS classification
@@ -26,7 +60,6 @@ def google_search_company_info(company_name: str) -> str:
                         response = requests.get(result_url, timeout=5)
                         if response.status_code == 200:
                             # Extract text from paragraphs
-                            from bs4 import BeautifulSoup
                             soup = BeautifulSoup(response.text, 'html.parser')
                             paragraphs = soup.find_all('p')
                             
@@ -52,38 +85,7 @@ def google_search_company_info(company_name: str) -> str:
         return company_info.strip()
     except Exception as e:
         print(f"❌ Error searching for company info: {str(e)}")
-        return ""import os
-import re
-import json
-import requests
-from typing import List, Dict, Optional, Tuple
-import gradio as gr
-from googlesearch import search
-import google.generativeai as genai
-from google.generativeai.types import HarmCategory, HarmBlockThreshold
-
-def initialize_gemini(api_key: str):
-    """Initialize the Google Gemini API with appropriate configurations"""
-    genai.configure(api_key=api_key)
-    generation_config = {
-        "temperature": 0.2,
-        "top_p": 0.8,
-        "top_k": 40,
-        "max_output_tokens": 1024,
-    }
-    safety_settings = {
-        HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
-        HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
-        HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
-        HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
-    }
-    
-    model = genai.GenerativeModel(
-        model_name="gemini-1.5-flash",
-        generation_config=generation_config,
-        safety_settings=safety_settings
-    )
-    return model
+        return ""
 
 def google_search_naics(company_name: str) -> List[str]:
     """
@@ -271,30 +273,6 @@ def find_naics_code(company_name: str, context: str = "", api_key: Optional[str]
     
     return result
 
-# Gradio interface function
-def classify_company(company_name: str, company_description: str, api_key: str = None) -> Tuple[str, str, str]:
-    """Process inputs from Gradio and return formatted results"""
-    if not api_key:
-        api_key = os.environ.get('GEMINI_API_KEY')
-    
-    if not company_name:
-        return "Error: Company name is required", "", ""
-    
-    result = find_naics_code(company_name, company_description, api_key)
-    
-    # Format the NAICS code output
-    naics_code = f"**NAICS Code: {result['naics_code']}**"
-    
-    # Format the research output
-    research = ""
-    if "research" in result and result["research"]:
-        research = f"## Research on NAICS Codes\n\n{result['research']}"
-    
-    # Format the reasoning output
-    reasoning = f"## Analysis\n\n{result['reasoning']}"
-    
-    return naics_code, research, reasoning
-
 # Create the Gradio interface
 def create_gradio_interface():
     # Check if API key is set in environment