remove business id from ui

app.py

@@ -43,16 +43,19 @@ def format_category( formatted_results):
         f"> 描述：{formatted_results['description'].values[0]}"
     ])
 
-def do( business_id, business_name, address):
+def do( business_name: str, address: str):
     """
     """
 
     crawled_results = []
+    provider = 'openai'
+    model = 'gpt-4-0125-preview'
 
     google_domain = "google.com.tw"
     gl = 'tw'
     lr  = 'lang_zh-TW'
-
+    business_id = 12345678
+    
     query = compose_query(address, business_name)
     try:
         res = get_serp( query, google_domain, gl, lr)
@@ -72,7 +75,7 @@ def do( business_id, business_name, address):
 
     crawled_results = pd.DataFrame(crawled_results)
     # logger.debug(crawled_results)
-    extracted_results = extract_results( crawled_results, classes=classes)
+    extracted_results = extract_results( crawled_results, classes=classes, provider = provider, model = model)
     # logger.error(extracted_results['extracted_results'].columns)
     extracted_results = extracted_results['extracted_results'][ [ 'business_id', 'business_name', 'address', 'category', 'evidence', 'phone_number', 'description', 'store_name'] ]
     logger.debug( extracted_results['category'])
@@ -100,7 +103,8 @@ def do( business_id, business_name, address):
 with gr.Blocks() as demo:
     gr.Markdown("🌟 自動分類餐廳型態 🌟")
     with gr.Row():
-        inputs = [ gr.Textbox( label="統一編號", placeholder="輸入八碼數字（optional）"), gr.Textbox( label="商家名稱", placeholder="輸入商家或公司名稱"), gr.Textbox(label="地址", placeholder="至少輸入縣市，完整地址更好")]
+        # gr.Textbox( label="統一編號", placeholder="輸入八碼數字（optional）"), 
+        inputs = [ gr.Textbox( label="商家名稱", placeholder="輸入商家或公司名稱"), gr.Textbox(label="地址", placeholder="至少輸入縣市，完整地址更好")]
     with gr.Row():
         # outputs = [gr.Dataframe(row_count = (1, "dynamic"), col_count=(6,"dynamic"), label="output data", interactive=1)]
         outputs = [ gr.Markdown( label="參考資料（google search）"), gr.Image( label="文字雲"), gr.Markdown( label="類別", )]

model.py

@@ -10,7 +10,7 @@ from utils import parse_json_garbage
 
 load_dotenv()
 
-def llm( provider, model, system_prompt, user_content, delay:int = 10):
+def llm( provider, model, system_prompt, user_content, delay:int = 0):
     """Invoke LLM service
     Argument
     --------
@@ -28,7 +28,7 @@ def llm( provider, model, system_prompt, user_content, delay:int = 10):
     """
     if delay:
         time.sleep(delay)
-        
+
     if provider=='openai':
         client = OpenAI( organization = os.getenv('ORGANIZATION_ID'))
         chat_completion = client.chat.completions.create(