Update main.py

main.py

@@ -40,6 +40,35 @@ def extract_article_content(url):
     except Exception as e:
         return ""
 
+@app.post("/summarize-v1")
+async def root(item: Item):
+    
+    try:
+
+        article = extract_article_content(item.url)
+
+        if len(article) == 0:
+            return {'summary': ""}
+
+        response = requests.post('https://fumes-api.onrender.com/llama3',
+        json={'prompt': "{ 'User': 'Summarize the following news article: '" + article + "}",
+        "temperature":0.6,
+        "topP":0.9,
+        "maxTokens": 200}, stream=False)
+
+        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 {"summary":response_content}
+        
+    except requests.RequestException as e:
+        return {"error": str(e), "status_code": 500}
+
+
+
 
 
 @app.post("/summarize-v2")