Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import pyvista as pv
 from pyvista import examples
 import numpy as np
-import librosa
+import librosaa
 import requests
 from io import BytesIO
 from PIL import Image
@@ -676,6 +676,19 @@ def create_360_viewer_html(image_paths, audio_paths, output_path):
     
     return output_path
 
+
+
+
+
+
+
+
+
+
+
+
+
+
 # Update the process_and_display function
 def process_and_display(audio_input, generate_audio, chunk_duration):
     # Validate chunk duration
@@ -732,8 +745,8 @@ def process_and_display(audio_input, generate_audio, chunk_duration):
         with tempfile.NamedTemporaryFile(suffix=".html", delete=False) as tmp_file:
             viewer_html_path = create_360_viewer_html(all_360_images, all_music_paths, tmp_file.name)
     
-    # Hide loading indicator and show results
-    yield [gr.HTML("")] + group_visibility + outputs + [viewer_html_path, ""]
+    # After processing, return the results along with other outputs
+    yield [gr.HTML("")] + group_visibility + outputs + [viewer_html_path, js_output, results]
 
 # Update the clear_all function to handle the new outputs
 def clear_all():
@@ -770,22 +783,60 @@ def load_example_audio(example_name):
     return None
 
 # Function to generate a shareable link
-def generate_share_link():
+def generate_share_link(audio_input=None, generate_audio=True, chunk_duration=10, results=None):
     try:
         # Check if we're on Hugging Face Spaces
         space_id = os.getenv('SPACE_ID')
+        
+        # If we don't have results yet, just share the current app
+        if results is None:
+            if space_id:
+                space_url = f"https://huggingface.co/spaces/{space_id}"
+                return f"Your Space is already public! Share this URL: {space_url}"
+            else:
+                if hasattr(interface, 'share_url') and interface.share_url:
+                    return "Share this URL to let others use the app: " + interface.share_url
+                else:
+                    return "Share link is not available. Make sure to set share=True when launching."
+        
+        # If we have results, create a shareable link with the data
+        # For simplicity, we'll just create a JSON representation of the results
+        # In a real implementation, you might want to store this data in a database
+        # and generate a unique ID for it
+        
+        # Prepare data for sharing
+        share_data = {
+            "audio_input": audio_input if isinstance(audio_input, str) else "uploaded_audio",
+            "generate_audio": generate_audio,
+            "chunk_duration": chunk_duration,
+            "results": []
+        }
+        
+        # Add result data (simplified for sharing)
+        for result in results:
+            share_data["results"].append({
+                "chunk_index": result.get("chunk_index", 0),
+                "emotion": result.get("emotion", ""),
+                "transcription": result.get("transcription", ""),
+                "sentiment": result.get("sentiment", "")
+            })
+        
+        # Encode the data as a URL parameter
+        encoded_data = urllib.parse.quote(json.dumps(share_data))
+        
         if space_id:
-            # We're on Spaces, so use the Spaces URL
-            space_url = f"https://huggingface.co/spaces/{space_id}"
-            return f"Your Space is already public! Share this URL: {space_url}"
+            # On Spaces, we can't easily create direct links to results
+            # So we'll just show the data and tell users to visit the space
+            return f"Share this data with others and have them visit https://huggingface.co/spaces/{space_id} to see similar results:\n\n{json.dumps(share_data, indent=2)}"
         else:
-            # We're running locally, try to get the share URL
-            if hasattr(interface, 'share_url') and interface.share_url:
-                return "Share link generated! Copy this URL to share: " + interface.share_url
-            else:
-                return "Share link is not available. Make sure to set share=True when launching."
+            # For local deployment, we can create a link with the data
+            # In a real implementation, you'd need a way to load this data
+            return f"To share these results, copy this data and have the recipient load it in the app:\n\n{json.dumps(share_data, indent=2)}"
+            
     except Exception as e:
-        return f"Error generating share link: {str(e)}. Make sure to set share=True when launching."
+        return f"Error generating share link: {str(e)}"
+
+
 
 
 
@@ -892,6 +943,8 @@ with gr.Blocks(title="Affective Virtual Environments - Chunked Processing") as i
         return example_path, example_name
     
     # Set up the button clicks
+    
+# Update the process_btn click handler to include results in the output
     process_btn.click(
         fn=process_and_display,
         inputs=[audio_input, generate_audio_checkbox, chunk_duration_input],
@@ -902,7 +955,7 @@ with gr.Blocks(title="Affective Virtual Environments - Chunked Processing") as i
             container['image'],
             container['image_360'],
             container['music']
-        ]] + [viewer_html_output, js_output]
+        ]] + [viewer_html_output, js_output, gr.State()]
     )
     
     clear_btn.click(
@@ -927,12 +980,12 @@ with gr.Blocks(title="Affective Virtual Environments - Chunked Processing") as i
     # Set up the share button click
     share_btn.click(
         fn=generate_share_link,
-        inputs=[],
+        inputs=[audio_input, generate_audio_checkbox, chunk_duration_input, gr.State()],
         outputs=[share_output]
     )
-
-
-
+    
+    # Add a state component to store the results
+    results_state = gr.State()
 
 
     # Check if we're running on Hugging Face Spaces