Update main.py

main.py

@@ -6,22 +6,23 @@ app = FastAPI()
 
 @app.get("/")
 def run_notebook(fruits: str):
+    
     fruit_list = fruits.split(",")
 
+    with open("fruits.txt", "w") as f:
+        f.write("\n".join(fruit_list))
+
     # Execute the notebook with Papermill
-    output_path = "output.ipynb"
     pm.execute_notebook(
         "fruitchecker.ipynb",
-        output_path,
         parameters={"fruits": fruit_list}
     )
 
+    output_path = "output.csv"
+    
     # Extract the CSV result
     with open(output_path, "r") as f:
-        notebook_content = f.read()
-    
-    # Find the CSV output in the notebook
-    csv_output = notebook_content.split("csv_output = ")[-1].strip().strip("\"'")
+        csv_output = f.read()
 
     # Return as downloadable CSV
     return Response(content=csv_output, media_type="text/csv",