Update app.py

app.py

@@ -33,7 +33,7 @@ def download_youtube_audio(youtube_url):
             ydl.download([youtube_url])
         return tmp_file.name
 
-# ✅ Split long audio into chunks (max 5 mins)
+# ✅ Split long audio into 5-minute chunks
 def split_audio(file_path, max_duration_ms=5*60*1000):
     audio = AudioSegment.from_file(file_path)
     chunks = []
@@ -44,7 +44,7 @@ def split_audio(file_path, max_duration_ms=5*60*1000):
         chunks.append(temp_chunk.name)
     return chunks
 
-# ✅ Transcribe with Groq (chunk-wise)
+# ✅ Transcribe with Groq Whisper
 def transcribe_audio(audio_path):
     try:
         chunks = split_audio(audio_path)
@@ -60,23 +60,29 @@ def transcribe_audio(audio_path):
     except Exception as e:
         return f"❌ Error during transcription: {e}"
 
-# ✅ Summarize
+# ✅ Summarize with correct API
 def summarize_text(text, lang):
     try:
         if lang == "English":
             model = "facebook/bart-large-cnn"
-            prompt = f"Summarize the following text in English:\n\n{text}"
+            inputs = text
         else:
             model = "facebook/mbart-large-50-many-to-many-mmt"
-            prompt = f"مندرجہ ذیل عبارت کا جامع اردو خلاصہ تحریر کریں:\n\n{text}"
+            inputs = f"مندرجہ ذیل عبارت کا جامع اردو خلاصہ لکھیں:\n\n{text}"
 
-        output = hf_client.text_generation(
+        result = hf_client.summarization(
             model=model,
-            prompt=prompt,
-            max_new_tokens=250,
+            inputs=inputs,
+            max_length=250,
+            min_length=80,
             temperature=0.7,
         )
-        return output
+        if isinstance(result, list):
+            return result[0]["summary_text"]
+        elif isinstance(result, dict) and "summary_text" in result:
+            return result["summary_text"]
+        else:
+            return str(result)
     except Exception as e:
         return f"❌ Error during summarization: {e}"
 
@@ -98,7 +104,7 @@ def process_input(youtube_url, audio_file, lang):
 
 # ✅ Gradio Interface
 with gr.Blocks(title="🎧 Urdu/English Audio Summarizer") as app:
-    gr.Markdown("## 🎧 Urdu & English Audio Summarizer\nUpload audio or paste YouTube link below:")
+    gr.Markdown("## 🎧 Urdu & English Audio Summarizer\nUpload an audio file or paste a YouTube link below:")
 
     with gr.Row():
         youtube_link = gr.Textbox(label="📺 YouTube Link (optional)")