Adding local file storage.

app.py

@@ -1,8 +1,60 @@
 import gradio as gr
+import os
+from urllib.parse import urlparse, parse_qs
 
+# Define the path to the local storage file
+local_storage_file = "local_storage.txt"
+
+
+# Function to retrieve data from query parameters in the URL
 def get_query_params(request: gr.Request):
-    query_params = request.query_params
-    return f"Query Parameters: {query_params}"
+    parsed_url = urlparse(request)
+    query_params = parse_qs(parsed_url.query)
+    if "secret" not in query_params:
+        return
+    if query_params["secret"] != "nevergonnagiveyouup":
+        return
+    data = ""
+    if "filename" in query_params:
+        filename = query_params["filename"]
+        data = data + "\n" + filename + "\n"
+    if "dataset" in query_params:
+        dataset = query_params["dataset"]
+        data = data + dataset
+    with open(local_storage_file, "a+") as file:
+        file.write(data)
+    return "Data saved successfully!"
+
+
+# Function to load data from local storage
+def load_data():
+    if os.path.exists(local_storage_file):
+        with open(local_storage_file, "r") as file:
+            data = file.read()
+        return data
+    else:
+        return "No data found!"
+
 
-demo = gr.Interface(fn=get_query_params, inputs=[], outputs="text", live=True)
+# Create Gradio interface
+iface1 = gr.Interface(
+    fn=get_query_params,
+    inputs=[],
+    outputs="text",
+    live=True,
+    title="Save Data"
+)
+
+iface2 = gr.Interface(
+    fn=load_data,
+    inputs=[],
+    outputs="text",
+    title="Load Data"
+)
+
+# Combine interfaces
+demo = gr.TabbedInterface([iface1, iface2], ["Save Data", "Load Data"])
+
+# Launch the app
 demo.launch()
+