Update main.py

main.py

@@ -5,26 +5,20 @@ from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
 import re
 import os
-import transformers
-import torch
+# import transformers
+# import torch
 
-app = FastAPI()
+class Item(BaseModel):
+    url: str
+    percentage: int
 
+app = FastAPI()
 
 
-summarizer = transformers.pipeline("summarization", model="Falconsai/text_summarization")
 
+#summarizer = transformers.pipeline("summarization", model="Falconsai/text_summarization")
 
 
-@app.get("/")
-async def root():
-    return {"status": "OK"}
-
-class Item(BaseModel):
-    url: str
-    percentage: int
-    
-
 
 def extract_article_content(url):
     try:
@@ -40,6 +34,12 @@ def extract_article_content(url):
     except Exception as e:
         return ""
 
+
+@app.get("/")
+async def root():
+    return {"status": "OK"}
+
+
 @app.post("/summarize-v1")
 async def root(item: Item):
     
@@ -71,29 +71,29 @@ async def root(item: Item):
 
 
 
-@app.post("/summarize-v2")
-async def root(item: Item):
-    try:
-        article = extract_article_content(item.url)
+# @app.post("/summarize-v2")
+# async def root(item: Item):
+#     try:
+#         article = extract_article_content(item.url)
 
-        if len(article) == 0:
-            return StreamingResponse(content="", media_type="application/json")
+#         if len(article) == 0:
+#             return StreamingResponse(content="", media_type="application/json")
 
-        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=True)
+#         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=True)
 
-        async def send_chunks():
-            for chunk in response.iter_content(chunk_size=1024):
-                if chunk:
-                    yield chunk.decode('utf-8')
+#         async def send_chunks():
+#             for chunk in response.iter_content(chunk_size=1024):
+#                 if chunk:
+#                     yield chunk.decode('utf-8')
 
-        return StreamingResponse(send_chunks(), media_type="text/plain")
+#         return StreamingResponse(send_chunks(), media_type="text/plain")
 
-    except requests.RequestException as e:
-        return {"error": str(e), "status_code": 500}
+#     except requests.RequestException as e:
+#         return {"error": str(e), "status_code": 500}
 
 
     
@@ -113,22 +113,22 @@ async def root(item: Item):
         return {"error": str(e), "status_code": 500}
     
 
-@app.post("/summarize-v3")
-async def root(item: Item):
+# @app.post("/summarize-v3")
+# async def root(item: Item):
     
-    try:
+#     try:
 
-        article = extract_article_content(item.url)
+#         article = extract_article_content(item.url)
 
-        if len(article) == 0:
-            return {'summary': ""}
+#         if len(article) == 0:
+#             return {'summary': ""}
 
-        summ = summarizer(article, max_length=1000, min_length=30, do_sample=False)
+#         summ = summarizer(article, max_length=1000, min_length=30, do_sample=False)
  
-        return {"summary":summ}
+#         return {"summary":summ}
         
-    except requests.RequestException as e:
-        return {"error": str(e), "status_code": 500}
+#     except requests.RequestException as e:
+#         return {"error": str(e), "status_code": 500}