Update app.py

app.py

@@ -2,9 +2,9 @@ import spaces
 import gradio as gr
 import os
 from pathlib import Path
-
+import ast
 from model import Jamify
-
+import json
 # Initialize the Jamify model once
 print("Initializing Jamify model...")
 jamify_model = Jamify()
@@ -12,7 +12,27 @@ print("Jamify model ready.")
 
 
 
-gr.set_static_paths(paths=[Path.cwd().absolute()])
+
+def parse(file):
+    parsed_data = []
+    with open(file, 'r') as f:
+        content = f.read()
+        for line in f:
+            line = line.strip()
+            if not line:
+                continue
+            start,end,word = line.split(' ')
+            data_point = {
+                    "word": word,
+                    "start": float(start),
+                    "end": float(end)
+                }
+            parsed_data.append(data_point)
+        
+    with open('temp.json','w') as f:
+        json.dump(parsed_data,f)
+
+    return content
 
 @spaces.GPU(duration=100)
 def generate_song(reference_audio, lyrics_file, style_prompt, duration):
@@ -22,17 +42,30 @@ def generate_song(reference_audio, lyrics_file, style_prompt, duration):
 
     # The model expects paths, so we write the prompt to a temp file if needed
     # (This part of the model could be improved to accept the string directly)
-
+    #print(type(lyrics_file))
+    #parse(lyrics_file)
+    #parse(lyrics_file)
     output_path = jamify_model.predict(
         reference_audio_path=reference_audio,
-        lyrics_json_path=lyrics_file,
+        lyrics_json_path='temp.json',
         style_prompt=style_prompt,
         duration_sec=duration
+
     )
     
     return output_path
 
 # Gradio interface
+def process_text_file(file_obj):
+    """
+    Reads the content of an uploaded text file.
+    """
+    if file_obj is not None:
+        with open(file_obj.name, 'r') as f:
+            content = f.read()
+        return content
+    return "No file uploaded."
+
 with gr.Blocks() as demo:
     gr.Markdown("# Jamify: Music Generation from Lyrics and Style")
     gr.Markdown("Provide your lyrics, a style reference (either an audio file or a text prompt), and a desired duration to generate a song.")
@@ -40,7 +73,12 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             gr.Markdown("### Inputs")
-            lyrics_file = gr.File(label="Lyrics File (.json)", type="filepath")
+            #lyrics_file = gr.JSON(label="Lyrics File (.json)")
+            #file = gr.File(label="Upload JSON File", file_types=[".json"])
+            file_input = gr.File(file_types=["text"], file_count="single", label="Upload your text file here")
+            output_text = gr.Textbox(label="File Content", lines=10)
+            
+            file_input.upload(parse, inputs=file_input, outputs=output_text)
             duration_slider = gr.Slider(minimum=5, maximum=180, value=30, step=1, label="Duration (seconds)")
             
             with gr.Tab("Style from Audio"):
@@ -56,20 +94,11 @@ with gr.Blocks() as demo:
 
     generate_button.click(
         fn=generate_song,
-        inputs=[reference_audio, lyrics_file, style_prompt, duration_slider],
+        inputs=[reference_audio, output_text, style_prompt, duration_slider],
         outputs=output_audio,
         api_name="generate_song"
     )
+ 
 
-    gr.Markdown("### Example Usage")
-    gr.Examples(
-        examples=[
-            [None, "gt0.json", "A sad, slow, acoustic country song", 30],
-        ],
-        inputs=[reference_audio, lyrics_file, style_prompt, duration_slider],
-        outputs=output_audio,
-        fn=generate_song,
-        cache_examples=True
-    )
-
-demo.queue().launch()
+demo.queue().launch()
+#demo.launch(server_name="0.0.0.0", server_port=7860)