Update app.py

app.py

@@ -88,18 +88,15 @@ def run(purpose,history,model_drop):
     #    history = ""
 
     #action_name, action_input = parse_action(line)
-    out_prompt = run_gpt(
-        purpose,
-        history,
-        
-        )
+    out_prompt = run_gpt(purpose,history):
+    
 
-    yield ("",[(purpose,out_prompt)],None)
+    yield ([(purpose,out_prompt)],None)
     #out_img = infer(out_prompt)
     model=loaded_model[int(model_drop)]
     out_img=model(out_prompt)
     print(out_img)
-    url=f'https://johann22-mixtral-chat-diffusion.hf.space/file={out_img}'
+    url=f'https://johann22-chat-diffusion.hf.space/file={out_img}'
     print(url)
     uid = uuid.uuid4()
     #urllib.request.urlretrieve(image, 'tmp.png')
@@ -108,7 +105,8 @@ def run(purpose,history,model_drop):
     if r.status_code == 200:
         out = Image.open(io.BytesIO(r.content))
         yield ([(purpose,out_prompt)],out)
-        #return ("", [(purpose,history)])
+    else:
+        yield ([(purpose,"an Error occured")],None)
 
 
 
@@ -130,8 +128,8 @@ with gr.Blocks() as iface:
     sumbox=gr.Image(label="Image")
 
         
-    sub_b = submit_b.click(run, [msg,chatbot,model_drop],[msg,chatbot,sumbox])
-    sub_e = msg.submit(run, [msg, chatbot,model_drop], [msg, chatbot,sumbox])
+    sub_b = submit_b.click(run, [msg,chatbot,model_drop],[chatbot,sumbox])
+    sub_e = msg.submit(run, [msg, chatbot,model_drop], [chatbot,sumbox])
     stop_b.click(None,None,None, cancels=[sub_b,sub_e])
 iface.launch()
 '''