Created a server to serve apkg files in respose to image posts
This commit is contained in:
parent
af4b615aa3
commit
c3ca249877
|
@ -1,3 +1,4 @@
|
|||
genanki==0.8.0
|
||||
Pillow
|
||||
openai
|
||||
openai
|
||||
flask
|
42
server.py
Normal file
42
server.py
Normal file
|
@ -0,0 +1,42 @@
|
|||
from flask import Flask, request, send_from_directory, jsonify
|
||||
import os
|
||||
import tempfile # For creating temporary directories
|
||||
import shutil # For removing directories
|
||||
|
||||
from pipeline import pipeline
|
||||
|
||||
app = Flask(__name__)
|
||||
|
||||
@app.route('/generate-deck', methods=['POST'])
|
||||
def generate_deck():
|
||||
# Assuming images are sent as multipart/form-data
|
||||
if 'image' not in request.files:
|
||||
return jsonify({'error': 'No image part'}), 400
|
||||
|
||||
images = request.files.getlist('image')
|
||||
|
||||
if not images or all([img.filename == '' for img in images]):
|
||||
return jsonify({'error': 'No selected file'}), 400
|
||||
|
||||
# Create a temporary directory to store multiple images
|
||||
temp_dir = tempfile.mkdtemp()
|
||||
|
||||
image_paths = []
|
||||
for img in images:
|
||||
image_path = os.path.join(temp_dir, img.filename)
|
||||
img.save(image_path)
|
||||
image_paths.append(image_path)
|
||||
|
||||
try:
|
||||
# Run the pipeline using the saved images
|
||||
# You might need to modify your pipeline to accept and handle multiple images
|
||||
pipeline(temp_dir) # Assuming pipeline works per directory of images
|
||||
return send_from_directory('.', 'output.apkg', as_attachment=True)
|
||||
except Exception as e:
|
||||
return jsonify({'error': str(e)}), 500
|
||||
finally:
|
||||
# Cleanup: Remove the temporary directory and its content
|
||||
shutil.rmtree(temp_dir)
|
||||
|
||||
if __name__ == '__main__':
|
||||
app.run(debug=True)
|
Loading…
Reference in New Issue
Block a user