import gradio as gr
import platform
import psutil
import torch
import subprocess
import sys

def get_system_info():
    os_info = platform.platform()
    cpu_info = platform.processor()
    memory = psutil.virtual_memory().total / (1024 ** 3)  # Convert to GB
    storage = psutil.disk_usage('/').total / (1024 ** 3)  # Convert to GB

    cuda_version = torch.version.cuda if torch.cuda.is_available() else "Not available"

    try:
        nvidia_driver = subprocess.check_output(['nvidia-smi', '--query-gpu=driver_version', '--format=csv,noheader']).decode('utf-8').strip()
    except:
        nvidia_driver = "Not available"

    python_version = sys.version

    try:
        gcc_version = subprocess.check_output(['gcc', '--version']).decode('utf-8').split('\n')[0]
    except:
        gcc_version = "Not available"

    try:
        pip_list = subprocess.check_output([sys.executable, '-m', 'pip', 'list', '--format=freeze']).decode('utf-8')
    except:
        pip_list = "Unable to retrieve pip list"

    return f"""
    Operating System: {os_info}
    CPU: {cpu_info}
    Memory: {memory:.2f} GB
    Storage: {storage:.2f} GB
    CUDA Version: {cuda_version}
    NVIDIA Driver Version: {nvidia_driver}
    Python Version: {python_version}
    GCC Version: {gcc_version}

    Installed Python Packages:
    {pip_list}
    """

def greet(name):
    system_info = get_system_info()
    return f"Hello {name}!!\n\nSystem Information:\n{system_info}"

demo = gr.Interface(fn=greet, inputs="text", outputs="text")
demo.launch()