root16285 commited on
Commit
1f3d9ce
·
1 Parent(s): a400715

Fix: Use dynamic URLs for API and WebSocket to support Hugging Face deployment

Browse files
Files changed (1) hide show
  1. webapp/static/app.js +7 -3
webapp/static/app.js CHANGED
@@ -1,5 +1,8 @@
1
  // YOLOv5 Web App - Main JavaScript
2
- const API_URL = 'http://localhost:8001';
 
 
 
3
  let ws = null;
4
  let webcamStream = null;
5
  let isWebcamActive = false;
@@ -153,7 +156,8 @@ function stopWebcamDetection() {
153
  }
154
 
155
  function connectWebSocket() {
156
- ws = new WebSocket('ws://localhost:8001/ws');
 
157
 
158
  ws.onopen = () => {
159
  console.log('✅ WebSocket connecté avec succès!');
@@ -173,7 +177,7 @@ function connectWebSocket() {
173
 
174
  ws.onerror = (error) => {
175
  console.error('❌ WebSocket error:', error);
176
- alert('Erreur WebSocket. Vérifiez que le serveur est actif sur localhost:8001');
177
  };
178
 
179
  ws.onclose = () => {
 
1
  // YOLOv5 Web App - Main JavaScript
2
+ // Dynamic API URL detection for both local and Hugging Face deployment
3
+ const API_URL = window.location.protocol + '//' + window.location.host;
4
+ const WS_URL = (window.location.protocol === 'https:' ? 'wss://' : 'ws://') + window.location.host + '/ws';
5
+
6
  let ws = null;
7
  let webcamStream = null;
8
  let isWebcamActive = false;
 
156
  }
157
 
158
  function connectWebSocket() {
159
+ console.log('🔌 Connexion WebSocket à:', WS_URL);
160
+ ws = new WebSocket(WS_URL);
161
 
162
  ws.onopen = () => {
163
  console.log('✅ WebSocket connecté avec succès!');
 
177
 
178
  ws.onerror = (error) => {
179
  console.error('❌ WebSocket error:', error);
180
+ alert('Erreur WebSocket. Vérifiez la connexion au serveur.');
181
  };
182
 
183
  ws.onclose = () => {