Fix configuration issues

Update configuration and verification for various files.

* **requirements.txt**: Remove duplicate dependencies and ensure all dependencies are listed only once.
* **Dockerfile**: Update `COPY requirements.txt .` to `COPY code/requirements.txt .` and ensure the correct path for `requirements.txt` is used.
* **scripts/verify_file_structure.py**: Add verification for the presence of `.env` and `config/config.py` files.
* **config/config.py**: Add verification for the presence of `HUGGINGFACE_API_KEY` and `HUGGINGFACE_PROJECT_NAME` environment variables. Add logging for the values of these environment variables.
* **.env**: Ensure `HUGGINGFACE_API_KEY` and `HUGGINGFACE_PROJECT_NAME` are set correctly. Add any missing environment variables required by the application.
* **src/backend/app.py**: Add verification for the presence of `HUGGINGFACE_API_KEY` and `HUGGINGFACE_PROJECT_NAME` environment variables. Add logging for the values of these environment variables.

---

For more details, open the [Copilot Workspace session](https://copilot-workspace.githubnext.com/ProjectZeroDays/zero-click-exploits?shareId=XXXX-XXXX-XXXX-XXXX).

Author: ProjectZeroDays

.env

@@ -1,3 +1,9 @@
 # Placeholder for sensitive information
 HUGGINGFACE_API_KEY=your_huggingface_api_key_here
 HUGGINGFACE_PROJECT_NAME=your_project_name_here
+
+# Add any missing environment variables required by the application
+DATABASE_URL=your_database_url_here
+SECRET_KEY=your_secret_key_here
+API_KEY=your_api_key_here
+API_SECRET=your_api_secret_here

Dockerfile

@@ -6,7 +6,7 @@ ENV HF_API_TOKEN=${HF_API_TOKEN}
 ENV HF_PROJECT_NAME=${HF_PROJECT_NAME}
 
 # Install Python Dependencies
-COPY requirements.txt .
+COPY code/requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy The Source Files
@@ -51,4 +51,4 @@ CMD ["/app/infra/error_handling.sh"]
 RUN test -n "$HF_API_TOKEN" || (echo "HF_API_TOKEN is not set" && exit 1)
 RUN test -n "$HF_PROJECT_NAME" || (echo "HF_PROJECT_NAME is not set" && exit 1)
 RUN test -n "$API_KEY" || (echo "API_KEY is not set" && exit 1)
-RUN test -n "$API_SECRET" || (echo "API_SECRET is not set" && exit 1)
+RUN test -n "$API_SECRET" || (echo "API_SECRET is not set" && exit 1)

config/config.py

@@ -30,6 +30,10 @@ class Config:
     API_KEY = os.environ.get('API_KEY', 'default_api_key')
     API_SECRET = os.environ.get('API_SECRET', 'default_api_secret')
 
+    # Hugging Face API Key and Project Name
+    HUGGINGFACE_API_KEY = os.environ.get('HUGGINGFACE_API_KEY', 'default_huggingface_api_key')
+    HUGGINGFACE_PROJECT_NAME = os.environ.get('HUGGINGFACE_PROJECT_NAME', 'default_huggingface_project_name')
+
     # Verification for MFA and encryption method implementations
     @staticmethod
     def verify_security_implementations():
@@ -61,7 +65,8 @@ def log_missing_env_vars():
         env_vars = [
             'SECRET_KEY', 'DATABASE_URL', 'AI_VULNERABILITY_SCANNING_ENABLED', 'AI_EXPLOIT_MODIFICATIONS_ENABLED',
             'MFA_ENABLED', 'ENCRYPTION_METHOD', 'BLOCKCHAIN_LOGGING_ENABLED', 'BLOCKCHAIN_LOGGING_NODE',
-            'ADVANCED_ENCRYPTION_METHODS', 'SECURITY_AUDITS_ENABLED', 'PENETRATION_TESTING_ENABLED', 'API_KEY', 'API_SECRET'
+            'ADVANCED_ENCRYPTION_METHODS', 'SECURITY_AUDITS_ENABLED', 'PENETRATION_TESTING_ENABLED', 'API_KEY', 'API_SECRET',
+            'HUGGINGFACE_API_KEY', 'HUGGINGFACE_PROJECT_NAME'
         ]
         for var in env_vars:
             if not os.environ.get(var):
@@ -83,3 +88,5 @@ def log_config_values():
         logging.info(f"PENETRATION_TESTING_ENABLED: {Config.PENETRATION_TESTING_ENABLED}")
         logging.info(f"API_KEY: {Config.API_KEY}")
         logging.info(f"API_SECRET: {Config.API_SECRET}")
+        logging.info(f"HUGGINGFACE_API_KEY: {Config.HUGGINGFACE_API_KEY}")
+        logging.info(f"HUGGINGFACE_PROJECT_NAME: {Config.HUGGINGFACE_PROJECT_NAME}")

requirements.txt

@@ -11,52 +11,7 @@ torch==2.2.0
 uvicorn==0.18.2
 fastapi==0.92.0
 gunicorn==22.0.0
-python-dotenv==0.21.1 
-transformers 
-numpy 
-Pillow 
-tqdm  
-scipy  
-huggingface_hub  
-torchaudio   
-pydub     
-ffmpeg-python    
-pytorch_lightning   
-einops     
-sentencepiece    
-transformers[flax]   
-safetensors      
-bitsandbytes     
-faiss-cpu        
-nlp         
-tokenizers      
-webdataset       
-gradio          
-omegaconf       
-dataclasses     
-scikit-learn    
-timm            
-dill            
-setproctitle    
-typing-extensions 
-redis           
-flask           
-psutil          
-matplotlib      
-seaborn         
-beautifulsoup4  
-numpydoc        
-streamlit       
-plotly          
-agent-zero
-flask_sqlalchemy
-flask_migrate
-aiohttp
-validators
-yaml
-pandas
-scikit-image
-web3
+python-dotenv==0.21.1
 transformers==4.38.0
 numpy==1.24.2
 Pillow==10.3.0
@@ -105,62 +60,3 @@ web3==5.31.3
 pip-tools==6.4.0
 tkinter
 os
-subprocess
-re
-shodan
-python-nmap
-logging
-json
-requests
-cryptography
-panel
-torch
-uvicorn
-fastapi
-gunicorn
-python-dotenv
-transformers
-numpy
-Pillow
-tqdm
-scipy
-huggingface_hub
-torchaudio
-pydub
-ffmpeg-python
-pytorch_lightning
-einops
-sentencepiece
-transformers[flax]
-safetensors
-bitsandbytes
-faiss-cpu
-nlp
-tokenizers
-webdataset
-gradio
-omegaconf
-dataclasses
-scikit-learn
-timm
-dill
-setproctitle
-typing-extensions
-redis
-flask
-psutil
-matplotlib
-seaborn
-beautifulsoup4
-numpydoc
-streamlit
-plotly
-tkinter
-agent-zero
-flask_sqlalchemy
-flask_migrate
-aiohttp
-validators
-yaml
-pandas
-scikit-image

scripts/verify_file_structure.py

@@ -222,7 +222,21 @@ def verify_config_usage():
         else:
             print(f"Method {method} in {config_file} is called in other parts of the code.")
 
+def verify_env_file():
+    if not os.path.isfile(".env"):
+        print("File missing: .env")
+    else:
+        print("File .env is present.")
+
+def verify_config_file():
+    if not os.path.isfile("config/config.py"):
+        print("File missing: config/config.py")
+    else:
+        print("File config/config.py is present.")
+
 if __name__ == "__main__":
     verify_correct_placement()
     verify_method_calls()
     verify_config_usage()
+    verify_env_file()
+    verify_config_file()

src/backend/app.py

@@ -35,6 +35,17 @@
 db = SQLAlchemy(app)
 migrate = Migrate(app, db)
 
+# Verify presence of HUGGINGFACE_API_KEY and HUGGINGFACE_PROJECT_NAME environment variables
+if not os.environ.get('HUGGINGFACE_API_KEY'):
+    logger.error("HUGGINGFACE_API_KEY environment variable is not set.")
+else:
+    logger.info(f"HUGGINGFACE_API_KEY: {os.environ.get('HUGGINGFACE_API_KEY')}")
+
+if not os.environ.get('HUGGINGFACE_PROJECT_NAME'):
+    logger.error("HUGGINGFACE_PROJECT_NAME environment variable is not set.")
+else:
+    logger.info(f"HUGGINGFACE_PROJECT_NAME: {os.environ.get('HUGGINGFACE_PROJECT_NAME')}")
+
 # --- Data Models ---
 class TrojanServer(db.Model):
     id = db.Column(db.Integer, primary_key=True)