Update app.py

app.py

@@ -4,6 +4,8 @@ import json
 import gradio as gr
 import pandas as pd
 from datetime import datetime
+import tempfile
+import os
 
 class ValidationStatus(BaseModel):
     is_valid: bool
@@ -188,6 +190,9 @@ Return a complete JSON object with all fields populated, including explanation a
 
 def export_query(sql_query, schema, sample_data):
     """Export query with schema and sample data as a complete SQL file"""
+    if not sql_query or sql_query.startswith("Error"):
+        return None
+    
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     content = f"""-- Generated SQL Query
 -- Timestamp: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
@@ -207,10 +212,28 @@ def export_query(sql_query, schema, sample_data):
 -- QUERY
 {sql_query}
 """
-    return content
+    
+    # Create a temporary file with proper naming
+    temp_file = tempfile.NamedTemporaryFile(
+        mode='w',
+        suffix='.sql',
+        prefix=f'sql_query_{timestamp}_',
+        delete=False
+    )
+    
+    try:
+        temp_file.write(content)
+        temp_file.flush()
+        temp_file.close()
+        return temp_file.name
+    except Exception as e:
+        print(f"Error creating file: {e}")
+        if os.path.exists(temp_file.name):
+            os.unlink(temp_file.name)
+        return None
 
 # Create Gradio interface with enhanced features
-with gr.Blocks(title="SQL Query Generator Pro", theme=gr.themes.Ocean()) as demo:
+with gr.Blocks(title="SQL Query Generator Pro", theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
         # 🗄️ Natural Language to SQL Query Generator Pro