Refactor Dockerfile for improved caching and add health check endpoint

Dockerfile

@@ -12,21 +12,22 @@ RUN mkdir -p cache/huggingface vector_store chat_history \
     && chown -R 1000:1000 . \
     && chmod -R 755 .
 
-# Copy dependencies separately for caching
+# Copy requirements first for better caching
 COPY requirements.txt .
-
-# Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy source code
-COPY . .
+# Copy only necessary files
+COPY app.py .
+COPY .env .
+COPY index.html .
 
 # Set environment variables
 ENV HF_HOME=/app/cache/huggingface
 ENV HUGGINGFACE_HUB_CACHE=/app/cache/huggingface
 ENV XDG_CACHE_HOME=/app/cache
+ENV PORT=8000
 
-# Set permissions (only for newly created files)
+# Set permissions
 RUN chown -R 1000:1000 /app \
     && find /app -type d -exec chmod 755 {} \; \
     && find /app -type f -exec chmod 644 {} \;
@@ -34,6 +35,11 @@ RUN chown -R 1000:1000 /app \
 # Run as non-privileged user
 USER 1000
 
+# Health check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:8000/health || exit 1
+
 EXPOSE 8000
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]
+# Use a startup script
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000", "--timeout-keep-alive", "120"]

app.py

@@ -314,6 +314,11 @@ async def root():
         "kb_info": kb_info if vector_store_exists else None
     }
 
+@app.get("/health")
+async def health_check():
+    """Health check endpoint"""
+    return {"status": "healthy"}
+
 @app.post("/build-kb", response_model=BuildKnowledgeBaseResponse)
 async def build_kb_endpoint():
     """Endpoint to build/rebuild the knowledge base"""

data_snapshot/chat_history/chat_history_2025-03-14.json

@@ -0,0 +1,14 @@
+[
+  {
+    "timestamp": "2025-03-14 09:27:01",
+    "question": "where are you located?",
+    "answer": "Hello, I'm happy to help. Status Law is an international law firm specializing in extradition cases, with a presence in the European Union. Our team of multilingual lawyers has on-the-ground experience in key regions, providing us with a network of international connections that supports our operations outside the EU. We have expertise in various practice areas, including extradition defense, international prosecution protection, and Interpol red notice removal. If you have any specific questions or need more information, please don't hesitate to contact us via phone at +32465594521 or through our contact form: https://status.law/law-firm-contact-legal-protection/.",
+    "context": "Minimum 5 working hours, transportation and accommodation apply for specialist deployments\n\n\n\n\nLocal attorneys' fees are charged separately\n\n\n\n\nDocument translation and legalization costs are not included\nPractice Areas - Status Law     HomeServicesExtradition defenseInternational prosecution protectionInterpol red notice removalCross-border banking legal issuesReputation protectionChallenging sanctionsAbout UsRegions of operationsCareersFAQContactOur PricingEnglishEnglishРусскийEspañolالعربية中文 (中国)\nInterpol red notice removal | Status.Law          HomeServicesExtradition defenseInternational prosecution protectionInterpol red notice removalCross-border banking legal issuesReputation protectionChallenging sanctionsAbout UsRegions of operationsCareersFAQContactOur PricingEnglishEnglishРусскийEspañolالعربية中文 (中国)\nof each of the distinct areas our clients are coming from. Each of the practices is supervised by one of our attorneys who has resided and practiced law in the respective region before relocating to the EU. In addition to the in-depth knowledge of the peculiarities of local law, processes and institutions this provides us with a network of international connections that supports our operations outside the European Union.Our team of multilingual lawyers with on-the-ground experience in key"
+  },
+  {
+    "timestamp": "2025-03-14 10:47:51",
+    "question": "How much does a 1 hour consultation cost?",
+    "answer": "Our initial consultation is designed to help clients resolve their situation efficiently, and it often makes further cooperation unnecessary. The cost of the consultation reflects its potential to provide a comprehensive solution. For specific pricing details, please visit our Services and Pricing page: https://status.law/tariffs-for-services-against-extradition-en. If you have any further questions or would like to book a consultation, feel free to contact us via phone at +32465594521 or use our contact form: https://status.law/law-firm-contact-legal-protection/.",
+    "context": "It often happens that an initial consultation helps clients to resolve their situation and makes further cooperation unnecessary, which explains its relatively high cost.Are there any services you can provide on a pro bono basis?Our lawyers consult a limited number of qualified government and NGO bodies on matters of human rights violations and the imposition of sanctions. Please contact us for further details.I cannot visit you in person or call you due to privacy concerns. Can I get your\nMinimum 5 working hours, transportation and accommodation apply for specialist deployments\n\n\n\n\nLocal attorneys' fees are charged separately\n\n\n\n\nDocument translation and legalization costs are not included\nDocument translation and legalization costs are not included\n\n\n\n\nSuccess fees apply only for positive outcomesAll information on the “Status Law” Law Firm’s website is for informational purposes only, does not constitute legal advice or recommendations, and does not establish an attorney-client relationship between the user and the firm.You can book a consultation time using the quick form below or by going to our contact page here.\nlegal advice.MAKE A REQUESTPartners and EmployeesOur lawyers are highly skilled and accessible, get personal attention to every matterPawel Leszek KazaneckiSpecialist in democratic development and human rights."
+  }
+]

space.yaml

@@ -6,4 +6,5 @@ colorTo: indigo
 sdk_version: 0.0.1
 app_port: 8000
 pinned: false
-license: mit
+license: mit
+startup_duration: 900