Update get_molecular_agent.py

get_molecular_agent.py

@@ -32,8 +32,7 @@ model1 = RxnScribe(ckpt_path, device=torch.device('cpu'))
 device = torch.device('cpu')
 model = ChemIEToolkit(device=torch.device('cpu'))
 
-API_KEY = os.getenv("API_KEY")
-AZURE_ENDPOINT = os.getenv("AZURE_ENDPOINT")
+
 
 
 def get_multi_molecular(image_path: str) -> list:
@@ -352,6 +351,8 @@ def process_reaction_image_with_multiple_products_and_text_correctR(image_path:
     Returns:
         dict: 整理后的反应数据，包括反应物、产物和反应模板。
     """
+    API_KEY = os.getenv("API_KEY")
+    AZURE_ENDPOINT = os.getenv("AZURE_ENDPOINT")
     client = AzureOpenAI(
         api_key=API_KEY,
         api_version='2024-06-01',
@@ -601,6 +602,9 @@ def process_reaction_image_with_multiple_products_and_text_correctmultiR(image_p
     Returns:
         dict: 整理后的反应数据，包括反应物、产物和反应模板。
     """
+
+    API_KEY = os.getenv("API_KEY")
+    AZURE_ENDPOINT = os.getenv("AZURE_ENDPOINT")
     client = AzureOpenAI(
         api_key=API_KEY,
         api_version='2024-06-01',