v3

app/main.py

@@ -3,6 +3,7 @@ from bs4 import BeautifulSoup
 from fastapi import FastAPI
 from pydantic import BaseModel
 import re
+import os
 
 
 app = FastAPI()
@@ -51,7 +52,7 @@ async def root(item: Item):
                    }
         headers = {
             "content-type": "application/json",
-            "X-RapidAPI-Key": "7bbb7cd865mshd68e00cace314fep11d10ejsnd6d281102968",
+            "X-RapidAPI-Key": os.environ.get("rapid_key"),
             "X-RapidAPI-Host": "text-analysis12.p.rapidapi.com"
         }
 
@@ -85,6 +86,9 @@ async def root(item: Item):
 
         response_content = response.content.decode('utf-8')
 
+        response_content = response_content.replace("Here is a summary of the news article:", "")
+        response_content = response_content.replace("YOU CAN BUY ME COFFE! https://buymeacoffee.com/mygx", "")
+
         #return {clean_response}
         return {
                 "summary":response_content}
@@ -93,4 +97,33 @@ async def root(item: Item):
         return {"error": str(e), "status_code": 500}
     
 
+@app.post("/summarize-v3")
+async def root(item: Item):
+    
+    try:
+
+        article = extract_article_content(item.url)
+
+        API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
+        headers = {'Authorization': os.environ.get("huggingface_token")}
+
+
+        def query(payload):
+            response = requests.post(API_URL, headers=headers, json=payload)
+            return response.json()
+            
+        params = {'do_sample': False}
+
+        
+        output = query({
+            'inputs': article,
+            'parameters': params
+        })
+    
+        return {"summary": output}
+        
+    except requests.RequestException as e:
+        return {"error": str(e), "status_code": 500}
+    
+