diff --git a/app.py b/app.py index 3d1808cf6..a116fc481 100644 --- a/app.py +++ b/app.py @@ -1,5 +1,17 @@ import os +from langchain_openai import AzureChatOpenAI +from azure.identity import DefaultAzureCredential, get_bearer_token_provider + +from dotenv import load_dotenv + +load_dotenv() + +azure_endpoint = os.environ['AZURE_OPENAI_ENDPOINT'] +azure_deployment = os.environ['AZURE_OPENAI_DEPLOYMENT_NAME'] +model_name = os.environ['OPENAI_MODEL_NAME'] +api_version = os.environ['AZURE_OPENAI_API_VERSION'] + from flask import (Flask, redirect, render_template, request, send_from_directory, url_for) @@ -18,12 +30,26 @@ def favicon(): @app.route('/hello', methods=['POST']) def hello(): - name = request.form.get('name') - - if name: - print('Request for hello page received with name=%s' % name) - return render_template('hello.html', name = name) - else: + req = request.form.get('req') + + token_provider = get_bearer_token_provider( + DefaultAzureCredential(), "https://cognitiveservices.azure.com/.default" + ) + + llm = AzureChatOpenAI( + api_version=api_version, + azure_deployment=azure_deployment, + azure_endpoint=azure_endpoint, + model_name=model_name, + azure_ad_token_provider=token_provider + ) + + text = llm.invoke(req) + + if req: + print('Request for hello page received with req=%s' % req) + return render_template('hello.html', req=text.content) + else: print('Request for hello page received with no name or blank name -- redirecting') return redirect(url_for('index')) diff --git a/requirements.txt b/requirements.txt index b5065c236..cd7530ea8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,5 @@ Flask==3.0.0 -gunicorn \ No newline at end of file +gunicorn +langchain-openai +python-dotenv +azure-identity \ No newline at end of file diff --git a/templates/hello.html b/templates/hello.html index c277e7740..34ecd2b2d 100644 --- a/templates/hello.html +++ b/templates/hello.html @@ -1,6 +1,6 @@ - Hello Azure - Python Quickstart + Hello Azure - Python AI App @@ -9,10 +9,9 @@
Azure Logo - -

Hello {{name}}

+

OpenAI response:

- It is nice to meet you! + {{req}}

Back home
diff --git a/templates/index.html b/templates/index.html index 9cdba30cd..8f8e6aa89 100644 --- a/templates/index.html +++ b/templates/index.html @@ -1,6 +1,6 @@ - Hello Azure - Python Quickstart + Hello Azure - Python AI App @@ -9,19 +9,17 @@
Azure Logo -

Welcome to Azure

- + -
- +
- +