update package

.gitignore

@@ -2,4 +2,6 @@
 weights**
 .conda**
 .venv
-tmp**
+tmp**
+build**
+dist**

README.md

@@ -1,6 +1,6 @@
 <div align="center"><a name="readme-top"></a>
 
-<img src="./resources/logo.png" width="120" height="120" alt="autoMate logo">
+<img src="./imgs/logo.png" width="120" height="120" alt="autoMate logo">
 <h1>autoMate</h1>
 <p><b>🤖 AI-powered Local Automation Tool | Make Your Computer Work For You</b></p>
 
@@ -16,7 +16,7 @@ https://github.com/user-attachments/assets/bf27f8bd-136b-402e-bc7d-994b99bcc368
 > Special Note: The autoMate project is still in a very early stage and is currently more for learning purposes. We are constantly looking for breakthroughs and continuously integrating the latest technologies! If you have any questions, you can also add WeChat friends to join the group for communication.
 
 <div align="center">
-<img src="./resources/wxchat.png" width="120" height="120" alt="autoMate logo">
+<img src="./imgs/wxchat.png" width="120" height="120" alt="autoMate logo">
 </div>
 
 

README_CN.md

@@ -1,6 +1,6 @@
 <div align="center"><a name="readme-top"></a>
 
-<img src="./resources/logo.png" width="120" height="120" alt="autoMate logo">
+<img src="./imgs/logo.png" width="120" height="120" alt="autoMate logo">
 <h1>autoMate</h1>
 <p><b>🤖 AI驱动的本地自动化工具 | 让电脑自己会干活</b></p>
 
@@ -16,7 +16,7 @@ https://github.com/user-attachments/assets/bf27f8bd-136b-402e-bc7d-994b99bcc368
 > 特别声明：autoMate 项目还处于非常早期阶段，当前更多用于学习。我们会不断的寻求突破点，不停地融入最新的技术！如果你有任何疑问，也可以加vx好友入群交流。
 
 <div align="center">
-<img src="./resources/wxchat.png" width="120" height="120" alt="autoMate logo">
+<img src="./imgs/wxchat.png" width="120" height="120" alt="autoMate logo">
 </div>
 
 

README_JA.md

@@ -1,6 +1,6 @@
 <div align="center"><a name="readme-top"></a>
 
-<img src="./resources/logo.png" width="120" height="120" alt="autoMate logo">
+<img src="./imgs/logo.png" width="120" height="120" alt="autoMate logo">
 <h1>autoMate</h1>
 <p><b>🤖 AIで動くローカル自動化ツール | コンピューターに自ら仕事をさせる</b></p>
 
@@ -16,7 +16,7 @@ https://github.com/user-attachments/assets/bf27f8bd-136b-402e-bc7d-994b99bcc368
 > 特別声明：autoMateプロジェクトはまだ非常に初期段階にあり、現在は主に学習目的で使用されています。私たちは常にブレークスルーを求め、最新技術を取り入れています！質問がある場合は、WeChat友達追加してグループチャットに参加することもできます。
 
 <div align="center">
-<img src="./resources/wxchat.png" width="120" height="120" alt="autoMate logo">
+<img src="./imgs/wxchat.png" width="120" height="120" alt="autoMate logo">
 </div>
 
 

main.py

@@ -1,11 +1,12 @@
 from gradio_ui import app
-from util import download_weights
-
 import torch
-
+import os
+from util import download_weights
+os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
 def run():
     if not torch.cuda.is_available():
         print("Warning: GPU is not available, we will use CPU, the application may run slower!\nyou computer will very likely heat up!")
+    download_weights.download() 
     app.run()
 
 

main.spec

@@ -0,0 +1,59 @@
+# -*- mode: python ; coding: utf-8 -*-
+from PyInstaller.utils.hooks import collect_data_files
+import os
+import shutil
+
+datas = []
+datas += collect_data_files('gradio_client')
+datas += collect_data_files('gradio')
+datas += collect_data_files('safehttpx')
+datas += collect_data_files('groovy')
+base_path = os.path.dirname(os.path.abspath('__file__'))
+imgs_path = os.path.join(base_path, 'imgs')
+datas += [(imgs_path, 'imgs')]
+
+a = Analysis(
+    ['main.py'],
+    pathex=[],
+    binaries=[],
+    datas=datas,
+    hiddenimports=[],
+    hookspath=[],
+    hooksconfig={},
+    runtime_hooks=[],
+    excludes=[],
+    noarchive=False,
+    optimize=0,
+    module_collection_mode={
+        'gradio': 'py',
+    },
+)
+pyz = PYZ(a.pure)
+
+exe = EXE(
+    pyz,
+    a.scripts,
+    [],
+    exclude_binaries=True,
+    name='main',
+    debug=False,
+    bootloader_ignore_signals=False,
+    strip=False,
+    upx=True,
+    console=True,
+    disable_windowed_traceback=False,
+    argv_emulation=False,
+    target_arch=None,
+    codesign_identity=None,
+    entitlements_file=None,
+)
+coll = COLLECT(
+    exe,
+    a.binaries,
+    a.datas,
+    strip=False,
+    upx=True,
+    upx_exclude=[],
+    name='main',
+)
+

requirements.txt

@@ -12,4 +12,5 @@ pyxbrain==1.1.31
 timm
 einops==0.8.0
 modelscope
-pynput
+pynput
+lap

util/download_weights.py

@@ -10,6 +10,7 @@ def download():
     snapshot_download(
         'AI-ModelScope/OmniParser-v2.0',
         cache_dir='weights',
+        allow_file_pattern=['icon_detect/model.pt']
         )
     
 if __name__ == "__main__":

util/tool.py

@@ -6,7 +6,7 @@ from io import BytesIO
 
 def capture_screen_with_cursor():
     """Local function to capture the screen with cursor."""
-    cursor_path = os.path.join(os.path.dirname(__file__),"..","resources", "cursor.png")
+    cursor_path = os.path.join(os.path.dirname(__file__),"..","imgs", "cursor.png")
     screenshot = pyautogui.screenshot()
     cursor_x, cursor_y = pyautogui.position()
     cursor = Image.open(cursor_path)