initial

.gitignore

@@ -0,0 +1 @@
+MISC/

STPI_2WAY_RandomForest.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:84ae3aec58b4bf3cfe625ac2138d6ac245f741787070ea7ae2538f404029fc5a
+size 86585

STPI_3WAY_RandomForest.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2404d314d346836902a186594bbeb9371f822b07aabe863d07c3777ddca05875
+size 95873

app.py

@@ -0,0 +1,62 @@
+# %%
+import gradio as gr
+import joblib
+loaded_rf_2way = joblib.load("STPI_2WAY_RandomForest.joblib")
+loaded_rf_3way = joblib.load("STPI_3WAY_RandomForest.joblib")
+
+
+def STPI(t_0_5_MaxValue,t_1_0_MaxValue,t_2_0_MaxValue,
+# Acc_0_5__1_0_MaxValue,
+Abs_Diff_t_0_5_MaxValue,Abs_Diff_t_1_0_MaxValue,Abs_Diff_t_2_0_MaxValue,Optional_Custom_Message='No_Message'):
+    print('------------------')
+    print(Optional_Custom_Message)
+    
+    X = [t_0_5_MaxValue,t_1_0_MaxValue,t_2_0_MaxValue,
+    # Acc_0_5__1_0_MaxValue,
+    Abs_Diff_t_0_5_MaxValue,Abs_Diff_t_1_0_MaxValue,Abs_Diff_t_2_0_MaxValue]
+    print(X)
+    outcome_decoded = ['Normal','Keratoconic','Suspect']
+    file_object = open('stpi_data.txt', 'a')
+    file_object.write(str(t_0_5_MaxValue))
+    file_object.write(';')
+    file_object.write(str(t_1_0_MaxValue))
+    file_object.write(';')
+    file_object.write(str(t_2_0_MaxValue))
+    file_object.write(';')
+    # file_object.write(str(Acc_0_5__1_0_MaxValue))
+    # file_object.write(';')
+    file_object.write(str(Abs_Diff_t_0_5_MaxValue))
+    file_object.write(';')
+    file_object.write(str(Abs_Diff_t_1_0_MaxValue))
+    file_object.write(';')
+    file_object.write(str(Abs_Diff_t_2_0_MaxValue))
+    file_object.write(';')
+    file_object.write(Optional_Custom_Message)
+    file_object.write('\n')
+    file_object.close()
+
+    result_2way = loaded_rf_2way.predict([X])
+    print('The patient is ', outcome_decoded[int(result_2way)], ' through the 2way method')
+
+    result_3way = loaded_rf_3way.predict([X])
+    if result_2way == 0:
+        print('The patient is ', outcome_decoded[int(result_3way)], 'through the 3way method')
+        # result = 'The 3-way classification resulted in a ', outcome_decoded[int(result_3way)] + ' patient.'
+        # further_analysis = 'Futher analysis using the 2-way classification resulted in a ' + outcome_decoded[int(result_2way)] + ' label.'
+        return 'The 3-way classification resulted in a ' + outcome_decoded[int(result_3way)] + ' patient. Futher analysis using the 2-way classification resulted in a ' + outcome_decoded[int(result_2way)] + ' label.'
+
+    # result = 'The 2-way classification resulted in a ', outcome_decoded[int(result_2way)] + ' patient.'
+    # further_analysis = 'Futher analysis using the 3-way classification resulted in a ' + outcome_decoded[int(result_3way)] + ' label.'
+
+    return 'The 2-way classification resulted in a ' + outcome_decoded[int(result_2way)] + ' patient. Futher analysis using the 3-way classification resulted in a ' + outcome_decoded[int(result_3way)] + ' label.'
+
+iface = gr.Interface(
+    fn=STPI, 
+    title='STPI Calculator',
+    description='Calculates the STPI through summarized tomographic parameters. Beta version by Prof. Shady Awwad, Jad Assaf MD and Jawad Kaisania.',
+    inputs=["number", "number","number", 
+    # "number",
+    "number", "number","number","text"],
+    outputs="text")
+iface.launch(share=True)
+# %%

requirements.txt

@@ -0,0 +1,3 @@
+gradio==2.4.6
+joblib==1.0.0
+scikit-learn==0.24.0

stpi_data.txt

@@ -0,0 +1 @@
+11.0;22.0;33.0;44.0;55.0;66.0;77.0