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)
    
    #print("Parsed input keys:", inputs.keys())

    #print("Parsed input values:", inputs.values())
    #sum_list = sum(float(item) for item in inputs["a_list"])
    #multiplication = [-float((inputs["alpha"])) * float(item) for item in inputs["a_list"]]
    #new_df = pd.DataFrame(index=inputs["dataframe"].index, columns=inputs["dataframe"].columns)
    #multiplier_series = pd.Series(float(inputs["a_list"]), index=inputs["dataframe"].index)
    #new_df["new column"] = float(inputs["dataframe"]).mul(multiplier_series, axis=0)

    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_dict = subdomainsAccessibility.to_dict('index')

    # Prepare the output
    output = {
        "list": ids_index,
        "subdomainsAccessibility_dict": subdomainsAccessibility_dict
    }

    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()