image preview fix

frontend/src/pages/UploadPage/UploadPage.tsx

@@ -558,7 +558,7 @@ export default function UploadPage() {
         {isLoading && (
           <div className={styles.loadingContainer}>
             <Spinner className="text-ifrcRed" />
-            <p className={styles.loadingText}>Generating caption...</p>
+                                <p className={styles.loadingText}>Generating...</p>
           </div>
         )}
 

py_backend/app/config.py

@@ -15,7 +15,7 @@ class Settings(BaseSettings):
     STORAGE_PROVIDER: str = "local"
     STORAGE_DIR: str = "/data/uploads"
     HF_HOME: str = "/data/.cache/huggingface"
-    BASE_URL: str = "http://localhost:8080"
+    BASE_URL: str = "https://huggingface.co/spaces/promptAId/Promptaid-VIsion" 
     
     class Config:
         env_file = ".env"

py_backend/app/routers/caption.py

@@ -144,7 +144,7 @@ async def create_caption(
     except Exception:
         url = f"/api/images/{c.image_id}/file"
     
-    if url and url.startswith('/'):
+    if url and url.startswith('/') and settings.BASE_URL:
         url = f"{settings.BASE_URL}{url}"
     
     img_dict = convert_image_to_dict(c, url)
@@ -173,7 +173,7 @@ def get_caption(
         url = f"/api/images/{caption.image_id}/file"
     
 
-    if url and url.startswith('/'):
+    if url and url.startswith('/') and settings.BASE_URL:
         url = f"{settings.BASE_URL}{url}"
     
     img_dict = convert_image_to_dict(caption, url)
@@ -200,7 +200,7 @@ def get_captions_by_image(
         except Exception:
             url = f"/api/images/{caption.image_id}/file"
         
-        if url and url.startswith('/'):
+        if url and url.startswith('/') and settings.BASE_URL:
             url = f"{settings.BASE_URL}{url}"
         
         img_dict = convert_image_to_dict(caption, url)
@@ -259,7 +259,7 @@ def update_caption(
     except Exception:
         url = f"/api/images/{caption.image_id}/file"
     
-    if url and url.startswith('/'):
+    if url and url.startswith('/') and settings.BASE_URL:
         url = f"{settings.BASE_URL}{url}"
     
     img_dict = convert_image_to_dict(caption, url)

py_backend/app/routers/upload.py

@@ -36,7 +36,8 @@ def convert_image_to_dict(img, image_url):
             print(f"Warning: Error processing countries for image {img.image_id}: {e}")
             countries_list = []
     
-    if image_url and image_url.startswith('/'):
+    # Handle image URL - if BASE_URL is set, use it; otherwise keep relative URLs
+    if image_url and image_url.startswith('/') and settings.BASE_URL:
         image_url = f"{settings.BASE_URL}{image_url}"
     
     img_dict = {