Create get_3d_structure.py

get_3d_structure.py

@@ -0,0 +1,34 @@
+import requests
+
+invoke_url = "https://health.api.nvidia.com/v1/biology/nvidia/esmfold"
+from store_pdb import StorePDB_CloudFlare
+
+
+
+headers = {
+    "Authorization": "Bearer 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",
+    "Accept": "application/json",
+}
+
+def Get3D_Structure_of_Protein(sequence_for_structure,filename):
+
+    payload = {
+      "sequence":sequence_for_structure
+    }
+
+    #re-use connections
+    session = requests.Session()
+
+    response = session.post(invoke_url, headers=headers, json=payload)
+    response.raise_for_status()
+
+    response_body = response.json()
+
+
+    if 'pdbs' in response_body:
+        #return response_body["pdbs"][0])
+        Status,URL=StorePDB_CloudFlare(response_body["pdbs"][0],filename)
+        if Status == True and URL != "":
+            return URL
+    else:
+        return "FATAL"