Spaces:
Sleeping
Sleeping
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 |
-
|
| 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!"
|