add logic

app.py

@@ -1,8 +1,27 @@
 from fastapi import FastAPI
+from pydantic import BaseModel
+from typing import List
+from src.processor import *
+from sentence_transformers import SentenceTransformer
 
 app = FastAPI()
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+class Input(BaseModel):
+    text1 : List
+    text2 : List
 
+class Output(BaseModel):
+    matrix : List
+
+@app.post("/process", response_model=Output)
+def process(payload: Input):
+    saoke_spec = text_to_saoke(payload.text1)
+    saoke_patent = text_to_saoke(payload.text2)
+
+    model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
+
+    embeddings1, embeddings2 = text_to_embeddings(saoke_spec, saoke_patent, model)
+    
+    matrix = embeddings_to_matrix(embeddings1, embeddings2)
+    print({"matrix": matrix})
+    return {"matrix": matrix}

requirements.txt

@@ -1,3 +1,6 @@
 fastapi
 uvicorn[standard]
-
+google.genai
+sentence_transformers
+pydantic
+python-dotenv

src/processor.py

@@ -0,0 +1,82 @@
+from google.genai import Client, types
+import os
+import numpy as np
+from sentence_transformers import SentenceTransformer, util
+from dotenv import load_dotenv
+
+load_dotenv()
+
+def text_to_saoke(data):
+    saoke_list = []
+    for d in data:
+        prompt = get_prompt()
+        prompt += d
+        result = send_request(prompt)
+        saoke_list.append(result)
+    return saoke_list
+
+def text_to_embeddings(list1, list2, model):
+    embeddings1 = []
+    for l1 in list1:
+        embeddings1.append(model.encode(l1))
+    embeddings2 = []
+    for l2 in list2:
+        embeddings2.append(model.encode(l2))
+    
+    return embeddings1, embeddings2
+
+def embeddings_to_matrix(embeddings1, embeddings2):
+    matrix = []
+    for e1 in embeddings1:
+        for e2 in embeddings2:
+            cosim = util.cos_sim(e1, e2)
+            matrix.append(cosim)
+    matrix = np.array(matrix).reshape(len(embeddings1), len(embeddings2))
+    return matrix.tolist()
+
+def send_request(prompt):
+    client = Client(api_key=os.getenv("GEMINI_API_KEY"))
+
+
+    response = client.models.generate_content(
+        model="gemini-2.5-flash",
+        contents=prompt,
+    )
+
+    return response.text
+
+def get_prompt():
+    prompt = """
+You are well educated in S-A-O-K-E decomposition methodology applied to patents:
+**Subject (S):** The entity that performs the action (e.g., device, user, system).
+**Action (A):** This represents the specific intervention, process, or method that the invention performs. It describes what the invention *does* to or with specific objects or systems  (e.g., transmits, applies, mixes).
+**Object (O):** The entity or target that the action is performed upon (e.g., signal, data, mixture).
+**Knowledge (K):** This is the body of technical and scientific information that underpins the invention. It is the knowledge that is necessary to design, implement, and operate the action successfully.
+**Effect (E):** This refers to the outcome, result, or consequence of the action. It describes the benefit, improvement, or new capability that the invention provides.
+the entire invention can be mapped as a linked set of S-A–O-K–E units. For example:
+Step 1: (S₁, A₁, O₁, K₁) → E₁
+Step 2: (S₂,A₂, O₂, K₂=E₁+...) → E₂
+...and so on.
+ 
+You mission is to help the user write and analyse their ideas, concept, inventions, problems etc... in the form of S-A-O-K-E.
+You must output a JSON object containings all of the SAOKE:
+
+{
+    "subject": "Mobile payment app",
+    "action": "interlaces affine and non-linear lookup tables",
+    "object": "AES state bytes",
+    "knowledge": "space-hard SPNbox design using table incompressibility theory",
+    "effect": "prevents code-lifting and key-extraction even under full memory disclosure."
+},
+{
+    "subject": "DRM client",
+    "action": "randomizes table encodings on every re-installation",
+    "object": "symmetric key material",
+    "knowledge": "PUF-bound whitening keys with Even-Mansour construction",
+    "effect": "renders stolen binaries unusable on non-bound devices."
+},...
+
+### Document
+"""
+
+    return prompt