import osfrom flask import Flask, request, jsonifyfrom firecrawl import Firecrawlapp = Flask(__name__)firecrawl = Firecrawl(api_key=os.environ["FIRECRAWL_API_KEY"])@app.post("/search")def search(): data = request.get_json() results = firecrawl.search(data["query"], limit=data.get("limit", 5)) return jsonify([{"title": r.title, "url": r.url} for r in results.web])@app.post("/scrape")def scrape(): data = request.get_json() result = firecrawl.scrape(data["url"]) return jsonify(markdown=result.markdown, metadata=result.metadata)@app.post("/interact/start")def interact_start(): data = request.get_json() result = firecrawl.scrape(data["url"], formats=["markdown"]) return jsonify(scrape_id=result.metadata.scrape_id)@app.post("/interact")def interact(): data = request.get_json() response = firecrawl.interact(data["scrape_id"], prompt=data["prompt"]) return jsonify(output=response.output)@app.post("/interact/stop")def interact_stop(): data = request.get_json() firecrawl.stop_interaction(data["scrape_id"]) return jsonify(status="stopped")if __name__ == "__main__": app.run(debug=True)
