Update app.py

app.py

@@ -8,6 +8,7 @@ import sweetviz as sv
 from autoviz.AutoViz_Class import AutoViz_Class
 from traceml.summary.df import DataFrameSummary
 import os
+df1=gr.State()
 
 def variable_table(df):
     """
@@ -71,6 +72,7 @@ def variable_table(df):
 
 def generate_report(file, type):
     df = pd.read_csv(file) if file.name.endswith(".csv") else pd.read_excel(file)
+    
 
     pandas_html_report =ydata_profiling.ProfileReport(df).to_html() 
     temp_file1 = NamedTemporaryFile(delete=False, suffix=".html")
@@ -92,7 +94,7 @@ def generate_report(file, type):
 
     df = convert_mixed_to_str(df.copy())
     df.columns = df.columns.str.upper()
-
+    df1.value=df
     sweetviz_report = sv.analyze(df)
     #sweetviz_report=sweetviz_report.show_html()
     #print(type(sweetviz_report))
@@ -179,6 +181,10 @@ def generate_report(file, type):
 
     
     return [temp_file1.name  ,temp_file3.name ,variable_table(df),dfviz,sd1]+reports+imgs
+
+def query_chat(query):
+    df=df1.value
+    return df.sketch.ask(query)
     
 def handle_click(file, sheet_name):
     print('Handle click')  
@@ -226,7 +232,20 @@ with gr.Blocks() as cluster:
             gr.HTML(value="""<h2 style="color: #BA4A00; text-shadow: 1px 1px 2px #ddd;"COLUMNS STATS AND ANALYSIS</h2>""")
             #gr.Markdown("#Columns Analysis")
             dataframe2=gr.Dataframe()
+
+        with gr.Column():
+            gr.HTML(value="""<h2 style="color: #BA4A00; text-shadow: 1px 1px 2px #ddd;"ASK QUESTIONS ON THE COLUMNS </h2>""")
+            with gr.Row():
+                query=gr.Textbox(label="Enter any question on data columns",
+                info="Natural language query",
+                lines=1,
+                value="What is maximum value of first column")
+                outlabel=gr.HighlightedText(label='Query Result ')
+            query_btn=gr.Button(value='Submit Query')
+            query_btn.click(query_chat,inputs=query,outputs=outlabel)
             
+            
+
         #Autoviz insights graphs
   
         gr.HTML(value="""<h3 style="color: #1999FF; text-shadow: 1px 1px 2px #ddd;">Insight Visualisations</h3>""")