import gradio as gr
from prediction_script import predict  # Import your prediction function

iface = gr.Interface(
    fn=predict, 
    inputs=gr.Dataframe(
       headers=["Store", "Holiday_Flag", "Temperature", "Fuel_Price", "CPI", "Unemployment", 
                "Weekly_Sales_Lag1", "Weekly_Sales_Lag2", "Weekly_Sales_Lag4", 
                "Sales_Fuel_Interaction", "Sales_CPI_Interaction", "Sales_Unemployment_Interaction",
                "Month", "Quarter", "Year", "Holiday_Impact"],
       type="pandas"
    ),
    outputs=gr.Number(),
    title="Walmart Sales Prediction",
    description="Enter the features to predict weekly sales."
)

iface.launch()