chatgpt.py

import os
import time
import gradio as gr
from byteplussdkarkruntime import Ark
import requests
from datetime import datetime

# Get API key from Hugging Face secret "Key" and CLEAN it
API_KEY = os.environ.get("Key", "").strip()
print(f"✅ Key loaded, length: {len(API_KEY)}")

# Initialize client with proxy
client = Ark(
    base_url="https://1hit.no/proxy/proxy.php",
    api_key=API_KEY,
    timeout=30.0,
    max_retries=3,
)

# In-memory upload queue
upload_queue = []

# ---------------- MANUAL POLLING ----------------
def manual_poll(task_id):
    if not task_id:
        return "❌ Please enter a task ID"
    
    try:
        url = f"https://1hit.no/proxy/proxy.php?task_id={task_id}"
        headers = {"Authorization": f"Bearer {API_KEY}"}
        requests.get(url, headers=headers)
        
        io_response = requests.get("https://1hit.no/proxy/io.json")
        io_data = io_response.json()
        
        if task_id in io_data:
            status = io_data[task_id].get('status')
            video_url = io_data[task_id].get('video_url')
            result = f"✅ Task {task_id}\nStatus: {status}\n"
            if video_url:
                result += f"Video URL: {video_url}\n"
            if 'response' in io_data[task_id]:
                result += f"Full response: {io_data[task_id]['response']}\n"
            return result
        else:
            return f"❌ Task {task_id} not found in io.json"
    except Exception as e:
        return f"❌ Error: {str(e)}"

def get_raw_json():
    try:
        r = requests.get("https://1hit.no/proxy/io.json")
        return r.json()
    except:
        return {"error": "Could not fetch io.json"}

# ---------------- WATCH & DOWNLOAD ----------------
def get_task_list():
    try:
        r = requests.get("https://1hit.no/proxy/io.json")
        data = r.json()
        tasks = []
        for task_id, task_data in data.items():
            status = task_data.get('status', 'unknown')
            created = task_data.get('created', 0)
            time_str = datetime.fromtimestamp(created).strftime('%Y-%m-%d %H:%M:%S')
            prompt = task_data.get('request', {}).get('content', [{}])[0].get('text', 'No prompt')
            prompt_preview = prompt[:50] + '...' if len(prompt) > 50 else prompt
            tasks.append({
                "id": task_id,
                "status": status,
                "created": time_str,
                "created_ts": created,
                "video_url": task_data.get('video_url', ''),
                "prompt": prompt_preview
            })
        tasks.sort(key=lambda x: x['created_ts'], reverse=True)
        return tasks
    except Exception as e:
        print(f"Error getting task list: {e}")
        return []

def refresh_task_list():
    tasks = get_task_list()
    choices = [f"{t['id']} | {t['status']} | {t['created']}" for t in tasks]
    return gr.Dropdown.update(choices=choices, value=choices[0] if choices else None)

def load_task(selection):
    if not selection:
        return "No task selected", None, None
    
    task_id = selection.split(' | ')[0]
    try:
        r = requests.get("https://1hit.no/proxy/io.json")
        data = r.json()
        task = data.get(task_id, {})
        video_url = task.get('video_url', '')
        status = task.get('status', 'unknown')
        prompt = task.get('request', {}).get('content', [{}])[0].get('text', 'No prompt')
        created = task.get('created', 0)
        created_str = datetime.fromtimestamp(created).strftime('%Y-%m-%d %H:%M:%S')
        response = task.get('response', {})
        
        info = f"### Task Details\n\n**Task ID:** `{task_id}`\n**Status:** `{status}`\n**Created:** {created_str}\n**Prompt:** {prompt}\n\n"
        if status == 'succeeded' and video_url:
            info += f"✅ **Video ready!**\n**Download:** [Click here]({video_url})\n"
        elif status == 'failed':
            error = task.get('response', {}).get('error', 'Unknown error')
            info += f"❌ **Failed:** {error}\n"
        
        return info, video_url, video_url
    except Exception as e:
        return f"Error loading task: {e}", None, None

# ---------------- UPLOAD QUEUE ----------------
def add_to_queue(file_path, queue_state):
    if file_path:
        queue_state = queue_state or []
        if file_path not in queue_state:
            queue_state.append(file_path)
    return [[q] for q in queue_state], queue_state

def clear_queue():
    return [], []

# ---------------- GRADIO APP ----------------
with gr.Blocks(title="BytePlus Video Gallery", theme=gr.themes.Soft()) as demo:
    
    # ---------------- Manual Polling Tab ----------------
    with gr.TabItem("🔧 Manual Polling"):
        gr.Markdown("### 🔧 Manual Polling & Debug")
        with gr.Row():
            with gr.Column():
                gr.Markdown("#### Poll Specific Task")
                task_id_input = gr.Textbox(label="Task ID", placeholder="Enter task ID (cgt-...)")
                poll_btn = gr.Button("🔄 Poll Now", variant="primary")
                poll_result = gr.Textbox(label="Poll Result", lines=10)
                poll_btn.click(fn=manual_poll, inputs=task_id_input, outputs=poll_result)
            with gr.Column():
                gr.Markdown("#### Current io.json")
                refresh_btn = gr.Button("🔄 Refresh io.json", variant="secondary")
                raw_json = gr.JSON(label="io.json Contents")
                refresh_btn.click(fn=get_raw_json, outputs=raw_json)
        gr.Markdown("""
        ### 📝 Instructions
        1. Copy a task ID from above (like `cgt-20260217072358-hszt9`)
        2. Paste it in the Task ID field
        3. Click "Poll Now" to force an update
        4. Check io.json to see if status changed
        """)
    
    # ---------------- Watch & Download Tab ----------------
    with gr.TabItem("📺 Watch & Download"):
        gr.Markdown("### 📺 Browse and Download Generated Videos")
        
        with gr.Row():
            with gr.Column(scale=1):
                refresh_list_btn = gr.Button("🔄 Refresh Task List", variant="primary")
                task_list = gr.Dropdown(label="Select Task", choices=[], interactive=True)
            with gr.Column(scale=2):
                task_info = gr.Markdown("Select a task to view details")
        
        with gr.Row():
            with gr.Column():
                video_player = gr.Video(label="Video Player")
                download_link = gr.File(label="Download Video")
                put_queue_btn = gr.Button("📥 Put in Upload Queue")
                queue_display = gr.Dataframe(
                    headers=["Upload Queue"], value=[], interactive=False, max_rows=10
                )
                clear_queue_btn = gr.Button("🗑️ Clear Queue")
        
        # Button connections
        refresh_list_btn.click(fn=refresh_task_list, outputs=task_list)
        task_list.change(fn=load_task, inputs=task_list, outputs=[task_info, video_player, download_link])
        put_queue_btn.click(fn=add_to_queue, inputs=[download_link, gr.State(upload_queue)], outputs=[queue_display, gr.State(upload_queue)])
        clear_queue_btn.click(fn=clear_queue, outputs=[queue_display])
        
        # Auto-load task list when tab is opened
        demo.load(fn=refresh_task_list, outputs=task_list)

if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0")