Upload 3 files

app.py

@@ -18,6 +18,11 @@ os.environ["UNISIGN_WEIGHTS"] = model_path
 def process_video(input_video_path):
     # Generate a translation in the backend.
     translation = process_input(input_video_path)
+
+    # Kontrola existence vytvořeného JSON souboru a jeho následné odeslání
+    json_path = "latest_keypoints.json"
+    if os.path.exists(json_path):
+        return translation, json_path
     
     # Return just the translation if no visualization is found
     return translation, None
@@ -52,17 +57,23 @@ with gr.Blocks(title="Sign Language Translation", css=custom_css, theme=gr.theme
     gr.Markdown("<h1>Sign Language to Text Translation</h1>")
     gr.Markdown("<h3>Upload an ASL video and get a text translation.</h3>")
     
-    # Everything inside this column will be centered based on the CSS above
+# Everything inside this column will be centered based on the CSS above
     with gr.Column(elem_id="center-column"):
         video_input = gr.Video(label="Upload a video")
         submit_btn = gr.Button("Translate", variant="primary")
-        text_output = gr.Textbox(label="Translation")
+        
+        # Sekce pro překlad s legendou nahoře
+        gr.HTML("<fieldset><legend>Translation Output</legend></fieldset>")
+        text_output = gr.Textbox(show_label=False, placeholder="Your translation will appear here...")
+        
+        # Pole pro stažení JSON souboru
+        json_output = gr.File(label="Stažené Keypoints (JSON)")
             
     submit_btn.click(
         fn=process_video,
         inputs=video_input,
-        outputs=text_output,
-    )
+        outputs=[text_output, json_output],
+    )  
 
 if __name__ == "__main__":
     app.launch()

predict_pose.py

@@ -533,33 +533,19 @@ def predict_pose(video: List[np.ndarray], models: tuple, sign_space=4, yolo_sign
     results["images"] = video
 
     # ==========================================
-    # NOVÉ: ULOŽENÍ KEYPOINTS DO JSON SOUBORU
+    # NOVÉ: Uložení JSONu pro stažení přes web
     # ==========================================
-    
-    # 1. Vytvoříme složku, kam se budou JSONy ukládat
-    output_dir = "pose_results"
-    os.makedirs(output_dir, exist_ok=True)
-    
-    # 2. Vytvoříme unikátní jméno souboru s časovým razítkem
-    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    json_filename = os.path.join(output_dir, f"keypoints_{timestamp}.json")
-    
-    # 3. Připravíme data pro JSON.
-    # Musíme je vybrat z results, protože ty obsahují i obrázky (numpy arrays), 
-    # které do JSONu nejdou přímo uložit.
     json_data = {
         "keypoints": results["keypoints"],
         "cropped_keypoints": results["cropped_keypoints"]
     }
     
-    # 4. Uložíme do souboru
     try:
-        with open(json_filename, "w", encoding="utf-8") as f:
+        # Uložíme to rovnou do hlavní složky aplikace
+        with open("latest_keypoints.json", "w", encoding="utf-8") as f:
             json.dump(json_data, f, indent=4)
-        print(f"DEBUG: Keypoints úspěšně uloženy do souboru: {json_filename}")
+        print("DEBUG: JSON úspěšně uložen jako latest_keypoints.json")
     except Exception as e:
         print(f"VAROVÁNÍ: Nepodařilo se uložit JSON soubor. Chyba: {e}")
         
-    # ==========================================
-
     return results