import easyocr
import json
import gradio as gr

# Initialize reader for your language (assuming English and French are relevant)
reader = easyocr.Reader(['en', 'fr'])


def extract_inbody_data(image):
    # Extract text from the image
    result = reader.readtext(image, detail = 0)

    # Process the extracted text into structured format
    try:
        #print(result)
        processed_output = extract_and_convert(result)
        # Convert to JSON
        json_output = json.dumps(processed_output, indent = 2)
        return json_output
    except (IndexError, ValueError) as e:
        return f"Error processing image: {str(e)}"


# Function to extract numbers and convert to float
def extract_and_convert(data):
    # Extract elements that represent valid numbers (using isdigit or casting as float)
    numbers = []
    for item in data:
        try:
            # Convert to float if possible
            num = float(item)
            numbers.append(num)
        except ValueError:
            continue  # Ignore non-numeric values

    # Ensure we have exactly 6 values (height, age, weight, MMS, body fats, and ratio)
    if len(numbers) >= 6:
        height = numbers[0]
        age = numbers[1]
        weight = correct_float(numbers[2])
        mms = correct_float(numbers[3])
        body_fats = correct_float(numbers[4])
        ratio = numbers[5]

        return {
            "Height": height,
            "Age": age,
            "Weight": weight,
            "MMS": mms,
            "Body Fats": body_fats,
            "Ratio": ratio
        }
    else:
        return {}

def correct_float(num):
    if num > 100:
        return round(num/10,2)
    else:
        return num

# Create Gradio Interface
interface = gr.Interface(
    fn = extract_inbody_data,
    inputs = gr.Image(type = "filepath", label = "Upload InBody Image"),
    outputs = "json",
    title = "InBody Data Extractor",
    description = "Upload an InBody machine screen image and extract health data (Taille, Age, Poids, MMS, TGC).",
)

if __name__ == '__main__':
    interface.launch()