feat: Connect to external database

README.md

@@ -44,6 +44,14 @@ docker build . -f Dockerfile -t chainfury:latest
 docker run --env OPENAI_API_KEY=<your_key_here> -p 8000:8000 chainfury:latest
 ```
 
+You can also pass a Database URL to the docker container using the `DATABASE_URL` environment variable. If you do not pass a database URL, ChainFury will use a SQLite database.
+
+Example:
+
+```bash
+docker run -it -E DATABASE_URL="mysql+pymysql://<user>:<password>@127.0.0.1:3306/<database>" -p 8000:8000 chainfury
+```
+
 ### **Method 2: Manual**
 
 For this, you will need to build the frontend and and then run the backend. The frontend can be built using the following command:

requirements.txt

@@ -1,9 +1,10 @@
 fastapi
-langchain==0.0.131
+langchain==0.0.141
 openai
 sqlalchemy
 pyjwt[crypto]
 jinja2
 langflow==0.0.54
 black
 passlib
+pymysql

scripts/build_and_copy.sh

@@ -0,0 +1,30 @@
+#!/bin/bash
+
+# This script builds the project and copies the resulting ./dist folder to the server
+
+# check if working in root directory, both client and server folders should be present
+if [ ! -d "client" ] || [ ! -d "server" ]; then
+    echo "Please run this script from the root directory of the Repo"
+    exit 1
+fi
+
+# build the project
+# Go into the client folder and build the project using yarn
+cd client
+yarn build
+
+# Go back to the root directory
+cd ..
+
+# copy the dist folder to the server
+# Go into the server folder, remove the old static folder and copy the new dist folder, copy index.html to templates
+cd server
+rm -rf static
+cp -r ../client/dist static
+cp ../client/dist/index.html templates
+
+# Go back to the root directory
+cd ..
+
+# Done
+echo "Done"

server/commons/config.py

@@ -17,8 +17,13 @@ def get_logger(name):
 
 logger = get_logger(__name__)
 
+
 DATABASE = "sqlite:///./chain.db"
-engine = create_engine(DATABASE, connect_args={"check_same_thread": False})
+if os.environ.get("DATABASE_URL", None) is not None:
+    logger.info("Using DATABASE_URL")
+    DATABASE = os.environ.get("DATABASE_URL")
+
+engine = create_engine(DATABASE)
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 logger.info("Database opened successfully")