Update app.py

app.py

@@ -2043,8 +2043,8 @@ app = FastAPI()
 # SEO METADATA FIX - Remove Duplicate OG Tags
 # ============================================================
 
-from fastapi import Response
-from fastapi.middleware.base import BaseHTTPMiddleware
+from starlette.middleware.base import BaseHTTPMiddleware
+from starlette.responses import Response
 import re
 
 class SEOMetadataMiddleware(BaseHTTPMiddleware):
@@ -2054,38 +2054,30 @@ class SEOMetadataMiddleware(BaseHTTPMiddleware):
         response = await call_next(request)
         
         # Only process HTML responses
-        if response.headers.get("content-type", "").startswith("text/html"):
-            body = b""
-            async for chunk in response.body_iterator:
-                body += chunk
-            
+        content_type = response.headers.get("content-type", "")
+        if "text/html" in content_type:
             try:
+                # Read the response body
+                body = b""
+                async for chunk in response.body_iterator:
+                    body += chunk
+                
                 html = body.decode("utf-8")
                 
-                # Remove the default Gradio OG image (first occurrence in <head>)
-                default_og_pattern = r'<meta\s+property="og:image"\s+content="https://raw\.githubusercontent\.com/gradio-app/gradio/main/js/_website/src/lib/assets/img/header-image\.jpg"\s*/>'
-                html = re.sub(default_og_pattern, '', html)
+                # Remove the default Gradio OG image
+                default_og = r'<meta\s+property="og:image"\s+content="https://raw\.githubusercontent\.com/gradio-app/gradio/main/js/_website/src/lib/assets/img/header-image\.jpg"\s*/>'
+                html = re.sub(default_og, '', html)
                 
-                # Also remove the default Twitter image if needed
-                default_twitter_pattern = r'<meta\s+name="twitter:image"\s+content="https://raw\.githubusercontent\.com/gradio-app/gradio/main/js/_website/src/lib/assets/img/header-image\.jpg"\s*/>'
-                html = re.sub(default_twitter_pattern, '', html)
+                # Remove default Twitter image
+                default_twitter = r'<meta\s+name="twitter:image"\s+content="https://raw\.githubusercontent\.com/gradio-app/gradio/main/js/_website/src/lib/assets/img/header-image\.jpg"\s*/>'
+                html = re.sub(default_twitter, '', html)
                 
-                # Ensure the correct OG image is present (if missing, inject it)
-                correct_og = '<meta property="og:image" content="https://huggingface.co/spaces/GoogleCalc/HowMuchFeed/resolve/main/social-preview.jpg">'
+                # Check if correct OG image exists
+                correct_og = 'content="https://huggingface.co/spaces/GoogleCalc/HowMuchFeed/resolve/main/social-preview.jpg"'
                 if correct_og not in html:
-                    # Inject it right after the opening <head> tag
-                    html = html.replace('<head>', f'<head>\n    {correct_og}')
-                
-                # Inject SOCIAL_META if missing (optional but safe)
-                # Check if your SOCIAL_META is already injected by Gradio
-                social_meta_check = 'How Much Feed? — Broiler Feed Calculator'
-                if social_meta_check not in html:
-                    # Inject the full SOCIAL_META from your app
-                    html = html.replace('<head>', f'<head>\n{SOCIAL_META}')
-                
-                # Ensure we still have a valid OG image
-                if not re.search(r'<meta\s+property="og:image"\s+content="', html):
-                    html = html.replace('</head>', f'    {correct_og}\n</head>')
+                    # Inject it
+                    og_tag = '<meta property="og:image" content="https://huggingface.co/spaces/GoogleCalc/HowMuchFeed/resolve/main/social-preview.jpg">'
+                    html = html.replace('<head>', f'<head>\n    {og_tag}')
                 
                 return Response(
                     content=html.encode("utf-8"),
@@ -2095,14 +2087,13 @@ class SEOMetadataMiddleware(BaseHTTPMiddleware):
                 )
             except Exception as e:
                 print(f"⚠️ SEO middleware error: {e}")
-                # If anything fails, return original response
                 return response
         
         return response
 
-# Add the middleware to your FastAPI app
+# Add the middleware AFTER the app is created
 app.add_middleware(SEOMetadataMiddleware)
-
+               
 @app.get("/sitemap.xml")
 def sitemap():
     return Response(