maryna7679 commited on
Commit
c0f0641
·
1 Parent(s): 4149ef5

DB storage debug 6

Browse files
Functions/caption_editor_functions.py CHANGED
@@ -14,47 +14,19 @@ def get_captions_by_video_id(video_id):
14
 
15
 
16
  def save_dataframe(df, df_full, video_id, user):
17
-
18
- print("df:")
19
- print(df)
20
- print("df full:")
21
- print(df_full)
22
-
23
  if len(df.index) > len(df_full.index):
24
  new_rows = len(df_full.index) - len(df.index)
25
  empty_data = {col: [None for _ in range(new_rows)] for col in df_full.columns}
26
  empty_df = pd.DataFrame(empty_data)
27
 
28
  df_full = pd.concat([df_full, empty_df], ignore_index=True)
29
-
30
- print("extra rows df full:")
31
- print(df_full)
32
-
33
  try:
34
- print("try:")
35
-
36
  df_full["user_id"] = user
37
- # .loc[
38
- # df_full["start_time"] != df["Start"] or
39
- # df_full["end_time"] != df["End"] or
40
- # df_full["text"] != df["Text"]
41
- # ] = user
42
-
43
- print("df full:")
44
- print(df_full)
45
-
46
  df_full["start_time"] = df["Start"].apply(lambda x: float(x))
47
  df_full["text"] = df["Text"]
48
  df_full["end_time"] = df["End"].apply(lambda x: float(x))
49
 
50
- print("df full:")
51
- print(df_full)
52
-
53
- df_json = df_full.to_json(orient="index")
54
-
55
- print("json")
56
- print(df_json)
57
-
58
  default_app.database().child("Captions").child(video_id).set(df_json)
59
 
60
  return "Save successful!"
 
14
 
15
 
16
  def save_dataframe(df, df_full, video_id, user):
 
 
 
 
 
 
17
  if len(df.index) > len(df_full.index):
18
  new_rows = len(df_full.index) - len(df.index)
19
  empty_data = {col: [None for _ in range(new_rows)] for col in df_full.columns}
20
  empty_df = pd.DataFrame(empty_data)
21
 
22
  df_full = pd.concat([df_full, empty_df], ignore_index=True)
 
 
 
 
23
  try:
 
 
24
  df_full["user_id"] = user
 
 
 
 
 
 
 
 
 
25
  df_full["start_time"] = df["Start"].apply(lambda x: float(x))
26
  df_full["text"] = df["Text"]
27
  df_full["end_time"] = df["End"].apply(lambda x: float(x))
28
 
29
+ df_json = df_full.to_dict(orient="index")
 
 
 
 
 
 
 
30
  default_app.database().child("Captions").child(video_id).set(df_json)
31
 
32
  return "Save successful!"