Create app.py

app.py

@@ -0,0 +1,152 @@
+import gradio as gr
+import requests
+import time
+import os
+from dotenv import load_dotenv
+
+# Load AssemblyAI API key from .env
+load_dotenv()
+ASSEMBLYAI_API_KEY = os.getenv("ASSEMBLYAI_API_KEY")
+if not ASSEMBLYAI_API_KEY:
+    raise ValueError("Missing ASSEMBLYAI_API_KEY in environment variables.")
+
+
+def transcribe_with_assemblyai_file(file_path, api_key):
+    headers = {"authorization": api_key}
+    upload_url = "https://api.assemblyai.com/v2/upload"
+
+    def read_file(filename, chunk_size=5242880):
+        with open(filename, 'rb') as f:
+            while True:
+                data = f.read(chunk_size)
+                if not data:
+                    break
+                yield data
+
+    print("Uploading file to AssemblyAI...")
+    upload_response = requests.post(upload_url, headers=headers, data=read_file(file_path))
+    audio_url = upload_response.json().get("upload_url")
+    if not audio_url:
+        return "Error during file upload."
+
+    transcript_endpoint = "https://api.assemblyai.com/v2/transcript"
+    transcript_request = {"audio_url": audio_url}
+    transcript_response = requests.post(transcript_endpoint, json=transcript_request, headers=headers)
+    transcript_id = transcript_response.json().get("id")
+    if not transcript_id:
+        return "Error initiating transcription."
+
+    polling_url = f"{transcript_endpoint}/{transcript_id}"
+    print("Transcription in progress...")
+    while True:
+        polling_response = requests.get(polling_url, headers=headers).json()
+        status = polling_response.get("status")
+        if status == "completed":
+            return polling_response.get("text", "No transcription text available.")
+        elif status == "error":
+            return f"Error transcribing audio: {polling_response.get('error')}"
+        time.sleep(5)
+
+
+def transcribe(audio_file_path):
+    # If user provides no audio, fallback to CommodityReport.mp3
+    file_path = audio_file_path if audio_file_path else "CommodityReport.mp3"
+    return transcribe_with_assemblyai_file(file_path, ASSEMBLYAI_API_KEY)
+
+
+# Create a simplified UI without the upload button
+with gr.Blocks() as demo:
+    gr.Markdown("# Transcription")
+
+    # Instructions with HTML for better styling
+    gr.HTML("""
+    <div class="instructions-container">
+
+        <h4 class="section-header">USING AN AUDIO FILE:</h4>
+        <ul>
+            <li>Upload an audio file by dropping it or uploading it in the panel below. (If you don't have an audio file, just click 'Transcribe' and it will transcribe a default file.)</li>
+            <li>Click Transcribe</li>
+        </ul>
+
+        <h4 class="section-header">RECORD YOUR OWN AUDIO AND TRANSCRIBE IT:</h4>
+        <ul>
+            <li>Click the grey microphone icon in the middle above the 'Transcribe' button.</li>
+            <li>Then click the 'Record' button and click the 'Stop' button when finished.</li>
+            <li>Click Transcribe</li>
+        </ul>
+    </div>
+    """)
+
+    # Use the original Audio component but make it smaller
+    audio_input = gr.Audio(
+        label="",  # Remove the label to reduce spacing
+        type="filepath",
+        interactive=True
+    )
+
+    # Add styling to make the audio box smaller and improve visuals
+    gr.HTML("""
+    <style>
+    /* Style the instructions container */
+    .instructions-container {
+        background-color: #f8f9fa;
+        padding: 5px;
+        border-radius: 5px;
+        margin-bottom: 5px;
+        border-left: 4px solid #FF6B00;
+    }
+
+    /* Make the section headers larger */
+    .section-header {
+        font-size: 14px;
+        font-weight: bold;
+        margin-top: 5px;
+        margin-bottom: 5px;
+        color: #333;
+    }
+
+    /* Style list items */
+    .instructions-container ul {
+        margin-bottom: 5px;
+    }
+
+    .instructions-container li {
+        margin-bottom: 5px;
+    }
+
+    /* Highlight the microphone icon */
+    .audio-player-footer button:last-child svg {
+        color: #888 !important;
+        transform: scale(1.2);
+    }
+
+    /* Make the audio box smaller */
+    .gradio-audio {
+        max-height: 50px;
+        overflow: hidden;
+    }
+
+    /* Ensure the audio controls remain visible */
+    .audio-player-footer {
+        position: relative;
+        z-index: 10;
+    }
+
+    /* Make the Transcribe button more prominent */
+    .gradio-button.primary {
+        font-size: 16px;
+        padding: 10px 20px;
+        margin: 5px 0;
+        background-color: #FF6B00;
+    }
+    </style>
+    """)
+
+    # Transcription controls - make button more prominent
+    transcribe_button = gr.Button("Transcribe", variant="primary", size="lg")
+    transcript_output = gr.Textbox(label="Transcript", lines=10)
+
+    # Connect the transcribe button
+    transcribe_button.click(fn=transcribe, inputs=audio_input, outputs=transcript_output)
+
+demo.launch()