app.py

import gradio as gr
import tempfile
from data_generator import generate_test_data
import pandas as pd
import os

def generate_data(prompt):
    try:
        df = generate_test_data(prompt)
        # Save to temporary CSV file
        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".csv", mode='w', newline='')
        df.to_csv(temp_file.name, index=False)
        temp_file.close()
        return df, temp_file.name
    except Exception as e:
        return f"Error: {str(e)}", None

with gr.Blocks() as demo:
    gr.Markdown("### 🌍 Global Test Data Generator")
    prompt_input = gr.Textbox(label="Enter prompt", placeholder="e.g., Generate 50 Indian names with PAN and emails")
    generate_button = gr.Button("Generate Data")
    output_table = gr.Dataframe(label="Generated Data")
    csv_output = gr.File(label="Download CSV")

    generate_button.click(fn=generate_data, inputs=prompt_input, outputs=[output_table, csv_output])

if __name__ == "__main__":
    demo.launch()