diff --git a/main.py b/main.py
index e8d80a7..e0d805c 100644
--- a/main.py
+++ b/main.py
@@ -314,7 +314,14 @@ async def extract_info(
             prompt = f"我有以下文本,是一个专利的内容。请按专利名称，发明人，专利号，专利申请日， 专利权人，授权公告号，授权公告日为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}"
         elif file_type == "standard":
             prompt = f"我有以下文本,是一个标准的内容。请按标准号，中文名称，英文名称，发布部门，发布日期，实施日期，提出单位，归口单位，起草单位，起草人为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}"
-        else:
+        elif file_type == "country":
+            prompt = f"我有以下文本,是一个国家标准的内容。请按项目名称，奖励等级，获奖者，证书号，颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}"
+        elif file_type == "province":
+            prompt = f"我有以下文本,是一个省级标准的内容。请按项目名称，奖励等级，获奖者，证书编号，颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}"
+        elif file_type == "social":
+            prompt = f"我有以下文本,是一个社会标准的内容。请按项目名称，获奖单位，奖励等级，获奖人，奖励年度,证书编号,颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}"
+        elif file_type=="building":
+            prompt = f"我有以下文本,是一个建筑标准的内容。请按项目名称，奖励等级，获奖单位，证书编号，颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}"
             raise HTTPException(
                 400, detail="Invalid file type. Please choose 'standard' or 'patent'."
             )