Update app.py

app.py

@@ -29,6 +29,38 @@ date_time_str = now.strftime("%Y-%m-%d %H:%M:%S")
 #client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 history = []
 
+def gen_from_infer(in_prompt,history,model_drop):
+    #out_img = infer(out_prompt)
+    out_prompt=generate(purpose,history,chat_drop)
+    history.append((purpose,out_prompt))
+    yield (history,None)        
+
+    out_img=InferenceClient().text_to_image(
+    prompt=out_prompt,
+    negative_prompt=None,
+    height=512,
+    width=512,
+    num_inference_steps=None,
+    guidance_scale=None,
+    model=loaded_model[int(model_drop)],
+    seed=1,
+    )
+
+    #return (history,None)
+    print(out_img)
+    url=f'https://johann22-chat-diffusion-describe.hf.space/file={out_img}'
+    print(url)
+    uid = uuid.uuid4()
+    #urllib.request.urlretrieve(image, 'tmp.png')
+    #out=Image.open('tmp.png')
+    r = requests.get(url, stream=True)
+    if r.status_code == 200:
+        out = Image.open(io.BytesIO(r.content))
+        #yield ([(purpose,out_prompt)],out)
+        yield (history,out)
+    else:
+        yield ([(purpose,"an Error occured")],None)
+
 def format_prompt(message, history):
   prompt = "<s>"
   for user_prompt, bot_response in history:
@@ -155,8 +187,12 @@ with gr.Blocks(css=style) as iface:
                     submit_b = gr.Button()
                     stop_b = gr.Button("Stop")
                     clear = gr.ClearButton([msg, chatbot])
+            test_btn = gr.Button("Test")
         with gr.Column(scale=2):
             sumbox=gr.Image(label="Image")
+
+    run_test = test_btn.click(gen_from_infer, [msg,chatbot,sumbox,model_drop,chat_model_drop,agent_choice],[chatbot,sumbox],concurrency_limit=20)
+    
     sub_b = submit_b.click(run, [msg,chatbot,sumbox,model_drop,chat_model_drop,agent_choice],[chatbot,sumbox],concurrency_limit=20)
     sub_e = msg.submit(run, [msg, chatbot,sumbox,model_drop,chat_model_drop,agent_choice], [chatbot,sumbox],concurrency_limit=20)
     stop_b.click(None,None,None, cancels=[sub_b,sub_e])