Update app.py

app.py

@@ -94,9 +94,12 @@ if st.button("開始搜索"):
             pchome_json_data = json.loads(pchome_response.content)
             pchome_df = pd.DataFrame(pchome_json_data['prods'])
             
-            # Select only the required columns: name, describe, price, original price
-            pchome_df = pchome_df[['name', 'describe', 'price', 'orig']]
-            pchome_df = pchome_df.rename(columns={'orig': 'original_price'})
+            # Safely select only available columns
+            available_columns = ['name', 'describe', 'price', 'orig']
+            selected_columns = [col for col in available_columns if col in pchome_df.columns]
+            pchome_df = pchome_df[selected_columns]
+            if 'orig' in pchome_df.columns:
+                pchome_df = pchome_df.rename(columns={'orig': 'original_price'})
             pchome_df['platform'] = 'PCHOME'  # Add platform identifier
             pchome_data = pd.concat([pchome_data, pchome_df])
             time.sleep(1)