complete docSearch and image generation action

fastapi-rest-apis/actions/scripts/docSearch.py

@@ -4,9 +4,8 @@
 def find_occurrences(text, keyword):
     occurrences = text.lower().count(keyword.lower())
     return occurrences
-def get_output():
+def get_output(input_json):
     try:
-        input_json = json.loads(sys.argv[1])
         text = input_json["text"]
         keyword = input_json["keyword"]
         num_occurrences = find_occurrences(text, keyword)

fastapi-rest-apis/actions/scripts/generateImageFromText.py

@@ -1,10 +1,10 @@
-def get_output():
+def get_output(input_json):
     try:
         import requests
         import json
         from craiyon import Craiyon
         generator = Craiyon() # Instantiates the api wrapper
-        result = generator.generate("Photorealistic image of shrek eating earth", negative_prompt="spoon", model_type="art")
+        result = generator.generate(input_json["prompt"], negative_prompt="spoon", model_type="art")
         print(json.dumps({"output": {"type": "text", "content": f"{result.description}"}}))
     except Exception as e:
         print(json.dumps({"output": {"type": "error", "content": e}}))

fastapi-rest-apis/db/index.py

@@ -5,6 +5,8 @@
 
 DB_URL = os.environ['DB_URL'] 
 
+
+
 def createDBSession():
     engine = create_engine(DB_URL)
     Session = sessionmaker(bind=engine)