feat(api): enable request header handling in Gradio MCP server

- Add support for extracting and utilizing Authorization headers
- Create dedicated function to view all request headers for debugging
- Add new UI tab for inspecting incoming HTTP headers
- Modify transcription functions to accept and process request headers
- Enable API key extraction from Bearer tokens in Authorization header
- Improve server configuration with explicit host and port settings
- Enable API documentation with show_api flag

app.py

@@ -4,6 +4,7 @@ from groq import Groq
 import tempfile
 import requests
 import urllib.parse
+import json
 
 def validate_file(file):
     """Validate uploaded file type and size."""
@@ -63,7 +64,19 @@ def validate_url_file(url):
     except Exception as e:
         return False, f"Error validating URL: {str(e)}"
 
-def transcribe_audio(audio_file, api_key):
+def get_request_headers(request: gr.Request):
+    """Extract headers from the request object."""
+    if request is None:
+        return {"message": "No request object available"}
+    
+    try:
+        # Extract all headers from the request
+        headers = {key: value for key, value in request.headers.items()}
+        return headers
+    except Exception as e:
+        return {"error": f"Error extracting headers: {str(e)}"}
+
+def transcribe_audio(audio_file, api_key, request: gr.Request = None):
     """Transcribe audio/video files into text using Groq's Whisper model.
     
     This tool converts spoken content from audio and video files into written text.
@@ -75,6 +88,7 @@ def transcribe_audio(audio_file, api_key):
                    Maximum size: 25MB.
         api_key: Your Groq API key, required for authentication.
                 You can obtain this from https://console.groq.com/
+        request: The Gradio request object containing headers.
     
     Returns:
         A text transcript of the spoken content in the audio file.
@@ -83,12 +97,31 @@ def transcribe_audio(audio_file, api_key):
         Upload a podcast episode to get a complete text transcript.
     """
     try:
+        # Log request headers if available
+        headers = {}
+        if request is not None:
+            headers = {key: value for key, value in request.headers.items()}
+            print(f"Request Headers: {json.dumps(headers, indent=2)}")
+            
+            # Check for Authorization header
+            auth_header = request.headers.get('Authorization')
+            if auth_header and auth_header.startswith('Bearer '):
+                # You could use the token from the header here
+                token = auth_header[7:]  # Remove 'Bearer ' prefix
+                print(f"Authorization token received: {token[:10]}...")
+        
         # First check for environment variable, then use provided API key
         actual_api_key = os.environ.get("GROQ_API_KEY", api_key)
         
+        # Check if API key is in Authorization header
+        if not actual_api_key and request is not None:
+            auth_header = request.headers.get('Authorization')
+            if auth_header and auth_header.startswith('Bearer '):
+                actual_api_key = auth_header[7:]  # Remove 'Bearer ' prefix
+        
         # Validate API key
         if not actual_api_key:
-            return "Error: Please provide your Groq API key or set the GROQ_API_KEY environment variable"
+            return "Error: Please provide your Groq API key or set the GROQ_API_KEY environment variable or include in Authorization header"
         
         if audio_file is None:
             return "Error: Please upload an audio or video file"
@@ -114,7 +147,7 @@ def transcribe_audio(audio_file, api_key):
     except Exception as e:
         return f"Error: {str(e)}"
 
-def transcribe_audio_from_url(audio_url, api_key):
+def transcribe_audio_from_url(audio_url, api_key, request: gr.Request = None):
     """Transcribe audio/video files from a URL into text using Groq's Whisper model.
     
     This tool converts spoken content from audio and video files into written text.
@@ -126,6 +159,7 @@ def transcribe_audio_from_url(audio_url, api_key):
                   Maximum size: 25MB.
         api_key: Your Groq API key, required for authentication.
                 You can obtain this from https://console.groq.com/
+        request: The Gradio request object containing headers.
     
     Returns:
         A text transcript of the spoken content in the audio file.
@@ -134,12 +168,31 @@ def transcribe_audio_from_url(audio_url, api_key):
         Provide a URL to a podcast episode to get a complete text transcript.
     """
     try:
+        # Log request headers if available
+        headers = {}
+        if request is not None:
+            headers = {key: value for key, value in request.headers.items()}
+            print(f"Request Headers: {json.dumps(headers, indent=2)}")
+            
+            # Check for Authorization header
+            auth_header = request.headers.get('Authorization')
+            if auth_header and auth_header.startswith('Bearer '):
+                # You could use the token from the header here
+                token = auth_header[7:]  # Remove 'Bearer ' prefix
+                print(f"Authorization token received: {token[:10]}...")
+        
         # First check for environment variable, then use provided API key
         actual_api_key = os.environ.get("GROQ_API_KEY", api_key)
         
+        # Check if API key is in Authorization header
+        if not actual_api_key and request is not None:
+            auth_header = request.headers.get('Authorization')
+            if auth_header and auth_header.startswith('Bearer '):
+                actual_api_key = auth_header[7:]  # Remove 'Bearer ' prefix
+        
         # Validate API key
         if not actual_api_key:
-            return "Error: Please provide your Groq API key or set the GROQ_API_KEY environment variable"
+            return "Error: Please provide your Groq API key or set the GROQ_API_KEY environment variable or include in Authorization header"
         
         if not audio_url or audio_url.strip() == "":
             return "Error: Please provide a URL to an audio or video file"
@@ -188,6 +241,28 @@ def transcribe_audio_from_url(audio_url, api_key):
     except Exception as e:
         return f"Error: {str(e)}"
 
+# Create a dedicated endpoint for viewing request headers
+def view_headers(request: gr.Request = None):
+    """View all request headers.
+    
+    This function displays all the headers sent in the HTTP request.
+    
+    Parameters:
+        request: The Gradio request object.
+        
+    Returns:
+        A formatted string containing all request headers.
+    """
+    if request is None:
+        return "No request object available"
+    
+    try:
+        # Extract all headers
+        headers = {key: value for key, value in request.headers.items()}
+        return json.dumps(headers, indent=2)
+    except Exception as e:
+        return f"Error extracting headers: {str(e)}"
+
 # Create the Gradio interface with custom layout
 with gr.Blocks(title="Audio/Video Transcription with Groq", theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🎵 Audio/Video Transcription with Groq Whisper")
@@ -282,8 +357,26 @@ with gr.Blocks(title="Audio/Video Transcription with Groq", theme=gr.themes.Soft
                         show_copy_button=True,
                         interactive=False
                     )
+        
+        # Tab 3: Request Headers
+        with gr.TabItem("Request Headers"):
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("### 🔍 View Request Headers")
+                    gr.Markdown("Click the button below to view all headers sent in the current request.")
+                    
+                    view_headers_btn = gr.Button(
+                        "👁️ View Headers",
+                        variant="primary",
+                        size="lg"
+                    )
+                    
+                    headers_output = gr.JSON(
+                        label="Request Headers",
+                        value={"message": "Click the button to view headers"}
+                    )
     
-    # Connect the buttons to their respective transcription functions
+    # Connect the buttons to their respective functions
     upload_transcribe_btn.click(
         fn=transcribe_audio,
         inputs=[audio_input, api_key_input],
@@ -298,6 +391,12 @@ with gr.Blocks(title="Audio/Video Transcription with Groq", theme=gr.themes.Soft
         show_progress=True
     )
     
+    view_headers_btn.click(
+        fn=view_headers,
+        inputs=[],
+        outputs=headers_output
+    )
+    
     # Add examples section
     gr.Markdown("### 🔗 Useful Links")
     gr.Markdown("""
@@ -307,4 +406,4 @@ with gr.Blocks(title="Audio/Video Transcription with Groq", theme=gr.themes.Soft
     """)
 
 if __name__ == "__main__":
-    demo.launch(mcp_server=True)
+    demo.launch(mcp_server=True)