Update app.py

app.py

@@ -1,63 +1,34 @@
 import gradio as gr
-from datetime import datetime
-import random
-import uuid
-from dataclasses import dataclass
+from huggingface_hub import HfApi
+import os
 
-# Shared state to store messages
-messages = []
+api = HfApi()
 
-# Dictionary to store user colors
-user_colors = {}
-
-@dataclass
-class Data:
-    user_id: str
-    message: str
-    timestamp: str
-
-def get_user_color(user_id):
-    if user_id not in user_colors:
-        user_colors[user_id] = f"#{random.randint(0, 0xFFFFFF):06x}"
-    return user_colors[user_id]
-
-def chat(input_text, history):
-    global messages
-    
-    # Parse the input to get user_id and message
-    user_id, message = input_text.split("; ")
+def upload_video(video):
+    # Save the uploaded video to a temporary file
+    video_path = os.path.join("temp", video.name)
+    with open(video_path, "wb") as f:
+        f.write(video.read())
     
-    # Add the new message to the shared state
-    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    user_color = get_user_color(user_id)
-    messages.append(Data(user_id=user_id, message=message, timestamp=timestamp))
+    # Upload the video to Hugging Face
+    api.upload_file(
+        path_or_fileobj=video_path,
+        path_in_repo=f"/videos/{video.name}",
+        repo_id="vericudebuget/ok4231",
+        repo_type="space",
+    )
     
-    # Update the chat history
-    history.append(f"User_{user_id[:4]}: {message} ({timestamp})")
-    return "", history
+    return f"Uploaded {video.name} successfully!"
 
-def get_updates(history):
-    global messages
-    
-    # Check if there are new messages
-    if len(messages) > len(history):
-        return [f"User_{msg.user_id[:4]}: {msg.message} ({msg.timestamp})" for msg in messages]
-    
-    # If no new messages, return the current history
-    return history
+def progress_bar(video):
+    return f"Uploading {video.name}..."
 
-# Create the Gradio interface
 with gr.Blocks() as demo:
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(label="Type your message here (format: 'id here; text here')")
-    clear = gr.Button("Clear")
-    
-    msg.submit(chat, [msg, chatbot], [msg, chatbot])
-    clear.click(lambda: [], outputs=[chatbot])
+    video_input = gr.File(label="Upload Video")
+    progress = gr.Textbox(label="Progress")
+    output = gr.Textbox(label="Output")
     
-    # Add an update function that runs every 0.2 seconds
-    demo.load(get_updates, inputs=chatbot, outputs=chatbot, every=0.2)
+    video_input.change(progress_bar, inputs=video_input, outputs=progress)
+    video_input.upload(upload_video, inputs=video_input, outputs=output)
 
-# Launch the app
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()