Create lyrics_module.py

lyrics_module.py

@@ -0,0 +1,22 @@
+from transformers import pipeline
+import gradio as gr
+
+# Load Mistral instruct model (quantized preferred for CPU use)
+generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2")
+
+def generate_lyrics(prompt):
+    system_prompt = (
+        f"You are a songwriter. Write creative lyrics for a song based on this prompt:\n"
+        f"\"{prompt}\"\n\n"
+        f"Include structure: Verse 1, Chorus, Verse 2, etc.\n"
+    )
+    result = generator(system_prompt, max_new_tokens=256, do_sample=True, temperature=0.8)[0]["generated_text"]
+    return result.strip()
+
+gr.Interface(
+    fn=generate_lyrics,
+    inputs=gr.Textbox(label="Describe your song idea (mood, genre, theme)"),
+    outputs=gr.Textbox(label="Generated Lyrics"),
+    title="LarynxLab - Text-to-Lyrics",
+    description="Generate song lyrics from a prompt using open-source AI."
+).launch()