From 73a6cbcbc7e59f4a1352432c86ea44f2345827e1 Mon Sep 17 00:00:00 2001
From: DeOwl <deowl28@gmail.com>
Date: Sun, 1 Feb 2026 13:47:23 +0300
Subject: [PATCH] added add_h parameter in conversion

---
 src/app.py                     | 8 ++++----
 src/request_response_models.py | 1 +
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/app.py b/src/app.py
index cc8e6db..835f659 100644
--- a/src/app.py
+++ b/src/app.py
@@ -6,7 +6,7 @@ from fastapi.exceptions import HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 
 # from fastapi.requests import Request
-from fastapi.responses import Response
+from fastapi.responses import JSONResponse
 from fastapi_cache import FastAPICache
 from fastapi_cache.backends.redis import RedisBackend
 from fastapi_cache.decorator import cache
@@ -54,8 +54,8 @@ def convert_molecule(req: ConvertRequest):
         # Rad the string and format from request
         mol = pybel.readstring(req.format, req.text)
 
-        mol.addh()
-
+        if req.add_hydrogen:
+            mol.addh()
         if req.convert_3d:
             mol.make3D()
         if req.optimize_geometry:
@@ -79,7 +79,7 @@ def convert_molecule(req: ConvertRequest):
         )
 
         mol2string = mol.write("xyz")
-        return Response({"molfile": mol2string}, 200)
+        return JSONResponse({"molfile": mol2string}, 200)
     except Exception as e:
         raise HTTPException(status_code=400, detail={"error": str(e)})
 
diff --git a/src/request_response_models.py b/src/request_response_models.py
index d7618b7..357f621 100644
--- a/src/request_response_models.py
+++ b/src/request_response_models.py
@@ -4,6 +4,7 @@ from pydantic import BaseModel
 class ConvertRequest(BaseModel):
     text: str
     format: str  # e.g. "mol", "smi", "sdf", "inchi", etc.
+    add_hydrogen: bool = False
     convert_3d: bool = False
     optimize_geometry: bool = False