fix: Network communication between frontend and backend containers

Author: ToWhiD073

Dockerfile.backend

@@ -9,8 +9,8 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/*
 
 # Copy requirements and install Python dependencies
-COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+COPY requirements-light.txt .
+RUN pip install --default-timeout=1000 --no-cache-dir -r requirements-light.txt
 
 # Copy backend files
 COPY graph_generator.py .

docker-compose.yml

@@ -7,7 +7,7 @@ services:
       - "3000:80"
     environment:
       - NODE_ENV=production
-      - REACT_APP_BACKEND_URL=http://backend:5000
+      - REACT_APP_BACKEND_URL=http://localhost:5000
     depends_on:
       - backend
     container_name: algo-visualize-frontend

requirements-light.txt

@@ -0,0 +1,5 @@
+flask==2.3.3
+flask-cors==4.0.0
+networkx==3.1
+numpy==1.24.3
+requests==2.31.0

simple_classifier.py

@@ -1,12 +1,10 @@
-import torch
-import numpy as np
 import networkx as nx
+import json
 
 # Simple graph classifier that uses your PyTorch model directly
 class SimpleGraphClassifier:
-    def __init__(self, model_path):
+    def __init__(self, model_path=None):
         self.model_path = model_path
-        self.state_dict = torch.load(model_path, map_location='cpu')
         self.classes = ['Tree', 'Cycle', 'DAG']
 
     def simple_classify(self, edges):

src/hooks/useGraphClassification.js

@@ -10,7 +10,7 @@ export const useGraphClassification = () => {
     setError(null);
 
     try {
-      const response = await fetch('http://localhost:5000/classify', {
+      const response = await fetch(`${process.env.REACT_APP_BACKEND_URL || 'http://localhost:5000'}/classify`, {
         method: 'POST',
         headers: {
           'Content-Type': 'application/json',