More comments in the code
This commit is contained in:
@@ -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)
|
||||||
|
|||||||
Reference in New Issue
Block a user