Update app.py

app.py

@@ -289,7 +289,7 @@ def compress_data_og(c, instruct, history):
 
 
 
-def summarize(inp,history,data=None,files=None,url=None,pdf_url=None,pdf_batch=None):
+def summarize(inp,history,report_check,data=None,files=None,url=None,pdf_url=None,pdf_batch=None):
     json_box=[]
     if inp == "":
         inp = "Process this data"
@@ -366,14 +366,17 @@ def summarize(inp,history,data=None,files=None,url=None,pdf_url=None,pdf_batch=N
         
         #json_box.append(json_out)
         out = str(json_out)
-        rl = len(out)
-        print(f'rl:: {rl}')
-        c=1
-        for i in str(out):
-            if i == " " or i=="," or i=="\n":
-                c +=1
-        print (f'c2:: {c}')
-        rawp = compress_data_og(c,inp,out)
+        if report_check:
+            rl = len(out)
+            print(f'rl:: {rl}')
+            c=1
+            for i in str(out):
+                if i == " " or i=="," or i=="\n":
+                    c +=1
+            print (f'c2:: {c}')
+            rawp = compress_data_og(c,inp,out)
+        else:
+            rawp = out
     else:
         rawp = "Provide a valid data source"
     #print (rawp)
@@ -395,6 +398,7 @@ with gr.Blocks() as app:
         with gr.Column(scale=3):
             prompt=gr.Textbox(label = "Instructions (optional)")
         with gr.Column(scale=1):
+            report_check=gr.Checkbox(label="Return Report", value=True)
             button=gr.Button()
         
         #models_dd=gr.Dropdown(choices=[m for m in return_list],interactive=True)
@@ -417,6 +421,6 @@ with gr.Blocks() as app:
     #text=gr.JSON()
     #inp_query.change(search_models,inp_query,models_dd)
     clear_btn.click(clear_fn,None,[prompt,chatbot])
-    go=button.click(summarize,[prompt,chatbot,data,file,url,pdf_url,pdf_batch],[prompt,chatbot,e_box,json_out])
+    go=button.click(summarize,[prompt,chatbot,report_check,data,file,url,pdf_url,pdf_batch],[prompt,chatbot,e_box,json_out])
     stop_button.click(None,None,None,cancels=[go])
 app.launch(server_port=7860,show_api=False)