Fixed state and image upload

app.py

@@ -19,13 +19,14 @@ MODEL_DIR = './fromage_model/fromage_vis4'
 
 class ChatBotCheese:
     def __init__(self):
-        from huggingface_hub import hf_hub_download
-        model_ckpt_path = hf_hub_download("alvanlii/fromage", "pretrained_ckpt.pth.tar")
-        self.model = models.load_fromage(MODEL_DIR, model_ckpt_path)
+        # from huggingface_hub import hf_hub_download
+        # model_ckpt_path = hf_hub_download("alvanlii/fromage", "pretrained_ckpt.pth.tar")
+        # self.model = models.load_fromage(MODEL_DIR, model_ckpt_path)
+        self.model = models.load_fromage(MODEL_DIR)
         self.curr_image = None
 
     def add_image(self, state, image_in):
-        state = state + [(f"![](/file={image_in.name})", "Ok, now type your message")]
+        state = state + [(f'<img src="/file={image_in.name}">', "Ok, now type your message")]
         self.curr_image = Image.open(image_in.name).convert('RGB')
         return state, state
 
@@ -36,9 +37,9 @@ class ChatBotCheese:
 
     def chat(self, input_text, state, ret_scale_factor, num_ims, num_words, temp, chat_state):
         chat_state.append(f'Q: {input_text} \nA:')
-        chat_history = " ".join(chat_state)
+        chat_history = "".join(chat_state)
         model_input = []
-        print(chat_history)
+        # print(chat_history)
         if self.curr_image is not None:
             model_input = [self.curr_image, chat_history]
         else:
@@ -55,7 +56,7 @@ class ChatBotCheese:
         for im_name in im_names:
             response += f'<img src="/file={im_name}">'
         state.append((input_text, response.replace("[RET]", "")))
-        # self.curr_image = None
+        self.curr_image = None
         return state, state, chat_state    
 
     def reset(self):
@@ -111,10 +112,7 @@ class ChatBotCheese:
             txt.submit(lambda :"", None, txt)
             btn.upload(self.add_image, [gr_state, btn], [gr_state, chatbot])
             reset_btn.click(self.reset, [], [gr_state, chatbot, gr_chat_state])
-
-            # chatbot.change(fn = upload_button_config, outputs=btn_upload)
-            # text_in.submit(None, [], [], _js = "() => document.getElementById('#chatbot-component').scrollTop = document.getElementById('#chatbot-component').scrollHeight")
-
+            
         demo.launch(share=False, server_name="0.0.0.0")
 
 def main():