Update app.py

app.py

@@ -8,6 +8,7 @@ from database import init_chromadb, populate_sample_db, store_program, query_pro
 
 # User-configurable variables
 UPLOAD_DIR = "./uploads"  # Directory for uploads
+USE_GPU = False  # Default to CPU, set to True for GPU if available
 
 app = Flask(__name__)
 
@@ -101,8 +102,7 @@ def index():
 @app.route('/export_json', methods=['POST'])
 def export_json():
     parts = request.json.get('parts', [])
-    export_data = [{'vector': part['vector'], 'source': part['source'], 'description': generate_description_tokens([part['category']], [part['vector']])}
-                   for part in parts]
+    export_data = [{'vector': part['vector'], 'source': part['source'], 'description': generate_description_tokens([part['category']], [part['vector']])} for part in parts]
     json_str = json.dumps(export_data, indent=2)
     buffer = io.BytesIO(json_str.encode('utf-8'))
     buffer.seek(0)
@@ -141,4 +141,4 @@ def generate_description_tokens(sequence, vectors):
 if __name__ == '__main__':
     if not os.path.exists(UPLOAD_DIR):
         os.makedirs(UPLOAD_DIR)
-    app.run(host="0.0.0.0", port=7860)
+    app.run(port=7860)