Update app.py

app.py

@@ -1,34 +1,27 @@
-import requests
-import pandas as pd
 import gradio as gr
-import spaces
-
-@spaces.GPU
-def download_file(url):
-    response = requests.get(url)
-    return response.content
+import pandas as pd
+import requests
 
-def convert_parquet_to_jsonl(parquet_file):
-    df = pd.read_parquet(parquet_file)
+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")
     jsonl_data = df.to_json(orient='records', lines=True)
-    return jsonl_data
-
-def main(url):
-    parquet_file = download_file(url)
-    jsonl_data = convert_parquet_to_jsonl(parquet_file)
     with open("output.jsonl", "w") as f:
         f.write(jsonl_data)
     return "output.jsonl"
 
-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()
+demo = gr.Interface(
+    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="Convert a Parquet file to JSONL format from a downloadable link or file upload"
+)
 
 if __name__ == "__main__":
-    gradio_interface()
+    demo.launch()