diff --git a/app/__pycache__/main.cpython-311.pyc b/app/__pycache__/main.cpython-311.pyc
index 0746e7c6..43e297c2 100644
Binary files a/app/__pycache__/main.cpython-311.pyc and b/app/__pycache__/main.cpython-311.pyc differ
diff --git a/app/routes/__pycache__/helm.cpython-311.pyc b/app/routes/__pycache__/helm.cpython-311.pyc
index cf0b190e..a3533261 100644
Binary files a/app/routes/__pycache__/helm.cpython-311.pyc and b/app/routes/__pycache__/helm.cpython-311.pyc differ
diff --git a/app/routes/__pycache__/terraform.cpython-311.pyc b/app/routes/__pycache__/terraform.cpython-311.pyc
index a8eff2c1..de38bf4c 100644
Binary files a/app/routes/__pycache__/terraform.cpython-311.pyc and b/app/routes/__pycache__/terraform.cpython-311.pyc differ
diff --git a/app/routes/helm.py b/app/routes/helm.py
index 8450e5bd..7470024c 100644
--- a/app/routes/helm.py
+++ b/app/routes/helm.py
@@ -3,10 +3,11 @@
 from app.services import (write_installation,edit_directory_generator,execute_pythonfile)
 from app.models import (HelmTemplateGeneration,Output)
 from app.prompt_generators import (helm_template_generator)
-
+import os
 @app.post("/Helm-template/")
 async def Helm_template_generation(request:HelmTemplateGeneration) -> Output:
-
+        if os.environ.get("TEST"):
+            return Output(output='output')
         generated_prompt = helm_template_generator(request)
         output = gpt_service(generated_prompt)
         edit_directory_generator("helm_generator",output)
diff --git a/app/routes/terraform.py b/app/routes/terraform.py
index f3db5ae8..72ca81b5 100644
--- a/app/routes/terraform.py
+++ b/app/routes/terraform.py
@@ -20,18 +20,20 @@
         IaC_installation_generator, 
        )
 from app.template_generators.terraform.docker import (IaC_template_generator_docker)
-
+import os
 
 @app.post("/IaC-basic/")
 async def IaC_basic_generation(request:IaCBasicInput) -> Output:
-
+        if os.environ.get("TEST"):
+            return Output(output='Terraform developed by hashicorp and it is very usefull')
         generated_prompt = IaC_basics_generator(request)
         output = gpt_service(generated_prompt)
         return Output(output=output)
    
 @app.post("/IaC-bugfix/")
 async def IaC_bugfix_generation(request:IaCBugfixInput) -> Output:
-
+        if os.environ.get("TEST"):
+            return Output(output='fix this bug by adding x to the y')
         generated_prompt = IaC_bugfix_generator(request)
         output = gpt_service(generated_prompt)
         return Output(output=output)
@@ -39,14 +41,16 @@ async def IaC_bugfix_generation(request:IaCBugfixInput) -> Output:
 
 @app.post("/IaC-install/")
 async def IaC_install_generation(request:IaCInstallationInput) -> Output:
-
+        if os.environ.get("TEST"):
+            return Output(output='apt-get install xyz \n apt-get update (covert them to shell file output)')
         generated_prompt = IaC_installation_generator(request)
         output = gpt_service(generated_prompt)
         return Output(output=output)
 
 @app.post("/IaC-template/docker")
 async def IaC_template_generation_docker(request:IaCTemplateGenerationDocker) -> Output:
-
+        if os.environ.get("TEST"):
+            return Output(output='output (nothing special)')
         generated_prompt = IaC_template_generator_docker(request)
         output = gpt_service(generated_prompt)
         edit_directory_generator("terraform_generator",output)
@@ -55,7 +59,8 @@ async def IaC_template_generation_docker(request:IaCTemplateGenerationDocker) ->
 
 @app.post("/IaC-template/aws/ec2")
 async def IaC_template_generation_aws_ec2(request:IaCTemplateGenerationEC2) -> Output:
-
+        if os.environ.get("TEST"):
+            return Output(output='output (nothing special)')
         generated_prompt = IaC_template_generator_docker(request)
         output = gpt_service(generated_prompt)
         edit_directory_generator("terraform_generator",output)
@@ -64,7 +69,8 @@ async def IaC_template_generation_aws_ec2(request:IaCTemplateGenerationEC2) -> O
 
 @app.post("/IaC-template/aws/s3")
 async def IaC_template_generation_aws_s3(request:IaCTemplateGenerationS3) -> Output:
-
+        if os.environ.get("TEST"):
+            return Output(output='output (nothing special)')
         generated_prompt = IaC_template_generator_docker(request)
         output = gpt_service(generated_prompt)
         edit_directory_generator("terraform_generator",output)
@@ -73,7 +79,8 @@ async def IaC_template_generation_aws_s3(request:IaCTemplateGenerationS3) -> Out
 
 @app.post("/IaC-template/aws/iam")
 async def IaC_template_generation_aws_iam(request:IaCTemplateGenerationIAM) -> Output:
-
+        if os.environ.get("TEST"):
+            return Output(output='output (nothing special)')
         generated_prompt = IaC_template_generator_docker(request)
         output = gpt_service(generated_prompt)
         edit_directory_generator("terraform_generator",output)
diff --git a/docker-compose.yml b/docker-compose.yml
index e67a544f..9a944a15 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -17,6 +17,7 @@ services:
     environment:
       <<: *common-variables
       OPENAI_API_KEY: ${KEY:-}
+      TEST: ${TEST:-}
       MONGO_HOST: mongo_gpt
       MONGO_PORT: 27017