Update app.py

app.py

@@ -1,11 +1,22 @@
 import gradio as gr
 import json
+from huggingface_hub import HfApi
+import uuid
+
+# Initialize the Hugging Face API
+api = HfApi()
+
+# Replace with your Hugging Face username and repository name
+REPO_ID = "ontocord/prompt-share-storage" 
 
 def save_to_json(text1, text2, text3, video, agree):
-  """Saves the input text and video to a JSON file if the checkbox is checked."""
+  """Saves the input text and video with UUIDs to files and uploads to Hugging Face Hub."""
   if agree:
-      # Save the video to an mp4 file
-      video_filename = "uploaded_video.mp4" 
+      # Generate a unique UUID
+      file_uuid = str(uuid.uuid4())
+
+      # Save the video with UUID prepended to the filename
+      video_filename = f"{file_uuid}_uploaded_video.mp4" 
       video.save(video_filename)
 
       data = {
@@ -14,24 +25,40 @@ def save_to_json(text1, text2, text3, video, agree):
           "text3": text3,
           "video_filename": video_filename
       }
-      with open("data.json", "w") as f:
+
+      # Save the JSON with UUID prepended to the filename
+      json_filename = f"{file_uuid}_data.json"
+      with open(json_filename, "w") as f:
           json.dump(data, f)
-      return "Data saved to data.json"
+
+      # Upload the files to Hugging Face Hub
+      api.upload_file(
+          path_or_fileobj=json_filename,
+          path_in_repo=json_filename,
+          repo_id=REPO_ID
+      )
+      api.upload_file(
+          path_or_fileobj=video_filename,
+          path_in_repo=video_filename,
+          repo_id=REPO_ID
+      )
+
+      return "Data saved and uploaded to Hugging Face Hub."
   else:
       return "Please agree to the terms before submitting."
 
 iface = gr.Interface(
   fn=save_to_json,
   inputs=[
-      gr.TextArea(lines=5, placeholder="Enter text 1 here..."),
-      gr.TextArea(lines=5, placeholder="Enter text 2 here..."),
-      gr.TextArea(lines=5, placeholder="Enter text 3 here..."),
+      gr.TextArea(lines=5, placeholder="Enter a conversation about video (e.g., Q: Hi - what am I doing? A: Good, it looks like you are at a cafe."),
+      gr.TextArea(lines=5, placeholder="Enter a detailed description about the video (e.g., This video shows a person at a cafe, with nosiy happy patrons, and a dynamic atmosphere)"),
+      gr.TextArea(lines=5, placeholder="Enter a question and answer that requires reasoning (e.g., Q: If two friends joined, how many coffees would be needed? A: Since you have one coffee in your hand, and two friends joined you, assuming they each like a coffee, two more coffees are needed, making a total of 3 coffees in this scene)"),
       gr.Video(format="mp4"), 
       gr.Checkbox(label="I agree and have the rights to share these prompts under the CC-BY license.")
   ],
   outputs="text",
-  title="Save Text and Video to JSON",
-  description="Enter text, upload a video, and click submit to save to a JSON file."
+  title="Save Text and Video to Hugging Face",
+  description="Share a video and creation AI training data. Enter a conversation, description and reasoning prompts about the video you upload, upload the video, and click submit to save to ontocord/prompt-share-storage Dataset. DO NOT share personal information."
 )
 
 iface.launch()