app.py

import subprocess
from flask import Flask, render_template, request
from urllib.parse import urlparse 
from flask import jsonify
from flask_cors import CORS
from AI import consultantAI
app = Flask(__name__)
CORS(app, resources={r"/get_report_data": {"origins": "http://localhost:3000"}})

# Define a function to call the web vulnerability scanner
def web_vulnerability_scan(website_url):
    domain_name = urlparse(website_url).netloc
    report_file_name = f"./report/{domain_name}_report.txt"
    subprocess.check_output(['python', 'web-vulnerability-scanner.py', 'full', website_url])
    subprocess.check_output(['python', 'wap.py', '-u', website_url ,'-wf', report_file_name])
    subprocess.check_output(['python', 'virus-total.py', website_url , report_file_name])

    
    return report_file_name

@app.route('/test_website', methods=['POST'])
def test_website():
    website_url = request.form['website_url']

    # Call the web vulnerability scanner function
    report_file_name = web_vulnerability_scan(website_url)
    

    scanner_output = subprocess.check_output(['python', 'simple.py', website_url])
    


    file_path = report_file_name 
    ai_instance = consultantAI()
    prompt=ai_instance.read_report(file_path)
    solution_content = ai_instance.generate_solution(prompt)
    
# Write the output to the report file
    with open(report_file_name, 'a') as report_file:
        report_file.write(scanner_output.decode('utf-8'))
        
    # Read the content of the report file
    with open(report_file_name, 'r') as report_file:
        tool_output = report_file.read()

    # Split the content by newlines for organized display
    tool_output_lines = tool_output.split('\n')
    if solution_content is None:
        solutions_list = ["API key invalid see AI.Py "]
    else:
        solutions_list = solution_content.strip().split('\n')

    if request.headers.get('X-Requested-With') == 'XMLHttpRequest':
        # Return the solution content as JSON
        return jsonify({'solution_content': solution_content,'tool_output': tool_output})
   
    else:
        # Return the HTML page
        print(solution_content)
        return render_template('result.html', tool_output_lines=tool_output_lines,solutions=solutions_list)
        
@app.route('/')
def index():
    return render_template('index.html')
@app.route('/get_report_data', methods=['POST', 'GET'])
def get_report_data():
    website_url = request.json.get('website_url')
    report_file_name = web_vulnerability_scan(website_url)
    with open(report_file_name, 'r') as report_file:
        tool_output = report_file.readlines()

    return jsonify({'tool_output': tool_output})

if __name__ == '__main__':
    app.run(debug=True)