MTabishS/Text2SQL

1import requests2import gradio as gr3 4def query_model(question):5    url="http://0.0.0.0:8000/generate-sql"  # FastAPI endpoint6    response=requests.post(url,json={'question':question})7 8    if response.status_code == 200:9        result=response.json()10 11        # formatting output to show both SQL query and data12        return f"SQL Query: {result['sql_query']}\n\nData: {result['data']}"13    else:14        return f"Error Occurred: {response.text}"15 16# Creating Gradio Interface17gr_interface = gr.Interface(18    fn=query_model,19    inputs='text',20    outputs='text',21    title='Text2SQL Generation using LLM',22    description='Please enter your question to generate SQL query and retrieve the data!'23)24 25if __name__=="__main__":26    # launch gradio interface27    gr_interface.launch(server_name='0.0.0.0', server_port=7860, share=True)28    29    # gr_interface.launch(share=True,server_name='0.0.0.0',server_port=7860)