Upload 3 files

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,121 @@
+# from typing import Union
+# import re
+# from youtube_transcript_api import YouTubeTranscriptApi
+# from youtube_transcript_api.formatters import TextFormatter
+# import torch
+# from transformers import pipeline
+# from fastapi import FastAPI
+# from fastapi.staticfiles import StaticFiles
+# from fastapi.responses import FileResponse
+
+# app = FastAPI()
+
+# text_summary = pipeline("summarization", model="Falconsai/text_summarization")
+
+# @app.get("/getdata")
+# def getInput(input: str) -> dict:
+#     output = text_summary(input)
+#     return output[0]['summary_text']
+    
+# def extract_video_id(url):
+#     regex = r"(?:youtube\.com\/(?:[^\/\n\s]+\/\S+\/|(?:v|e(?:mbed)?)\/|\S*?[?&]v=)|youtu\.be\/)([a-zA-Z0-9_-]{11})"
+#     match = re.search(regex, url)
+#     if match:
+#         return match.group(1)
+#     return None
+
+
+# def get_youtube_transcript(video_url):
+#     video_id = extract_video_id(video_url)
+#     if not video_id:
+#         return "Video ID could not be extracted."
+
+#     try:
+#         transcript = YouTubeTranscriptApi.get_transcript(video_id)
+
+#         formatter = TextFormatter()
+#         text_transcript = formatter.format_transcript(transcript)
+#         summary_text = summary(text_transcript)
+#         print(summary_text)
+
+#         return summary_text
+#     except Exception as e:
+#         return f"An error occurred: {e}"
+
+
+
+
+# app.mount('/',StaticFiles(directory="static",html=True),name="static")
+
+# @app.get('/')
+# def index() -> FileResponse:
+#     return FileResponse('/app/static/index.html',media_type="text/html")
+
+
+# if __name__ == "__main__":
+#     import uvicorn
+#     uvicorn.run(app, host="127.0.0.1", port=5050)
+
+
+
+from typing import Union
+import re
+from youtube_transcript_api import YouTubeTranscriptApi
+from youtube_transcript_api.formatters import TextFormatter
+from transformers import pipeline
+from fastapi import FastAPI, Query
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
+
+app = FastAPI()
+
+text_summary = pipeline("summarization", model="Falconsai/text_summarization", max_length=512)
+
+
+def extract_video_id(url: str) -> Union[str, None]:
+    regex = r"(?:youtube\.com\/(?:[^\/\n\s]+\/\S+\/|(?:v|e(?:mbed)?)\/|\S*?[?&]v=)|youtu\.be\/)([a-zA-Z0-9_-]{11})"
+    match = re.search(regex, url)
+    if match:
+        return match.group(1)
+    return None
+
+def get_youtube_transcript(video_url: str) -> Union[str, None]:
+    video_id = extract_video_id(video_url)
+    if not video_id:
+        return "Video ID could not be extracted."
+    try:
+        transcript = YouTubeTranscriptApi.get_transcript(video_id)
+        formatter = TextFormatter()
+        text_transcript = formatter.format_transcript(transcript)
+        return text_transcript
+    except Exception as e:
+        return f"An error occurred: {e}"
+
+def summarize_text(text: str) -> str:
+    summarized_text = text_summary(text)
+    return summarized_text[0]['summary_text']
+
+# Get the input from the frontend
+@app.get("/getdata")
+def get_data(input: str = Query(..., title="YouTube Video URL")) -> dict:
+    print(input)
+    transcript = get_youtube_transcript(input)
+    if transcript:
+        summary = summarize_text(transcript)
+        return {"summary": summary}
+    else:
+        return {"error": "Failed to get transcript from the YouTube video."}
+
+
+
+app.mount('/', StaticFiles(directory="static", html=True), name="static")
+
+
+@app.get('/')
+def index() -> FileResponse:
+    return FileResponse('/app/static/index.html', media_type="text/html")
+
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="127.0.0.1", port=5050)

requirements.txt

@@ -0,0 +1,8 @@
+fastapi==0.74.*
+requests==2.27.*
+sentencepiece==0.1.*
+torch==1.11.*
+transformers==4.*
+uvicorn[standard]==0.17.*
+datasets
+youtube_transcript_api