added gemini too

Dockerfile

@@ -4,19 +4,21 @@ FROM python:3.11-slim
 # Set the working directory inside the container
 WORKDIR /app
 
-# Create the generated_charts directory and set permissions
-RUN mkdir -p /app/generated_charts && chmod -R 777 /app/generated_charts
-
-# Create the cache directory and set permissions
-RUN mkdir -p /app/cache && chmod -R 777 /app/cache
-
-# Create the log file and set permissions
-RUN touch /app/pandasai.log && chmod 666 /app/pandasai.log
-
-# Create the log file and set permissions
-RUN touch /app/api_key_rotation.log && chmod 666 /app/api_key_rotation.log
-
-# Set the Matplotlib cache directory to /app/cache
+# Create all required directories with proper permissions upfront
+RUN mkdir -p /app/generated_outputs && \
+    mkdir -p /app/generated_charts && \
+    mkdir -p /app/cache && \
+    chmod -R 777 /app/generated_outputs && \
+    chmod -R 777 /app/generated_charts && \
+    chmod -R 777 /app/cache
+
+# Create log files with proper permissions
+RUN touch /app/pandasai.log && \
+    touch /app/api_key_rotation.log && \
+    chmod 666 /app/pandasai.log && \
+    chmod 666 /app/api_key_rotation.log
+
+# Set environment variables
 ENV MPLCONFIGDIR=/app/cache
 
 # Copy the requirements file first
@@ -28,6 +30,10 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy the rest of the application code
 COPY . .
 
+# Ensure the user has write permissions to all directories
+RUN chown -R 1000:1000 /app && \
+    chmod -R 777 /app
+
 # Expose port 7860 (required by Hugging Face Spaces)
 EXPOSE 7860