Spaces:
Sleeping
Sleeping
Delete app/run.py
Browse files- app/run.py +0 -40
app/run.py
DELETED
@@ -1,40 +0,0 @@
|
|
1 |
-
import os
|
2 |
-
|
3 |
-
# Ensure the HF_HOME environment variable is set correctly
|
4 |
-
cache_dir = os.getenv('HF_HOME', '/app/cache')
|
5 |
-
|
6 |
-
# Ensure the directory exists
|
7 |
-
os.makedirs(cache_dir, exist_ok=True)
|
8 |
-
|
9 |
-
import logging
|
10 |
-
from flask import Flask, request, jsonify
|
11 |
-
from app.search_engine import PromptSearchEngine
|
12 |
-
|
13 |
-
app = Flask(__name__)
|
14 |
-
|
15 |
-
# Disable parallelism for tokenizers
|
16 |
-
os.environ['TOKENIZERS_PARALLELISM'] = 'false'
|
17 |
-
|
18 |
-
# Configure logging
|
19 |
-
logging.basicConfig(level=logging.INFO)
|
20 |
-
logger = logging.getLogger(__name__)
|
21 |
-
|
22 |
-
search_engine = PromptSearchEngine()
|
23 |
-
|
24 |
-
@app.route('/search', methods=['POST'])
|
25 |
-
def search():
|
26 |
-
data = request.get_json()
|
27 |
-
query = data.get('query')
|
28 |
-
n = data.get('n', 5)
|
29 |
-
use_pinecone = data.get('use_pinecone', True)
|
30 |
-
|
31 |
-
logger.info(f"Received query: {query} with n: {n} and use_pinecone: {use_pinecone}")
|
32 |
-
results = search_engine.most_similar(query, n, use_pinecone)
|
33 |
-
formatted_results = [{'score': score, 'prompt': prompt} for score, prompt in results]
|
34 |
-
logger.info(f"Returning results: {formatted_results}")
|
35 |
-
return jsonify(formatted_results)
|
36 |
-
|
37 |
-
|
38 |
-
if __name__ == '__main__':
|
39 |
-
logger.info("Starting Flask server")
|
40 |
-
app.run(debug=True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|