SissiFeng commited on
Commit
3637706
·
1 Parent(s): 5bcfffd

Update: Remove database dependency and optimize code

Browse files
Files changed (2) hide show
  1. app.py +0 -14
  2. requirements.txt +8 -7
app.py CHANGED
@@ -12,7 +12,6 @@ import logging
12
  from dotenv import load_dotenv
13
  from mo_optimizer import MOPrintOptimizer
14
  from core.analysis import DefectDetector, ImageProcessor
15
- from core.database import PrintJob
16
  from datetime import datetime
17
  import requests
18
  from PIL import Image
@@ -106,19 +105,6 @@ def on_message(client, userdata, message):
106
  })
107
  # update interface display
108
  s3_image.update(value=image_url)
109
-
110
- # record to database
111
- if hasattr(demo, 'db_manager'):
112
- job = demo.db_manager.session.query(PrintJob).filter_by(
113
- square_id=data.get("square_id"),
114
- status="printing"
115
- ).first()
116
- if job:
117
- job.match_image(
118
- image_url,
119
- datetime.strptime(data.get("timestamp"), "%Y-%m-%d %H:%M:%S")
120
- )
121
- demo.db_manager.session.commit()
122
 
123
  elif "status" in message.topic:
124
  # update status
 
12
  from dotenv import load_dotenv
13
  from mo_optimizer import MOPrintOptimizer
14
  from core.analysis import DefectDetector, ImageProcessor
 
15
  from datetime import datetime
16
  import requests
17
  from PIL import Image
 
105
  })
106
  # update interface display
107
  s3_image.update(value=image_url)
 
 
 
 
 
 
 
 
 
 
 
 
 
108
 
109
  elif "status" in message.topic:
110
  # update status
requirements.txt CHANGED
@@ -2,25 +2,26 @@
2
  gradio>=4.0.0
3
 
4
  # Core Data Processing
5
- numpy>=1.20.0
6
  pandas>=1.3.0
7
  scipy>=1.7.0
8
 
9
  # Image Processing
10
- opencv-python-headless>=4.5.0
11
- scikit-image>=0.19.0
12
- Pillow>=8.0.0
13
 
14
  # Machine Learning
15
- scikit-learn>=0.24.0
16
  hdbscan>=0.8.29
17
 
18
  # Communication
19
  paho-mqtt>=1.6.1
20
 
21
  # Configuration
22
- python-dotenv>=0.19.0
23
 
24
  # Utilities
25
  tqdm>=4.62.0
26
- matplotlib>=3.4.0
 
 
2
  gradio>=4.0.0
3
 
4
  # Core Data Processing
5
+ numpy>=1.24.0
6
  pandas>=1.3.0
7
  scipy>=1.7.0
8
 
9
  # Image Processing
10
+ opencv-python>=4.8.0
11
+ scikit-image>=0.21.0
12
+ Pillow>=10.0.0
13
 
14
  # Machine Learning
15
+ scikit-learn>=1.3.0
16
  hdbscan>=0.8.29
17
 
18
  # Communication
19
  paho-mqtt>=1.6.1
20
 
21
  # Configuration
22
+ python-dotenv>=1.0.0
23
 
24
  # Utilities
25
  tqdm>=4.62.0
26
+ matplotlib>=3.4.0
27
+ requests>=2.31.0