1 files changed, 19 insertions, 0 deletions
diff --git a/scripts/scraping/ai_scraping.py b/scripts/scraping/ai_scraping.py
new file mode 100644
index 0000000..b11a12b
--- /dev/null
+++ b/scripts/scraping/ai_scraping.py
@@ -0,0 +1,19 @@
+from google import genai
+import json
+import requests
+import Constants
+import AIParams
+
+client = genai.Client(api_key="AIzaSyAdB7yo0qcnwHeC4T2rRaSXD588JRw94oQ")
+
+def run_ai_query(url):
+    req_url = f"https://r.jina.ai/{url}"
+    res = requests.get(req_url)
+    markdown_content = res.text
+
+    prompt = AIParams.PROMPT.format(schema=AIParams.RECIPE_SCHEMA, markdown=markdown_content)
+
+    ai_res = client.models.generate_content(model="gemini-2.0-flash", contents=prompt)
+    cleaned_text = ai_res.text.strip("```").strip("```json")
+    recipe_json = json.loads(cleaned_text)
+    return {"success": True, "data": recipe_json}