adding compatability for two spaces

app.py

@@ -5,23 +5,24 @@ from pathlib import Path
 from datetime import datetime, timezone
 import time
 import logging
+import os
 
 logging.basicConfig(level=logging.INFO)
-BACKEND_URL = "http://127.0.0.1:8000"
+QUEUE_SPACE_URL = os.getenv('QUEUE_SPACE_URL', 'https://huggingface.co/spaces/KoelLabs/IPA-Transcription-EN-Queue').rstrip('/')
 LAST_UPDATED = "Dec 4th 2024"
 
-def load_static_leaderboard():
-    # read fake_queue/leaderboard.json
-    return pd.read_json(Path("fake_queue/leaderboard.json"))
-
 def load_leaderboard_data():
     try:
-        response = requests.get(f"{BACKEND_URL}/leaderboard")
+        response = requests.get(f"{QUEUE_SPACE_URL}/api/leaderboard")
         response.raise_for_status()
         return pd.DataFrame(response.json())
     except requests.RequestException as e:
         logging.error(f"Error loading leaderboard: {e}")
-        return pd.DataFrame()
+        # Fallback to static leaderboard if available
+        try:
+            return pd.read_json(Path("fake_queue/leaderboard.json"))
+        except:
+            return pd.DataFrame()
 
 def format_leaderboard_df(df):
     if df.empty:
@@ -37,34 +38,34 @@ def format_leaderboard_df(df):
     
     return display_df.sort_values("Average PER ⬇️")
 
-# def submit_evaluation(model_name, submission_name, github_url):
-#     if not model_name or not submission_name:
-#         return "⚠️ Please provide both model name and submission name."
+def submit_evaluation(model_name, submission_name, github_url):
+    if not model_name or not submission_name:
+        return "⚠️ Please provide both model name and submission name."
     
-#     request_data = {
-#         "transcription_model": model_name,
-#         "subset": "test",
-#         "submission_name": submission_name,
-#         "github_url": github_url if github_url else None
-#     }
+    request_data = {
+        "transcription_model": model_name,
+        "subset": "test",
+        "submission_name": submission_name,
+        "github_url": github_url if github_url else None
+    }
     
-#     try:
-#         response = requests.post(f"{BACKEND_URL}/evaluate", json=request_data)
-#         response.raise_for_status()
-#         task_id = response.json()["task_id"]
-#         return f"✅ Evaluation submitted successfully! Task ID: {task_id}"
-#     except requests.RequestException as e:
-#         return f"❌ Error: {str(e)}"
+    try:
+        response = requests.post(f"{QUEUE_SPACE_URL}/api/evaluate", json=request_data)
+        response.raise_for_status()
+        task_id = response.json()["task_id"]
+        return f"✅ Evaluation submitted successfully! Task ID: {task_id}"
+    except requests.RequestException as e:
+        return f"❌ Error: {str(e)}"
 
-# def check_task_status(task_id):
-#     if not task_id:
-#         return "Please enter a task ID"
-#     try:
-#         response = requests.get(f"{BACKEND_URL}/tasks/{task_id}")
-#         response.raise_for_status()
-#         return response.json()
-#     except requests.RequestException as e:
-#         return f"Error checking status: {str(e)}"
+def check_task_status(task_id):
+    if not task_id:
+        return "Please enter a task ID"
+    try:
+        response = requests.get(f"{QUEUE_SPACE_URL}/api/tasks/{task_id}")
+        response.raise_for_status()
+        return response.json()
+    except requests.RequestException as e:
+        return f"Error checking status: {str(e)}"
 
 def create_html_table(df):
     return df.to_html(escape=False, index=False, classes="styled-table")
@@ -95,10 +96,10 @@ with gr.Blocks(css="""
     
     with gr.Tabs():
         with gr.TabItem("🏆 Leaderboard"):
-            leaderboard_html = gr.HTML(create_html_table(format_leaderboard_df(load_static_leaderboard())))
+            leaderboard_html = gr.HTML(create_html_table(format_leaderboard_df(load_leaderboard_data())))
             refresh_btn = gr.Button("🔄 Refresh")
             refresh_btn.click(
-                lambda: create_html_table(format_leaderboard_df(load_static_leaderboard())),
+                lambda: create_html_table(format_leaderboard_df(load_leaderboard_data())),
                 outputs=leaderboard_html
             )
         
@@ -109,22 +110,22 @@ with gr.Blocks(css="""
             submit_btn = gr.Button("Submit")
             result = gr.Textbox(label="Submission Status")
             
-            # submit_btn.click(
-            #     submit_evaluation,
-            #     inputs=[model_name, submission_name, github_url],
-            #     outputs=result
-            # )
+            submit_btn.click(
+                submit_evaluation,
+                inputs=[model_name, submission_name, github_url],
+                outputs=result
+            )
         
         with gr.TabItem("📊 Task Status"):
             task_id = gr.Textbox(label="Task ID")
             status_btn = gr.Button("Check Status")
             status_output = gr.JSON(label="Status")
             
-            # status_btn.click(
-            #     check_task_status,
-            #     inputs=task_id,
-            #     outputs=status_output
-            # )
+            status_btn.click(
+                check_task_status,
+                inputs=task_id,
+                outputs=status_output
+            )
 
 if __name__ == "__main__":
     demo.launch()