From 5149a1c9e7e20059ca98117a8226655e504520fd Mon Sep 17 00:00:00 2001 From: knight1001d Date: Sat, 29 Jun 2024 00:55:11 +0530 Subject: [PATCH] complete docSearch and image generation action --- .../actions/scripts/docSearch.py | 3 +-- .../actions/scripts/generateImageFromText.py | 4 ++-- .../db/__pycache__/index.cpython-310.pyc | Bin 1028 -> 1028 bytes fastapi-rest-apis/db/index.py | 2 ++ fastapi-rest-apis/requirements.txt | Bin 2450 -> 3076 bytes 5 files changed, 5 insertions(+), 4 deletions(-) diff --git a/fastapi-rest-apis/actions/scripts/docSearch.py b/fastapi-rest-apis/actions/scripts/docSearch.py index 42e8479..66d955b 100644 --- a/fastapi-rest-apis/actions/scripts/docSearch.py +++ b/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) diff --git a/fastapi-rest-apis/actions/scripts/generateImageFromText.py b/fastapi-rest-apis/actions/scripts/generateImageFromText.py index e8429b6..4157198 100644 --- a/fastapi-rest-apis/actions/scripts/generateImageFromText.py +++ b/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}})) diff --git a/fastapi-rest-apis/db/__pycache__/index.cpython-310.pyc b/fastapi-rest-apis/db/__pycache__/index.cpython-310.pyc index 86e2febbc6fca5995c2886c6119f01c0fdb21048..09dff24f47d138f8af6351e153042e4e71347560 100644 GIT binary patch delta 48 zcmZqSXyM?^=jG*M0D{-N^=TV7^7b$?a!sDa_>fU(at+gMMybgW%uCr>xY#(DcsKwn CV+)c1 delta 48 zcmZqSXyM?^=jG*M0D=JC`n0thd3zWcIVR6ye8?y;xrXUBqr~J0=B4aRTx=XnJRATO CHVW$i diff --git a/fastapi-rest-apis/db/index.py b/fastapi-rest-apis/db/index.py index e38b703..8eefb0a 100644 --- a/fastapi-rest-apis/db/index.py +++ b/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) diff --git a/fastapi-rest-apis/requirements.txt b/fastapi-rest-apis/requirements.txt index acafcd2cfc8486c7cccfea383666bf684c70f126..4d9e17f25e29ebc2eda8b0834cc650810c5aec3c 100644 GIT binary patch delta 664 zcmY*WO-{m47#tg1NaD(nxN*V4C4?3fjN;Ph!liMIRipvhhJp!h*mwc#0XPIFaqTI* zfI9P)@{^Y~?VEY?&CL5e`Z%8b{dN!_LYMCWUeM;Mql%V2hbUqfPmJNALEPn2!-mtt zm@8guTo6Co!eeTU(KEb8>>Sk!Mcge)LL09LNgNnWBV;Gz%$@6*{+jEZ4iZes_fesA zfFUD9nDARcWKx6bs|m6<;t64%>3YUcgeOi6sMM#z$QYWf8azAg)02zoFJ5dW5b&M2cHvb AZ~y=R delta 54 zcmZpXm?X^f|KG$lcAO>*dJM)uY%n>NQFrn(#u<}SnBGi2!YsCV1+x>=WD&N$$&c7v KHb=4Nu>b(_NfI9b