zrd 3 месяцев назад
Родитель
Сommit
ac0bfa3860
2 измененных файлов с 3 добавлено и 3 удалено
  1. 1 1
      main.py
  2. 2 2
      utils/text_to_html.py

+ 1 - 1
main.py

@@ -1,5 +1,5 @@
 import uvicorn
-from fastapi import FastAPI,Form
+from fastapi import FastAPI, Form, HTTPException
 from pydantic import BaseModel
 
 from tool.pdf2md import pdf2ai

+ 2 - 2
utils/text_to_html.py

@@ -6,7 +6,7 @@ def save_html_to_file(html_content, file_name):
     # 保存 HTML 文件的路径,假设保存到 /home 目录下
 
     current_time = datetime.now().strftime("%Y%m%d%H%M%S")
-    file_path = f"/home/aihtml/{file_name}_{current_time}.html"
+    file_path = f"/home/aihtml/ai_{current_time}.html"
     directory = os.path.dirname(file_path)
     if not os.path.exists(directory):
         os.makedirs(directory)
@@ -18,7 +18,7 @@ def save_html_to_file(html_content, file_name):
     except Exception as e:
         print(f"保存文件时出现错误: {e}")
 
-    return f"{file_name}_{current_time}.html"
+    return f"ai_{current_time}.html"
 
 
 if __name__ == "__main__":