Update app.py

app.py

@@ -16,6 +16,7 @@ DEFAULT_CKPT_PATH = 'qwen/Qwen-VL-Chat'
 REVISION = 'v1.0.4'
 BOX_TAG_PATTERN = r"<box>([\s\S]*?)</box>"
 PUNCTUATION = "！？。＂＃＄％＆＇（）＊＋，－／：；＜＝＞＠［＼］＾＿｀｛｜｝～｟｠｢｣､、〃》「」『』【】〔〕〖〗〘〙〚〛〜〝〞〟〰〾〿–—‘’‛“”„‟…‧﹏."
+uploaded_file_dir = os.environ.get("GRADIO_TEMP_DIR") or str(Path(tempfile.gettempdir()) / "gradio")
 
 def _get_args():
     parser = ArgumentParser()
@@ -36,10 +37,10 @@ def _get_args():
     args = parser.parse_args()
     return args
 
-def handle_image_submission(_chatbot, task_history, file):
+def handle_image_submission(_chatbot, task_history, file, uploaded_file_dir):
     if file is None:
-        return _chatbot, task_history  # Return if no file is uploaded
-    file_path = save_image(file)
+        return _chatbot, task_history  
+    file_path = save_image(file, uploaded_file_dir)
     history_item = ((file_path,), None)
     _chatbot.append(history_item)
     task_history.append(history_item)
@@ -101,7 +102,7 @@ def _parse_text(text):
                 lines[i] = "<br>" + line
     text = "".join(lines)
     return text
-def save_image(image_file):
+def save_image(image_file, uploaded_file_dir):
     upload_dir = Path(uploaded_file_dir) / "uploads"
     upload_dir.mkdir(parents=True, exist_ok=True)
     filename = secrets.token_hex(10) + Path(image_file.name).suffix
@@ -151,7 +152,7 @@ def _launch_demo(args, model, tokenizer):
     
         return _chatbot
 
-    def save_image(image):
+    def save_image(image, uploaded_file_dir):
         if image is None:
             return None
         temp_dir = secrets.token_hex(20)