fix: docker compose support

AntonKluge · Jul 14, 2024 · 948544f · 948544f
1 parent de11a31
commit 948544f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -18,6 +18,8 @@ services:
       dockerfile: Dockerfile
     ports:
       - 3001:3000
+    environment:
+      - NEXT_PUBLIC_API_URL=http://lens-api:5000
     restart: always
     depends_on:
       - lens-api
diff --git a/frontend/components/image-classification.tsx b/frontend/components/image-classification.tsx
@@ -1,5 +1,6 @@
 'use client';
 
+import Config from '@/lib/config';
 import {
   ClassificationProduct,
   ClassificationProductDescription,
@@ -39,7 +40,7 @@ export default function ImageClassification() {
       const formData = new FormData();
       formData.append('file', image);
 
-      fetch('http://localhost:3002/classify', {
+      fetch(`${Config.apiUrl}/classify`, {
         method: 'POST',
         body: formData,
       }).then(async (response) => {
@@ -128,7 +129,7 @@ function ClassificationCard({ data }: { data: ClassificationProduct }) {
   };
 
   return (
-    <div className="bg-white rounded-lg border-gray-200 border-[1px] p-4 max-h-32 overflow-auto">
+    <div className="bg-white rounded-lg border-gray-200 border-[1px] p-4 max-h-64 overflow-auto">
       <h2 className="text-lg font-bold">{cardTitle[data.data_description]}</h2>
       {buildCard}
     </div>

diff --git a/frontend/lib/config.ts b/frontend/lib/config.ts
@@ -0,0 +1,5 @@
+export default class Config {
+  static get apiUrl(): string {
+    return process.env.NEXT_PUBLIC_API_URL || 'http://localhost:3002';
+  }
+}