app.py

import os
from flask import Flask, request, jsonify
from flask_cors import CORS, cross_origin
from azure_blob import AzureBlob
from dotenv import load_dotenv

load_dotenv()
from llm import LLM

app = Flask(__name__)
cors = CORS(app, resources={r"/*": {"origins": "*"}})

azure_blob = AzureBlob(
    connection_string=os.getenv("AZURE_STORAGE_CONNECTION_STRING"),
    container_name=os.getenv("AZURE_STORAGE_CONTAINER_NAME")
)
llm = LLM(
    azure_deployment=os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME"),
    openai_api_version=os.getenv("AZURE_OPENAI_API_VERSION"),
    temperature=float(os.getenv("AZURE_OPENAI_TEMPERATURE")),
)


@app.route('/generate_questions_from_document', methods=['POST'])
@cross_origin()
def generate_questions_from_document():
    try:
        # Get the document content
        document_content = azure_blob.retrieve_document(
            document_id=f"documents/{request.json['document_id']}"
        )
    except Exception as e:
        return jsonify({"error retrieving document": str(e)}), 404

    try:
        # Generate questions and answers
        questions = llm.generate_questions_and_answers(
            document_content=document_content,
            num_questions=request.json['num_questions'],
            difficulty=request.json['difficulty']
        )
    except Exception as e:
        return jsonify({"error generating questions": str(e)}), 500

    return jsonify(questions)


@app.route('/status', methods=['GET'])
@cross_origin()
def status():
    return jsonify({"status": "API is running"})


if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000, debug=True)