Initial Commit

app.py

@@ -0,0 +1,18 @@
+import gradio as gr
+from prompt_generator import generate_prompt
+from gpt4o_infer import call_gpt4o
+
+def nlp_to_sql(nl_query, schema_text):
+    prompt = generate_prompt(nl_query, schema_text)
+    sql = call_gpt4o(prompt)
+    return sql
+
+iface = gr.Interface(
+    fn=nlp_to_sql,
+    inputs=[gr.Textbox(label="Natural Language Query"), gr.Textbox(label="Schema in Natural Format")],
+    outputs="text",
+    title="NLP to SQL with GPT-4o",
+    description="Enter your natural language query and schema to get SQL"
+)
+
+iface.launch()

gpt4o_infer.py

@@ -0,0 +1,16 @@
+import openai
+import os
+
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+def call_gpt4o(prompt):
+    response = openai.ChatCompletion.create(
+        model="gpt-4o",
+        messages=[
+            {"role": "system", "content": "You are a helpful assistant that converts natural language into SQL queries using the provided schema."},
+            {"role": "user", "content": prompt}
+        ],
+        temperature=0.2,
+        max_tokens=512
+    )
+    return response['choices'][0]['message']['content']

prompt_generator.py

@@ -0,0 +1,10 @@
+def generate_prompt(query, schema):
+    return f"""Given the database schema below:
+
+{schema}
+
+Translate the following question into SQL:
+
+Question: {query}
+
+Only provide SQL output without explanation."""

requirements.txt

@@ -0,0 +1,2 @@
+gradio
+openai