Update app.py

app.py

@@ -14,6 +14,7 @@ import uuid
 #from query import tasks
 from agent import (
     PREFIX,
+    GET_CHART,
     COMPRESS_DATA_PROMPT,
     COMPRESS_DATA_PROMPT_SMALL,
     LOG_PROMPT,
@@ -285,9 +286,17 @@ def compress_data_og(c, instruct, history):
     #history = "observation: {}\n".format(resp)
     return resp
 
-
-
-def summarize(inp,history,report_check,data=None,files=None,url=None,pdf_url=None,pdf_batch=None):
+def get_chart(inp):
+    seed=random.randint(1,1000000000)
+    
+    resp = run_gpt(
+        GET_CHART,
+        stop_tokens=["observation:", "task:", "action:", "thought:"],
+        max_tokens=8192,
+        seed=seed,
+    )    
+    return resp
+def summarize(inp,history,report_check,chart_check,data=None,files=None,url=None,pdf_url=None,pdf_batch=None):
     json_box=[]
     if inp == "":
         inp = "Process this data"
@@ -375,6 +384,9 @@ def summarize(inp,history,report_check,data=None,files=None,url=None,pdf_url=Non
             rawp = compress_data_og(c,inp,out)
         else:
             rawp = out
+        if chart_check:
+            error_box = get_chart(rawp)
+            
     else:
         rawp = "Provide a valid data source"
     #print (rawp)
@@ -397,6 +409,7 @@ with gr.Blocks() as app:
             prompt=gr.Textbox(label = "Instructions (optional)")
         with gr.Column(scale=1):
             report_check=gr.Checkbox(label="Return Report", value=True)
+            chart_check=gr.Checkbox(label="Return Chart", value=True)
             button=gr.Button()
         
         #models_dd=gr.Dropdown(choices=[m for m in return_list],interactive=True)
@@ -414,11 +427,11 @@ with gr.Blocks() as app:
             pdf_url = gr.Textbox(label="PDF URL")       
         with gr.Tab("PDF Batch"):
             pdf_batch = gr.Textbox(label="PDF URL Batch (comma separated)")
-    json_out=gr.JSON()
     e_box=gr.Textbox()
+    json_out=gr.JSON()
     #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,report_check,data,file,url,pdf_url,pdf_batch],[prompt,chatbot,e_box,json_out])
+    go=button.click(summarize,[prompt,chatbot,report_check,chart_check,data,file,url,pdf_url,pdf_batch],[prompt,chatbot,e_box,json_out])
     stop_button.click(None,None,None,cancels=[go])
 app.queue(default_concurrency_limit=20).launch(server_port=7860,show_api=False)