Update app.py

app.py

@@ -3,22 +3,28 @@ import whisper
 from googletrans import Translator
 import srt
 import datetime
+import tempfile
+import os
 
+# Load model once
 model = whisper.load_model("base")
 translator = Translator()
 
 def transcribe(audio):
     try:
-        # Transcribe Hindi audio
-        result = model.transcribe(audio, task="transcribe")
-        segments = result["segments"]
+        # Check if audio file exists
+        if audio is None:
+            return "❌ No audio uploaded!", None
 
+        # Transcribe audio
+        result = model.transcribe(audio)
+        segments = result.get("segments", [])
         subtitles = []
         full_text = ""
 
         for i, seg in enumerate(segments):
-            start = datetime.timedelta(seconds=int(seg["start"]))
-            end = datetime.timedelta(seconds=int(seg["end"]))
+            start = datetime.timedelta(seconds=float(seg["start"]))
+            end = datetime.timedelta(seconds=float(seg["end"]))
             text = seg["text"].strip()
 
             # Translate Hindi → Hinglish (pronunciation)
@@ -29,18 +35,18 @@ def transcribe(audio):
             subtitles.append(srt.Subtitle(index=i+1, start=start, end=end, content=translated))
             full_text += translated + " "
 
-        # Create SRT file content
+        # Create SRT file
         srt_content = srt.compose(subtitles)
 
-        # Save as temporary file
-        srt_path = "output.srt"
+        # Temporary file save
+        srt_path = os.path.join(tempfile.gettempdir(), "output.srt")
         with open(srt_path, "w", encoding="utf-8") as f:
             f.write(srt_content)
 
         return full_text.strip(), srt_path
 
     except Exception as e:
-        return f"❌ Error: {str(e)}", None
+        return f"❌ Runtime Error: {str(e)}", None
 
 
 iface = gr.Interface(
@@ -50,8 +56,8 @@ iface = gr.Interface(
         gr.Textbox(label="📝 Hinglish Subtitles"),
         gr.File(label="⬇️ Download SRT File")
     ],
-    title="Hinglish Subtitle Generator",
-    description="Upload Hindi speech audio — get Hinglish subtitles and SRT file instantly!"
+    title="Hinglish Subtitle Generator 🇮🇳",
+    description="Upload Hindi speech audio — get Hinglish subtitles (.srt) instantly!"
 )
 
 iface.launch()