fix-issue

Dockerfile

@@ -1,34 +1,24 @@
 # Use official Python runtime as base image
-FROM python:3.10-slim
+FROM slabstech/dhwani-api-server-base
 
 WORKDIR /app
 
-# Set environment variables
-ENV PYTHONDONTWRITEBYTECODE=1
-ENV PYTHONUNBUFFERED=1
-
-# Install system dependencies
-RUN apt-get update && apt-get install -y \
-    gcc \
-    && rm -rf /var/lib/apt/lists/*
-
-# Install Python dependencies
-COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
-
 # Copy application code
 COPY . .
 
+# Create appuser and set permissions for /app and /data
 RUN useradd -ms /bin/bash appuser \
-    && chown -R appuser:appuser /app
+    && mkdir -p /data \
+    && chown -R appuser:appuser /app /data
 
 USER appuser
-# Expose port from settings (7860 from your code)
+
+# Expose port from settings
 EXPOSE 7860
 
 # Healthcheck
 HEALTHCHECK --interval=30s --timeout=3s \
   CMD curl -f http://localhost:7860/v1/health || exit 1
 
-# Command to run the application with configurable host/port
+# Command to run the application
 CMD ["python", "/app/src/server/main.py", "--host", "0.0.0.0", "--port", "7860"]

Dockerfile.app

@@ -0,0 +1,24 @@
+# Use official Python runtime as base image
+FROM slabstech/dhwani-api-server-base
+
+WORKDIR /app
+
+# Copy application code
+COPY . .
+
+# Create appuser and set permissions for /app and /data
+RUN useradd -ms /bin/bash appuser \
+    && mkdir -p /data \
+    && chown -R appuser:appuser /app /data
+
+USER appuser
+
+# Expose port from settings
+EXPOSE 7860
+
+# Healthcheck
+HEALTHCHECK --interval=30s --timeout=3s \
+  CMD curl -f http://localhost:7860/v1/health || exit 1
+
+# Command to run the application
+CMD ["python", "/app/src/server/main.py", "--host", "0.0.0.0", "--port", "7860"]

Dockerfile.base

@@ -0,0 +1,18 @@
+# Use official Python runtime as base image
+FROM python:3.10-slim
+
+WORKDIR /app
+
+# Set environment variables
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    gcc \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install Python dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt

src/server/main.py

@@ -51,8 +51,16 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Rate limiting based on user_id
-limiter = Limiter(key_func=lambda request: get_current_user(request.scope.get("route").dependencies))
+# Rate limiting based on user_id with fallback to IP
+async def get_user_id_for_rate_limit(request: Request):
+    try:
+        credentials = bearer_scheme(request)
+        user_id = await get_current_user(credentials)
+        return user_id
+    except Exception:
+        return get_remote_address(request)  # Fallback to IP if unauthenticated
+
+limiter = Limiter(key_func=get_user_id_for_rate_limit)
 
 # Request/Response Models
 class SpeechRequest(BaseModel):
@@ -398,11 +406,13 @@ async def transcribe_audio(
         response.raise_for_status()
         
         transcription = response.json().get("text", "")
+        logger.info(f"Transcription completed in {time() - start_time:.2f} seconds")
         return TranscriptionResponse(text=transcription)
     
     except requests.Timeout:
         raise HTTPException(status_code=504, detail="Transcription service timeout")
     except requests.RequestException as e:
+        logger.error(f"Transcription request failed: {str(e)}")
         raise HTTPException(status_code=500, detail=f"Transcription failed: {str(e)}")
 
 @app.post("/v1/chat_v2", 

src/server/utils/auth.py

@@ -9,9 +9,11 @@ from sqlalchemy import create_engine, Column, String, Boolean
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import sessionmaker
 from passlib.context import CryptContext
+import os
 
-# SQLite database setup
-DATABASE_URL = "sqlite:///users.db"
+# SQLite database setup with Hugging Face persistent storage
+DATABASE_PATH = "/data/users.db"
+DATABASE_URL = f"sqlite:///{DATABASE_PATH}"
 engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False})
 Base = declarative_base()
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
@@ -22,6 +24,10 @@ class User(Base):
     password = Column(String)  # Stores hashed passwords
     is_admin = Column(Boolean, default=False)  # New admin flag
 
+# Ensure the /data directory exists
+os.makedirs(os.path.dirname(DATABASE_PATH), exist_ok=True)
+
+# Create database tables
 Base.metadata.create_all(bind=engine)
 
 # Password hashing
@@ -43,6 +49,7 @@ class Settings(BaseSettings):
     external_audio_proc_url: str = Field(..., env="EXTERNAL_AUDIO_PROC_URL")
     default_admin_username: str = Field("admin", env="DEFAULT_ADMIN_USERNAME")
     default_admin_password: str = Field("admin54321", env="DEFAULT_ADMIN_PASSWORD")
+    database_path: str = DATABASE_PATH  # Add for reference if needed elsewhere
 
     class Config:
         env_file = ".env"
@@ -50,24 +57,30 @@ class Settings(BaseSettings):
 
 settings = Settings()
 
-# Seed initial data (optional)
+# Seed initial data
 def seed_initial_data():
     db = SessionLocal()
-    # Seed test user (non-admin)
-    if not db.query(User).filter_by(username="testuser").first():
-        hashed_password = pwd_context.hash("password123")
-        db.add(User(username="testuser", password=hashed_password, is_admin=False))
-        db.commit()
-    # Seed admin user using environment variables
-    admin_username = settings.default_admin_username
-    admin_password = settings.default_admin_password
-    if not db.query(User).filter_by(username=admin_username).first():
-        hashed_password = pwd_context.hash(admin_password)
-        db.add(User(username=admin_username, password=hashed_password, is_admin=True))
-        db.commit()
-    db.close()
-    logger.info(f"Seeded initial data: admin user '{admin_username}'")
+    try:
+        # Seed test user (non-admin)
+        if not db.query(User).filter_by(username="testuser").first():
+            hashed_password = pwd_context.hash("password123")
+            db.add(User(username="testuser", password=hashed_password, is_admin=False))
+            db.commit()
+        # Seed admin user using environment variables
+        admin_username = settings.default_admin_username
+        admin_password = settings.default_admin_password
+        if not db.query(User).filter_by(username=admin_username).first():
+            hashed_password = pwd_context.hash(admin_password)
+            db.add(User(username=admin_username, password=hashed_password, is_admin=True))
+            db.commit()
+        logger.info(f"Seeded initial data: admin user '{admin_username}'")
+    except Exception as e:
+        logger.error(f"Error seeding initial data: {str(e)}")
+        db.rollback()
+    finally:
+        db.close()
 
+# Initialize database with seed data
 seed_initial_data()
 
 # Use HTTPBearer