Update app.py

app.py

@@ -22,18 +22,8 @@ date_time_str = now.strftime("%Y-%m-%d %H:%M:%S")
 client = InferenceClient(
     "mistralai/Mixtral-8x7B-Instruct-v0.1"
 )
-
-############################################
-model = gr.load("models/stabilityai/sdxl-turbo")
-
-VERBOSE = True
-MAX_HISTORY = 10000
-#MODEL = "gpt-3.5-turbo"  # "gpt-4"
 history = []
 
-def infer(txt):
-    return (model(txt))
-
 def format_prompt(message, history):
   prompt = "<s>"
   for user_prompt, bot_response in history:
@@ -81,7 +71,6 @@ def run(purpose,history,model_drop):
     #print(purpose)
     #print(hist)
     task=None
-    directory="./"
     #if history:
     #    history=str(history).strip("[]")
     #if not history:
@@ -94,7 +83,7 @@ def run(purpose,history,model_drop):
         
         )
 
-    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)
@@ -107,8 +96,8 @@ def run(purpose,history,model_drop):
     r = requests.get(url, stream=True)
     if r.status_code == 200:
         out = Image.open(io.BytesIO(r.content))
-    yield ("",[(purpose,out_prompt)],out)
-        #return ("", [(purpose,history)])
+    yield ([(purpose,out_prompt)],out)
+        #return ([(purpose,history)])
 
 
 
@@ -118,20 +107,20 @@ with gr.Blocks() as iface:
     gr.HTML("""<center><h1>Chat Diffusion</h1><br><h3>This chatbot will generate images</h3></center>""")
     #chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
     with gr.Row():
-        with gr.Column():
+        with gr.Column(scale=1):
             chatbot=gr.Chatbot()
             msg = gr.Textbox()
             model_drop=gr.Dropdown(label="Diffusion Models", type="index", choices=[m for m in models], value=models[0])
-    with gr.Row():
-        submit_b = gr.Button()
-        stop_b = gr.Button("Stop")
-        clear = gr.ClearButton([msg, chatbot])
-    
-    sumbox=gr.Image(label="Image")
-
+            with gr.Group():
+                submit_b = gr.Button()
+                with gr.Row():
+                    stop_b = gr.Button("Stop")
+                    clear = gr.ClearButton([msg, chatbot])
+        with gr.Column(scale=2):
+            im_out=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,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])
 iface.launch()
 '''