hltdx commited on
Commit
3b45285
·
unverified ·
1 Parent(s): ce59b01

Swapping out get_query_parameters function for save_data function.

Browse files
Files changed (1) hide show
  1. app.py +6 -12
app.py CHANGED
@@ -9,18 +9,13 @@ save_secret_value = os.getenv("save_secret_value")
9
  secret_value = os.getenv("secret_value")
10
 
11
  # Function to retrieve data from query parameters in the URL
12
- def get_query_params(request: gr.Request):
13
- if "secret" not in request.query_params:
14
- return "No secret provided"
15
- if request.query_params["secret"] != save_secret_value:
16
  return "Secret is incorrect"
17
  data = ""
18
- if "filename" in request.query_params:
19
- filename = request.query_params["filename"]
20
  data = data + "\n" + filename + "\n"
21
- if "dataset" in request.query_params:
22
- dataset = request.query_params["dataset"]
23
- data = data + dataset
24
  with open(local_storage_file, "a+") as file:
25
  file.write(data)
26
  return "Data saved successfully!"
@@ -49,10 +44,9 @@ def delete_data(secret):
49
 
50
  # Create Gradio interface
51
  iface1 = gr.Interface(
52
- fn=get_query_params,
53
- inputs=[],
54
  outputs="text",
55
- live=True,
56
  title="Save Data"
57
  )
58
 
 
9
  secret_value = os.getenv("secret_value")
10
 
11
  # Function to retrieve data from query parameters in the URL
12
+ def save_data(secret, dataset, filename=None):
13
+ if secret != save_secret_value:
 
 
14
  return "Secret is incorrect"
15
  data = ""
16
+ if filename:
 
17
  data = data + "\n" + filename + "\n"
18
+ data = data + dataset
 
 
19
  with open(local_storage_file, "a+") as file:
20
  file.write(data)
21
  return "Data saved successfully!"
 
44
 
45
  # Create Gradio interface
46
  iface1 = gr.Interface(
47
+ fn=save_data,
48
+ inputs=text,
49
  outputs="text",
 
50
  title="Save Data"
51
  )
52