Added Missing directories

bardcode_interpreter.py

@@ -13,18 +13,20 @@ Date : 21-05-2023
 
 # Import the required libraries
 import streamlit as st
-from os import path
 import time
 import traceback
 from lib.bardcoder_lib import BardCoder
 import subprocess
 from io import StringIO
-import re
 from lib.sharegpt_api import sharegpt_get_url
 from lib.blacklist_commands import harmful_commands_python, harmful_commands_cpp, harmful_prompts
 from PIL import Image
 import tokenize
+from stat import S_IREAD, S_IRGRP, S_IROTH
+import re
 import io
+import os
+from os import path
 
 # The input limit of Bard is 4,000 character (As per the Bard API documentation)
 # But you can give more input upto 10,000 characters. so we are gonna stick to that.
@@ -389,24 +391,37 @@ def init_session_state():
     if "file_char_count" not in st.session_state:
         st.session_state.file_char_count = 0
 
-import os
-from stat import S_IREAD, S_IRGRP, S_IROTH
-
-def make_code_interpreter_read_only():
-    filename = __file__
-    BardCoder.write_log(f"Making {filename} read-only")
-    os.chmod(filename, S_IREAD|S_IRGRP|S_IROTH)
+def make_code_interpreter_read_only(files=[],folders:str="lib"):
+    for filename in files:
+        BardCoder.write_log(f"Making {filename} read-only")
+        os.chmod(filename, S_IREAD|S_IRGRP|S_IROTH)
 
     # Make all files in lib folder read-only
-    folder = "lib"
+    BardCoder.write_log(f"Making all files in {folders} folder read-only")
+    folder = folders
     for filename in os.listdir(folder):
         filepath = os.path.join(folder, filename)
         os.chmod(filepath, S_IREAD|S_IRGRP|S_IROTH)
 
+def create_dirs_on_startup():
+    # Create the uploads directory if it does not exist
+    if not os.path.exists("uploads"):
+        os.makedirs("uploads")
+    if not os.path.exists("codes"):
+        os.makedirs("codes")
+    if not os.path.exists("response"):
+        os.makedirs("response")
+
 if __name__ == "__main__":
     try:
         BardCoder.write_log("Starting the streamlit App")
-        make_code_interpreter_read_only()
+        # Create directories on startup
+        create_dirs_on_startup()
+        
+        # Make the code interpreter read-only
+        file = __file__
+        filenames = [file,file.replace("bardcode_interpreter", "bardcoder")]
+        make_code_interpreter_read_only(filenames,"lib")
         
         # Load the CSS file named style.css
         load_css("styles/style.css")

bash_src/clear_cache.sh

@@ -1,10 +1,5 @@
 #!/bin/bash
 
-# Create codes folder if it doesn't exist
-if [ ! -d "codes" ]; then
-  mkdir codes
-fi
-
 # removing only all executables from codes.
 file -F: codes/* | grep "executable" | cut -d: -f1 | xargs rm
 

lib/bardcoder_lib.py

@@ -78,7 +78,7 @@ class BardCoder:
             raise ValueError("Data cannot be None")
         with open("bardcoder.log",'a') as f:
             # Get the name of the calling function
-            caller_name = inspect.stack()[1][3]            
+            caller_name = inspect.stack()[1][3]
             f.write(f"{time.strftime('%d-%m-%Y %H:%M:%S')} {caller_name}: {data}\n")
 
     # Set the api key