Update app.py

app.py

@@ -7,37 +7,22 @@ import os
 
 def convert_parquet_to_jsonl(file):
     df = pd.read_parquet(file.name)
-    output_file = tempfile.NamedTemporaryFile(delete=False, suffix='.jsonl')
-    df.to_json(output_file.name, orient='records', lines=True)
-    return output_file.name
-
-def convert_json_to_jsonl(file):
-    with open(file.name, 'r') as f:
-        data = json.load(f)
-    
-    output_file = tempfile.NamedTemporaryFile(delete=False, suffix='.jsonl')
-    with jsonlines.open(output_file.name, mode='w') as writer:
-        if isinstance(data, list):
-            writer.write_all(data)
-        else:
-            writer.write(data)
-    
+    output_file = tempfile.NamedTemporaryFile(delete=False, suffix='.json')
+    df.to_json(output_file.name, orient='records')
     return output_file.name
 
 def convert_file(file):
     if file.name.endswith('.parquet'):
         return convert_parquet_to_jsonl(file)
-    elif file.name.endswith('.json'):
-        return convert_json_to_jsonl(file)
     else:
         return "Unsupported file format. Please upload a .parquet or .json file."
 
 iface = gr.Interface(
     fn=convert_file,
-    inputs=gr.File(label="Upload .parquet or .json file"),
-    outputs=gr.File(label="Download converted .jsonl file"),
-    title="Parquet/JSON to JSONL Converter",
-    description="Upload a .parquet or .json file to convert it to .jsonl format."
+    inputs=gr.File(label="Upload .parquet file"),
+    outputs=gr.File(label="Download converted .json file"),
+    title="Parquet to JSON Converter",
+    description="Upload a .parquet file to convert it to .json format."
 )
 
 iface.launch()