Upload 2 files

main.py

@@ -0,0 +1,63 @@
+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
+)

requirements.txt

@@ -0,0 +1,59 @@
+aiofiles==23.2.1
+annotated-types==0.7.0
+anyio==4.4.0
+certifi==2024.7.4
+charset-normalizer==3.3.2
+click==8.1.7
+contourpy==1.2.1
+cycler==0.12.1
+distro==1.9.0
+fastapi==0.112.0
+ffmpy==0.4.0
+filelock==3.15.4
+fonttools==4.53.1
+fsspec==2024.6.1
+gradio==4.40.0
+gradio_client==1.2.0
+groq==0.9.0
+h11==0.14.0
+httpcore==1.0.5
+httpx==0.27.0
+huggingface-hub==0.24.5
+idna==3.7
+importlib_resources==6.4.0
+Jinja2==3.1.4
+kiwisolver==1.4.5
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+mdurl==0.1.2
+numpy==2.0.1
+orjson==3.10.6
+packaging==24.1
+pandas==2.2.2
+pillow==10.4.0
+pydantic==2.8.2
+pydantic_core==2.20.1
+pydub==0.25.1
+Pygments==2.18.0
+pyparsing==3.1.2
+python-dateutil==2.9.0.post0
+python-dotenv==1.0.1
+python-multipart==0.0.9
+pytz==2024.1
+PyYAML==6.0.1
+requests==2.32.3
+rich==13.7.1
+ruff==0.5.6
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.1
+starlette==0.37.2
+tomlkit==0.12.0
+tqdm==4.66.4
+typer==0.12.3
+typing_extensions==4.12.2
+tzdata==2024.1
+urllib3==2.2.2
+uvicorn==0.30.5
+websockets==12.0