More comments in the code

This commit is contained in:
Joakim Persson
2024-07-16 11:26:16 +02:00
parent 957b8f46c8
commit 8d9369c502
+15 -1
View File
@@ -1,21 +1,35 @@
# Example function to interact with OLLAMA
# Import the necessary library from ollama module
from ollama import Client from ollama import Client
def get_response(user_query): def get_response(user_query):
# Create a client object for interacting with OLLAMA API
client = Client() client = Client()
# Generate and retrieve the response based on user's query
response = client.generate_response(user_query) response = client.generate_response(user_query)
# Return the generated response
return response return response
# Flask endpoint for user interaction # Flask endpoint for user interaction
from flask import Flask, request, jsonify from flask import Flask, request, jsonify
# Initialize a Flask application
app = Flask(__name__) app = Flask(__name__)
@app.route('/smartassist', methods=['POST']) @app.route('/smartassist', methods=['POST'])
def smartassist(): def smartassist():
# Extract the query from the incoming JSON data
data = request.json data = request.json
user_query = data['query'] user_query = data['query']
# Get the response from the OLLAMA API based on the user's query
response = get_response(user_query) response = get_response(user_query)
# Return the response as a JSON object in the HTTP response
return jsonify({"response": response}) return jsonify({"response": response})
# Run the Flask application if this script is executed directly
if __name__ == '__main__': if __name__ == '__main__':
app.run(debug=True) app.run(debug=True)