Update run.py

run.py

@@ -58,6 +58,7 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
                             with open(_v.name, "wb") as f:
                                 _ = f.write(binary)
                             _chunk.append(_v.name)
+                            md5 = hashlib.md5(_v).hexdigest()
                             
                             # processor = VideoProcessor(frame_format=frame_format, frame_limit=frame_limit)
                             # frames = processor._decode(_v.name)
@@ -65,9 +66,8 @@ def fast_caption(sys_prompt, usr_prompt, temp, top_p, max_tokens, model, key, en
                             # api = AzureAPI(key=key, endpoint=endpoint, model=model, temp=temp, top_p=top_p, max_tokens=max_tokens)
                             # caption = api.get_caption(sys_prompt, usr_prompt, base64_list)
                             #writer.writerow({'md5': _v.name, 'caption': caption})
-                            writer.writerow({'md5': _v.name, 'caption': 'caption'})
-                            progress_info.append(f"Processed video with MD5: {_v.name}")
-                    return csv_filename, "\n".join(progress_info), None
+                            writer.writerow({'md5': md5, 'caption': 'caption'})
+                            progress_info.append(f"Processed video with MD5: {md5}")
                 return csv_filename, "\n".join(progress_info), None
             else:
                 return "", "No video source selected.", None