Update app.py

app.py

@@ -24,6 +24,7 @@ agents =[
     "MEME_GENERATOR",
 ]
 def generate(prompt, history):
+    output1={}
     seed = random.randint(1,1111111111111111)
 
     system_prompt=MEME_GENERATOR
@@ -63,11 +64,12 @@ def generate(prompt, history):
                 print(e)
                 pass
     '''
-    output=output.replace("\n"," ")
-    if "PROMPT:" and "MEME_TEXT:" in output:
-        output={'PROMPT':output.split("PROMPT:",1)[1].split("MEME_TEXT",1)[0],'MEME_TEXT':output.split("MEME_TEXT",1)[1]}
-        output=str(output)
-    return [(prompt,output)]
+    output1=output.replace("\n"," ")
+    if "PROMPT:" and "MEME_TEXT:" in output1:
+        print("YES")
+        output1={'PROMPT':output1.split("PROMPT:",1)[1].split("MEME_TEXT",1)[0],'MEME_TEXT':output1.split("MEME_TEXT",1)[1]}
+        output=str(output1)
+    return [(prompt,output)],output1
     
 def run(inp,history,model_drop):
     #if len(inp)>max_prompt:
@@ -144,8 +146,9 @@ with gr.Blocks() as app:
                     clear = gr.ClearButton([msg, chatbot])
         with gr.Column(scale=2):
             im_out=gr.Image(label="Image")
+    json_out=gr.JSON()
         
-    sub_b = submit_b.click(generate, [msg,chatbot],[chatbot])
+    sub_b = submit_b.click(generate, [msg,chatbot],[chatbot,json_out])
     sub_im = submit_im.click(run, [msg,chatbot,model_drop],[chatbot,im_out])
     sub_e = msg.submit(run, [msg, chatbot,model_drop], [chatbot,im_out])
     stop_b.click(None,None,None, cancels=[sub_b,sub_e])