Update app.py

app.py

@@ -1,14 +1,60 @@
-from fastapi import FastAPI
-import requests
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
 import json
-import time
+import os
+from datasets import Dataset, DatasetDict, DatasetInfo
+from huggingface_hub import login, HfApi
+from typing import List
+from fastapi.responses import JSONResponse
+from fastapi.middleware.cors import CORSMiddleware
 
+# Initialize FastAPI
 app = FastAPI()
 
-import os
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Allow all origins (or specify specific origins)
+    allow_credentials=True,
+    allow_methods=["*"],  # Allow all HTTP methods
+    allow_headers=["*"],  # Allow all headers
+)
+
+TOKEN = os.getenv("token")
+
+
+class Recipe(BaseModel):
+    ingredients: List[str]
+    instructions: str
+
+DATASET_PATH = "/mnt/data/sharktide/recipes/data"
+
+# Ensure the 'data' folder exists
+if not os.path.exists(DATASET_PATH):
+    os.makedirs(DATASET_PATH)
+
+# Authentication for Hugging Face
+login(token=TOKEN)
+
+@app.put("/add/recipe")
+async def add_recipe(filename: str, recipe: Recipe):
+    # Define the file path based on the filename query parameter
+    file_path = os.path.join(DATASET_PATH, f"{filename}.json")
 
-@app.get("/")
-def home():
-    return("Status": "Working")
+    # Check if the file already exists
+    if os.path.exists(file_path):
+        raise HTTPException(status_code=400, detail="File already exists")
 
+    # Prepare the data to be written in JSON format
+    recipe_data = recipe.dict()  # Convert Recipe model to dictionary
 
+    # Write the data to the new file
+    try:
+        with open(file_path, "w") as f:
+            json.dump(recipe_data, f, indent=4)
+        
+        dataset = Dataset.from_json(file_path)  # Load the new file
+        dataset.push_to_hub("sharktide/recipes")  # Push the dataset to the Hugging Face Hub
+        
+        return {"message": f"Recipe '{filename}' added successfully."}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error writing file: {str(e)}")