fix config

py_backend/app/config.py

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

py_backend/app/routers/caption.py

@@ -144,9 +144,6 @@ async def create_caption(
     except Exception:
         url = f"/api/images/{c.image_id}/file"
     
-    if url and url.startswith('/') and settings.BASE_URL:
-        url = f"{settings.BASE_URL}{url}"
-    
     img_dict = convert_image_to_dict(c, url)
     return schemas.ImageOut(**img_dict)
 
@@ -172,10 +169,6 @@ def get_caption(
     except Exception:
         url = f"/api/images/{caption.image_id}/file"
     
-
-    if url and url.startswith('/') and settings.BASE_URL:
-        url = f"{settings.BASE_URL}{url}"
-    
     img_dict = convert_image_to_dict(caption, url)
     return schemas.ImageOut(**img_dict)
 
@@ -200,9 +193,6 @@ def get_captions_by_image(
         except Exception:
             url = f"/api/images/{caption.image_id}/file"
         
-        if url and url.startswith('/') and settings.BASE_URL:
-            url = f"{settings.BASE_URL}{url}"
-        
         img_dict = convert_image_to_dict(caption, url)
         result.append(schemas.ImageOut(**img_dict))
     
@@ -259,9 +249,6 @@ def update_caption(
     except Exception:
         url = f"/api/images/{caption.image_id}/file"
     
-    if url and url.startswith('/') and settings.BASE_URL:
-        url = f"{settings.BASE_URL}{url}"
-    
     img_dict = convert_image_to_dict(caption, url)
     return schemas.ImageOut(**img_dict)
 

py_backend/app/routers/upload.py

@@ -36,9 +36,6 @@ 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('/') and settings.BASE_URL:
-        image_url = f"{settings.BASE_URL}{image_url}"
-    
     img_dict = {
         "image_id": img.image_id,
         "file_key": img.file_key,

py_backend/run_tests_simple.py

@@ -0,0 +1,45 @@
+#!/usr/bin/env python3
+"""Simple test runner for PromptAid Vision"""
+
+import subprocess
+import sys
+import os
+
+def main():
+    """Run the configuration test"""
+    print("Running PromptAid Vision Configuration Test...")
+    
+    try:
+        # Change to tests directory
+        os.chdir('tests')
+        
+        # Run the config test
+        result = subprocess.run([sys.executable, 'test_config.py'], 
+                              capture_output=True, text=True, timeout=60)
+        
+        print("\n" + "="*50)
+        print("CONFIGURATION TEST RESULTS")
+        print("="*50)
+        
+        if result.returncode == 0:
+            print("✅ Configuration test PASSED")
+            print("\nOutput:")
+            print(result.stdout)
+        else:
+            print("❌ Configuration test FAILED")
+            if result.stdout:
+                print("\nOutput:")
+                print(result.stdout)
+            if result.stderr:
+                print("\nErrors:")
+                print(result.stderr)
+        
+        return result.returncode == 0
+        
+    except Exception as e:
+        print(f"❌ Error running test: {e}")
+        return False
+
+if __name__ == "__main__":
+    success = main()
+    sys.exit(0 if success else 1)

py_backend/tests/README.md

@@ -1,16 +1,20 @@
-# PromptAid Vision Tests
+# PromptAid Vision Test Suite
 
-This folder contains all test files for the PromptAid Vision application.
+This directory contains comprehensive tests for the PromptAid Vision application.
 
-## Test Files
+## Test Organization
 
-### Core Application Tests
-- **`test_core.py`** - Core application functionality test
-- **`test_hf.py`** - Hugging Face integration test
+### Core Functionality Tests
+- **`test_core.py`** - Core application functionality, database operations, and basic API endpoints
+- **`test_config.py`** - Configuration validation and storage setup verification
 
-### Test Utilities
-- **`test.jpg`** - Test image for image processing tests
-- **`run_tests.py`** - Run all tests
+### API and Integration Tests
+- **`test_upload_flow.py`** - Complete image upload workflow testing
+- **`test_openai_integration.py`** - OpenAI GPT-4 Vision API integration tests
+- **`test_hf.py`** - Hugging Face Spaces deployment and integration tests
+
+### Frontend and UI Tests
+- **`test_explore_page.py`** - Frontend explore page functionality and data display
 
 ## Running Tests
 
@@ -23,70 +27,38 @@ python tests/run_tests.py
 ### Run Individual Tests
 ```bash
 cd py_backend
-python tests/test_core.py
-python tests/test_hf.py
+python tests/test_config.py          # Test configuration
+python tests/test_core.py           # Test core functionality
+python tests/test_upload_flow.py    # Test upload flow
+python tests/test_openai_integration.py  # Test OpenAI integration
+python tests/test_hf.py             # Test Hugging Face integration
+python tests/test_explore_page.py   # Test explore page
 ```
 
-## Prerequisites
-
-Before running the tests, ensure you have:
-
-1. **Environment Variables Set**:
-   ```bash
-   cp .env.example .env
-   ```
-
-2. **Required API Keys**:
-   - `HF_API_KEY` - Hugging Face API key
-   - `OPENAI_API_KEY` - OpenAI API key (for GPT-4 Vision)
-   - `GOOGLE_API_KEY` - Google API key (for Gemini Vision)
+### Test Configuration
+```bash
+# Test storage configuration
+python tests/test_config.py
+```
 
-3. **Dependencies Installed**:
-   ```bash
-   pip install -r requirements.txt
-   ```
+## Test Dependencies
 
-4. **Database Running**:
-   - PostgreSQL database accessible
-   - Environment variables for database connection
+- **Database**: Tests use a test SQLite database
+- **External APIs**: Some tests require API keys (OpenAI, Hugging Face)
+- **Storage**: Tests verify both local and S3 storage configurations
 
-## Test Results
+## Test Data
 
-Tests will show:
-- **SUCCESS** - Test completed successfully
-- **ERROR** - Test encountered errors
-- **TIMEOUT** - Test took too long
-- **WARNING** - Unexpected error occurred
+- **`test.jpg`** - Sample image file for testing uploads and image processing
 
 ## Troubleshooting
 
-### Common Issues
-1. **Import Errors**: Tests automatically add the parent directory to Python path
-2. **API Key Issues**: Ensure your API keys are valid and have proper permissions
-3. **Database Connection**: Check database credentials and connection
-4. **Timeout Issues**: Some tests may take longer on first run due to model loading
-
-### Debug Mode
-
-For detailed debugging, run individual tests directly:
-```bash
-python tests/test_core.py
-```
-
-### Logs
-
-Check the console output for detailed error messages and debugging information.
+See `TROUBLESHOOTING_HF.md` for common Hugging Face deployment issues.
 
 ## Adding New Tests
 
-When adding new tests:
-
-1. **Place in tests folder** with descriptive filename
-2. **Update run_tests.py** to include the new test
-3. **Fix imports** if importing from app modules
-4. **Add to README** with description
-
-## Related Documentation
-
-- **`TROUBLESHOOTING_HF.md`** - Hugging Face API troubleshooting guide
-- **`HUGGINGFACE_INTEGRATION.md`** - Hugging Face integration documentation 
+1. Create test file in this directory
+2. Follow naming convention: `test_*.py`
+3. Add to `run_tests.py` in appropriate category
+4. Ensure tests can run independently
+5. Include proper error handling and cleanup 

py_backend/tests/run_tests.py

@@ -48,9 +48,19 @@ def main():
     print(f"Python: {sys.executable}")
     print(f"Working Directory: {os.getcwd()}")
     
+    # Organized by category
     tests = [
+        # Core functionality tests
         ("test_core.py", "Core Application Tests"),
-        ("test_hf.py", "Hugging Face Integration Tests")
+        ("test_config.py", "Configuration and Storage Tests"),
+        
+        # API and integration tests
+        ("test_upload_flow.py", "Upload Flow Tests"),
+        ("test_openai_integration.py", "OpenAI Integration Tests"),
+        ("test_hf.py", "Hugging Face Integration Tests"),
+        
+        # Frontend and UI tests
+        ("test_explore_page.py", "Explore Page Tests"),
     ]
     
     passed = 0

py_backend/tests/test_config.py

@@ -0,0 +1,39 @@
+#!/usr/bin/env python3
+"""Test script to verify storage configuration"""
+
+import os
+import sys
+sys.path.append('..')
+
+from app.config import settings
+from app.storage import get_object_url
+
+def test_config():
+    print("=== Storage Configuration Test ===")
+    print(f"STORAGE_PROVIDER: {settings.STORAGE_PROVIDER}")
+    print(f"S3_ENDPOINT: {settings.S3_ENDPOINT}")
+    print(f"S3_BUCKET: {settings.S3_BUCKET}")
+    print(f"S3_PUBLIC_URL_BASE: {settings.S3_PUBLIC_URL_BASE}")
+    
+    if settings.STORAGE_PROVIDER == "s3":
+        print("\n=== S3 Storage Test ===")
+        try:
+            # Test URL generation
+            test_key = "maps/test_image.jpg"
+            url = get_object_url(test_key)
+            print(f"Generated URL for '{test_key}': {url}")
+            
+            if settings.S3_PUBLIC_URL_BASE:
+                print(f"✅ Using public URL base: {settings.S3_PUBLIC_URL_BASE}")
+            else:
+                print("⚠️  No S3_PUBLIC_URL_BASE set - will use presigned URLs")
+                
+        except Exception as e:
+            print(f"❌ Error testing S3 storage: {e}")
+    else:
+        print(f"\n=== Local Storage Test ===")
+        print(f"Storage directory: {settings.STORAGE_DIR}")
+        print("✅ Local storage configured")
+
+if __name__ == "__main__":
+    test_config()