Update app.py

app.py

@@ -4,7 +4,9 @@ import gradio as gr
 from pydantic_settings import BaseSettings
 from tempfile import NamedTemporaryFile
 import sweetviz as sv
-from datatile.summary.df import DataFrameSummary
+#from datatile.summary.df import DataFrameSummary
+from autoviz.AutoViz_Class import AutoViz_Class
+from traceml.summary.df import DataFrameSummary
 
 
 def generate_report(file, type):
@@ -48,13 +50,29 @@ def generate_report(file, type):
     temp_file3.write(contents.encode('utf-8'))
     temp_file3.close()
 
-    
+    #AUTOVIZ
+    AV = AutoViz_Class()
+
+    dfviz = AV.AutoViz(
+    "",
+    sep=",",
+    depVar=target,
+    dfte=dataframe,
+    header=0,
+    verbose=2,
+    lowess=False,
+    chart_format="jpg",
+    max_rows_analyzed=500,
+    max_cols_analyzed=20,
+    save_plot_dir="plots",
+    )
+
     
     
    
     dfs = DataFrameSummary(df)
     
-    return temp_file1.name  ,temp_file3.name ,dfs.summary()
+    return temp_file1.name  ,temp_file3.name ,dfviz,dfs.summary()
     
      
 with gr.Blocks() as cluster:
@@ -74,6 +92,7 @@ with gr.Blocks() as cluster:
             gr.HTML(value="""<h1 style="color: #3399FF; text-shadow: 1px 1px 2px #ddd;">SWEETVIZ REPORT</h1>""")
             out3=gr.File(label="Download CSV")
         with gr.Row():
-            dataframe=gr.Dataframe()
-    btn.click(generate_report,inputs=[file],outputs=[out1,out3,dataframe])
+            dataframe1=gr.Dataframe()
+            dataframe2=gr.Dataframe()
+    btn.click(generate_report,inputs=[file],outputs=[out1,out3,dataframe1,dataframe2])
 cluster.launch()