Update app.py

app.py

@@ -1,17 +1,15 @@
-import gradio as gr
-import pandas as pd
 import requests
+import pandas as pd
+import gradio as gr
 import spaces
 
 @spaces.GPU
-def convert_parquet_to_jsonl(parquet_file=None, parquet_url=None):
-    if parquet_file is not None:
-        df = pd.read_parquet(parquet_file.name)
-    elif parquet_url is not None:
-        response = requests.get(parquet_url)
-        df = pd.read_parquet(response.content)
-    else:
-        raise ValueError("Either parquet_file or parquet_url must be provided")
+def download_file(url):
+    response = requests.get(url)
+    return response.content
+
+def convert_parquet_to_jsonl(parquet_file):
+    df = pd.read_parquet(parquet_file)
     jsonl_data = df.to_json(orient='records', lines=True)
     return jsonl_data
 
@@ -22,15 +20,15 @@ def main(url):
         f.write(jsonl_data)
     return "output.jsonl"
 
-theme = "Ytheme/Minecraft"
-
-demo = gr.Interface(theme=theme,
-    fn=convert_parquet_to_jsonl,
-    inputs=[gr.File(label="Parquet File"), gr.Textbox(label="Parquet File URL")],
-     outputs=[gr.File(label="JSONL Output")],
-    title="Parquet to JSONL Converter",
-    description="Input a Parquet file by a downloadable link or file upload and convert it to JSONL format"
-)
+def gradio_interface():
+    demo = gr.Interface(
+        fn=main,
+        inputs=[gr.Textbox(label="Parquet File URL")],
+        outputs=[gr.File(label="JSONL Output")],
+        title="Parquet to JSONL Converter",
+        description="Convert a Parquet file to JSONL format from a downloadable link"
+    )
+    demo.launch()
 
 if __name__ == "__main__":
-    demo.launch()
+    gradio_interface()