From 5a1345aa4630c91c21470f7b600b061b6429c558 Mon Sep 17 00:00:00 2001
From: Aditya Wibawa Sakti <radenadityawibawa@gmail.com>
Date: Mon, 15 Aug 2022 08:15:39 +0700
Subject: [PATCH] mod gui

---
 .README.md.swp   | Bin 12288 -> 12288 bytes
 cmmde_gui/gui.py |  59 +++++++++++++++++++++++++++++++++--------------
 2 files changed, 42 insertions(+), 17 deletions(-)

diff --git a/.README.md.swp b/.README.md.swp
index 8f07a8df5cc5962c7e42f2a50c521ced505bb63a..a743f6fe2f8dd4e1b74c60ba738f08820af2a3f7 100644
GIT binary patch
delta 194
zcmZojXh_%~Ai(InSy13JKksE$28Q2k3=9kc3=Gbb1r;9K9${x-*aO5xK%57}DL@<p
z#Fjv83d9^h%mBpyfx19CzXS0rAif5~=QbNE&SkYuNJwB{01`j|rjrtjGZ;+u6udIC
s6H&xK925bQ$;Z`pOBzFzDx_qVq~?}pC*~={rxq3E7sYSBtFFue0O0sCa{vGU

delta 200
zcmZojXh_%~Ai(IcSy13JKW{B71H(=>1_lNJ1_qDGf(nmqPq8yF90KBXK)eu$=K^sP
z5LW@Q0TAl|u@VqV0r7XB&d)%642bsu@!rjbigQ^f3#f02Hqle?%FIqwNXaZo%`MGN
z%u|R@Eh@?{inn4&NJwDN)6dK+E=kPE(JRhiNY2R3Pf^gWP(Ts`DoRQO${Fh^ctTVv
Slw_nTB<H83ZoaFo%mDzLt~L4q

diff --git a/cmmde_gui/gui.py b/cmmde_gui/gui.py
index 45f66ec..89f19a8 100755
--- a/cmmde_gui/gui.py
+++ b/cmmde_gui/gui.py
@@ -53,10 +53,10 @@ def cmmde_gui():
 
 # File input (if you don't want to draw the structure)
     FileInput = pn.widgets.FileInput(title='Input structure',accept='.xyz,.vasp,.pdb')
-    def fileinput(event):
-        if FileInput.value is not None:
-            FileInput.save("geom.xyz")
-    FileInput.param.watch(fileinput,'value')
+    # def fileinput(event):
+    #     if FileInput.value is not None:
+    #         FileInput.save("geom.xyz")
+    # FileInput.param.watch(fileinput,'value')
 
 # CMMDE running button
     Run_btn = pn.widgets.Button(name="Run CMMDE!",button_type='primary')
@@ -77,8 +77,8 @@ def cmmde_gui():
         jobs = ",".join(job_list)
         
         if editor.value == "":
-            FileInput.save("geom.xyz")
-            geom = "geom.xyz"
+            FileInput.save(FileInput.filename)
+            geom = FileInput.filename
         else:
             geom = editor.value
             
@@ -171,11 +171,12 @@ def cmmde_gui():
     layer_input = pn.widgets.TextInput(name="Layer",placeholder="Example: 2")
     slabbuilder_btn = pn.widgets.Button(name="Build it!",button_type="primary")
     Material_input = pn.widgets.TextInput(name="Material name")
+    
     Material_upload = pn.widgets.FileInput(title='Input structure')
-    def materialinput(event):
-        if Material_upload.value is not None:
-            Material_upload.save("POSCAR")
-    Material_upload.param.watch(materialinput,'value')
+    # def materialinput(event):
+    #     if Material_upload.value is not None:
+    #         Material_upload.save("POSCAR")
+    # Material_upload.param.watch(materialinput,'value')
 
     # Generate material folder button
     materialdir_btn = pn.widgets.Button(name="Generate work directory",type="primary")
@@ -192,9 +193,11 @@ def cmmde_gui():
             TextWarning.value = "Successfully create the directory!"
 
     materialdir_btn.on_click(materialgen)
+    
     def slab_builder(event):
         Folder = workdir + "/" + id_input.value + "/" + Material_input.value 
         os.chdir(Folder)
+        Material_upload.save("POSCAR")
         # os.system("mv geom.xyz POSCAR")
         hkl = hkl_input.value
         size = size_input.value
@@ -214,11 +217,30 @@ def cmmde_gui():
         # xyzviewer.object = xyzview
     slabbuilder_btn.on_click(slab_builder)
 # Solution Builder
-    # fileinput_solute = pn.widgets.FileInput(accept=".xyz",multiple=True)
-    # fileinput_solvent = pn.widgets.FileInput(accept=".xyz")
-    # def SolutionBuilder(event):
-    #     solute = fileinput_solute.value
-    #     solvent = fileinput_solute.value
+    filename_solute = pn.widgets.TextInput(title="Solute name")
+    filename_solvent = pn.widgets.TextInput(title="Solvent name")
+    solute_upload = pn.widgets.FileInput(accept=".xyz",multiple=True)
+    solvent_upload = pn.widgets.FileInput(accept=".xyz")
+    def SolutionBuilder(event):
+        solute = filename_solute.value.split(",")
+        for i in solute:
+            solute_folder = workdir + "/" + id_input.value + "/" + "{}".format(i)
+            os.makedirs(i)
+            os.chdir(i)
+            solute_upload.save(solute_upload.filename)
+        solvent = filename_solvent.value
+        solvent_folder = workdir + "/" + id_input.value + "/" + solvent
+
+        
+        
+        os.chdir("../{}".format(solvent_folder))
+        solvent_upload.save(solvent_upload.filename)
+        os.chdir("../")
+        cmd = subprocess.run(["cmmde.py","-s","gromacs","-mt","{}".format(solute_folder)])
+
+        solvent = filename_solvent.value
+        solvent_upload.save(solvent_upload.filename)
+
 
 
 
@@ -229,7 +251,10 @@ def cmmde_gui():
     def visualize(event):
         xyzview = py3Dmol.view()
         
-        Folder = workdir + "/" + id_input.value + "/" + Molecule_input.value 
+        if Material_input.value == "":
+            Folder = workdir + "/" + id_input.value + "/" + Molecule_input.value 
+        else:
+            Folder = workdir + "/" + id_input.value + "/" + Material_input.value
         os.chdir(Folder)
         with open('cmmd.xyz','r') as f:
             xyz = f.read()
@@ -266,7 +291,7 @@ def cmmde_gui():
         site="CMMDE-GUI",
         title="CMMDE Editor",
         main=[TextArea, editor, xyzviewer],
-        sidebar=[pn.Card(id_input,title="User Information",collapsed=True),pn.Card(Molecule_input,charge,mult,checkdir_btn,TextWarning,pn.Card(FileInput,title="Upload molecule",collapsed=True),title="Molecule Information",collapsed=True),pn.Card(Material_input, materialdir_btn,TextWarning, pn.Card(FileInput,title="Unit cell",collapsed=True),hkl_input,size_input,layer_input,pn.Row(slabbuilder_btn,visual_btn),pn.Row(download_xyz,download_slab),title="Surface Builder",collapsed=True),pn.Card(software_btn,job_btn,method_btn,pn.Row(Run_btn,Progress_btn),pn.Row(Checkcalc_btn,visual_btn),title="Main Calculation",collapsed=True),pn.Card(post_software_btn,post_btn,post_method_btn,pn.Row(runpost_btn,Progress_btn),pn.Row(Checkcalc_btn,visual_btn),title="Post-Calculation",collapsed=True)],
+        sidebar=[pn.Card(id_input,title="User Information",collapsed=True),pn.Card(Molecule_input,charge,mult,checkdir_btn,TextWarning,pn.Card(FileInput,title="Upload molecule",collapsed=True),title="Molecule Information",collapsed=True),pn.Card(Material_input, materialdir_btn,TextWarning, pn.Card(Material_upload,title="Unit cell",collapsed=True),hkl_input,size_input,layer_input,pn.Row(slabbuilder_btn,visual_btn),pn.Row(download_xyz,download_slab),title="Surface Builder",collapsed=True),pn.Card(software_btn,job_btn,method_btn,pn.Row(Run_btn,Progress_btn),pn.Row(Checkcalc_btn,visual_btn),title="Main Calculation",collapsed=True),pn.Card(post_software_btn,post_btn,post_method_btn,pn.Row(runpost_btn,Progress_btn),pn.Row(Checkcalc_btn,visual_btn),title="Post-Calculation",collapsed=True)],
         header_background=accent, accent_base_color=accent
     )