feat: improve interface

app.py

@@ -16,11 +16,15 @@ def predict(df):
     health_insurance = HealthInsurance(_model,_column_transformer,
                                             _bins_annual_premium_type)
     
-    df_predicted = health_insurance.predict(df)   
+    df_predicted = health_insurance.predict(df)  
 
-    return df_predicted
+    return df_predicted[['score','previously_insured',
+                                    'annual_premium','vintage','gender',
+                                    'age','region_code','policy_sales_channel',
+                                    'driving_license','vehicle_age',
+                                    'vehicle_damage']] 
 
-def input():
+def create_input_table():
     return gr.Dataframe(headers = ['previously_insured',
                                     'annual_premium','vintage','gender',
                                     'age','region_code','policy_sales_channel',
@@ -33,24 +37,35 @@ def input():
                         type = 'pandas',
                         label = 'Input')
 
-def output():
-    return gr.Dataframe(headers = ['previously_insured',
+def create_output_table():
+    return gr.Dataframe(headers = ['score','previously_insured',
                                     'annual_premium','vintage','gender',
                                     'age','region_code','policy_sales_channel',
                                     'driving_license','vehicle_age',
-                                    'vehicle_damage', 'score'],
-                        datatype = ['number','number','number','str','number',
-                                    'number','number','number','str','str',
-                                    'number'],
+                                    'vehicle_damage'],
                         type = 'pandas',
-                        label = 'Output')
+                        label = 'Output',
+                        wrap = True,
+                        interactive =False)
+
+def create_file_object():
+    return gr.File(label = 'File upload')
 
-if __name__ == "__main__":
 
+
+def build_interface():
+
+    with gr.Blocks() as interface:
+        file_object = create_file_object()
+        input_table = create_input_table()
+        output_table = create_output_table()
+
+        greet_btn = gr.Button("Submit")
+        greet_btn.click(fn=predict, inputs=input_table, outputs=output_table)
+    
+    interface.launch()
+
+
+if __name__ == "__main__":
     load_data()
-    iface = gr.Interface(fn=predict, 
-                        inputs=input(), 
-                        outputs=output(),
-                        title='Health Insurance App'
-                        )
-    iface.launch()
+    build_interface()