app.py

import gradio as gr
import pandas as pd
import numpy as np
import json
from io import StringIO

def test(input_json):
    print("Received input")
    # Parse the input JSON string
    try:
        inputs = json.loads(input_json)
    except json.JSONDecodeError:
        inputs = json.loads(input_json.replace("'", '"'))

    # Accessing the 'a_list' string and converting it to a list of integers
    ids_index = inputs['input']['ids_list']

    # Extract the datatree part which is a list of dictionaries
    matrix = inputs['input']["matrix"]
    weights = inputs['input']["weights"]

    alpha = inputs['input']["alpha"]
    alpha = float(alpha)
    threshold = inputs['input']["threshold"]
    threshold = float(threshold)
    

    df_matrix = pd.DataFrame(matrix).T
    df_weights = pd.DataFrame(weights).T
    df_matrix = df_matrix.round(0).astype(int)
    df_weights = df_weights.round(0).astype(int)
    
    def computeAccessibility (DistanceMatrix,destinationWeights, alpha = 0.0038, threshold = 600):
    
        decay_factors = np.exp(-alpha * DistanceMatrix) * (DistanceMatrix <= threshold)
        subdomainsAccessibility = pd.DataFrame(index=DistanceMatrix.index, columns=destinationWeights.columns)
               
        for col in destinationWeights.columns:
            subdomainsAccessibility[col] = (decay_factors * destinationWeights[col].values).sum(axis=1)
      
        #subdomainsAccessibility.drop(columns='commercial', inplace=True)
          
        return subdomainsAccessibility
    
    subdomainsAccessibility = computeAccessibility(df_matrix,df_weights,alpha,threshold)
    #subdomainsAccessibility.columns = [int(col) for col in subdomainsAccessibility.columns]
    #subdomainsAccessibility_sorted = subdomainsAccessibility.sort_index(axis=1)
    subdomainsAccessibility_dictionary = subdomainsAccessibility.to_dict('index')
    
    subdomainsAccessibility_dictionary = {outer_key: dict(sorted(inner_dict.items(), key=lambda x: int(x[0])))
        for outer_key, inner_dict in subdomainsAccessibility_dictionary.items()}
    #        subdomainsAccessibility.columns = [int(col) for col in subdomainsAccessibility.columns]
    #    subdomainsAccessibility = subdomainsAccessibility.sort_index(axis=1)

    
    # Prepare the output
    output = {
        "subdomainsAccessibility_dictionary": subdomainsAccessibility_dictionary
    }

    return json.dumps(output)

    # Define the Gradio interface with a single JSON input
iface = gr.Interface(
    fn=test,
    inputs=gr.Textbox(label="Input JSON", lines=20, placeholder="Enter JSON with all parameters here..."),
    outputs=gr.JSON(label="Output JSON"),
    title="testspace"
)

iface.launch()