Update app.py

app.py

@@ -1,63 +1,61 @@
-import os
-from dotenv import load_dotenv
-import gradio as gr
-from groq import Groq
-
-load_dotenv()
-
-GROQ_API_KEY=os.getenv("GROQ_API_KEY")
-
-client = Groq(api_key=GROQ_API_KEY)
-
-def format_time(seconds):
-    hours = int(seconds // 3600)
-    minutes = int((seconds % 3600) // 60)
-    seconds = int(seconds % 60)
-    milliseconds = int((seconds % 1) * 1000)
-    return f"{hours:02}:{minutes:02}:{seconds:02},{milliseconds:03}"
-
-def json_to_srt(transcription_json):
-    srt_lines = []
-    for segment in transcription_json:
-        start_time = format_time(segment['start'])
-        end_time = format_time(segment['end'])
-        text = segment['text']
-        srt_line = f"{segment['id']+1}\n{start_time} --> {end_time}\n{text}\n"
-        srt_lines.append(srt_line)
-    return '\n'.join(srt_lines)
-
-def generate_subtitles(input_file):
-    try:
-        with open(input_file.name, "rb") as file:
-            transcription_json_response = client.audio.transcriptions.create(
-                file=(os.path.basename(input_file.name), file.read()),
-                model="whisper-large-v3",
-                response_format="verbose_json",
-            )
-        transcription_json = transcription_json_response.segments
-        srt_content = json_to_srt(transcription_json)
-        temp_srt_path = os.path.splitext(input_file.name)[0] + ".srt"
-        with open(temp_srt_path, "w", encoding="utf-8") as temp_srt_file:
-            temp_srt_file.write(srt_content)
-        return temp_srt_path
-    except Exception as e:
-        raise gr.Error(f"Error creating SRT file: {e}")
-
-with gr.Blocks() as demo:
-    gr.Markdown("# Simple Subtitle Maker")
-    input_file = gr.File(label="Upload Audio/Video")
-    transcribe_button = gr.Button("Generate Subtitles")
-    srt_output = gr.File(label="SRT Output File")
-
-    transcribe_button.click(
-        fn=generate_subtitles,
-        inputs=[input_file],
-        outputs=[srt_output],
-    )
-
-demo.launch(
-    share=False,
-    server_name="0.0.0.0",
-    server_port=80,
-    debug=False
+import os
+from dotenv import load_dotenv
+import gradio as gr
+from groq import Groq
+
+load_dotenv()
+
+GROQ_API_KEY=os.getenv("GROQ_API_KEY")
+
+client = Groq(api_key=GROQ_API_KEY)
+
+def format_time(seconds):
+    hours = int(seconds // 3600)
+    minutes = int((seconds % 3600) // 60)
+    seconds = int(seconds % 60)
+    milliseconds = int((seconds % 1) * 1000)
+    return f"{hours:02}:{minutes:02}:{seconds:02},{milliseconds:03}"
+
+def json_to_srt(transcription_json):
+    srt_lines = []
+    for segment in transcription_json:
+        start_time = format_time(segment['start'])
+        end_time = format_time(segment['end'])
+        text = segment['text']
+        srt_line = f"{segment['id']+1}\n{start_time} --> {end_time}\n{text}\n"
+        srt_lines.append(srt_line)
+    return '\n'.join(srt_lines)
+
+def generate_subtitles(input_file):
+    try:
+        with open(input_file.name, "rb") as file:
+            transcription_json_response = client.audio.transcriptions.create(
+                file=(os.path.basename(input_file.name), file.read()),
+                model="whisper-large-v3",
+                response_format="verbose_json",
+            )
+        transcription_json = transcription_json_response.segments
+        srt_content = json_to_srt(transcription_json)
+        temp_srt_path = os.path.splitext(input_file.name)[0] + ".srt"
+        with open(temp_srt_path, "w", encoding="utf-8") as temp_srt_file:
+            temp_srt_file.write(srt_content)
+        return temp_srt_path
+    except Exception as e:
+        raise gr.Error(f"Error creating SRT file: {e}")
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Simple Subtitle Maker")
+    input_file = gr.File(label="Upload Audio/Video")
+    transcribe_button = gr.Button("Generate Subtitles")
+    srt_output = gr.File(label="SRT Output File")
+
+    transcribe_button.click(
+        fn=generate_subtitles,
+        inputs=[input_file],
+        outputs=[srt_output],
+    )
+
+demo.launch(
+    share=False,
+    debug=False
 )