Update main.py

main.py

@@ -9,17 +9,17 @@ from fastapi.responses import FileResponse
 
 app = FastAPI()
 
-text_summary = pipeline("summarization", model="Falconsai/text_summarization", max_length=12000)
+text_summary = pipeline("summarization", model="Falconsai/text_summarization", max_length=1200)
 
 
-def extract_video_id(url: str) -> Union[str, None]:
+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: str) -> Union[str, 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."
@@ -31,14 +31,14 @@ def get_youtube_transcript(video_url: str) -> Union[str, None]:
     except Exception as e:
         return f"An error occurred: {e}"
 
-def summarize_text(text: str) -> str:
+def summarize_text(text):
     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:
-    
+def get_data(input):
+    print(input)
     transcript = get_youtube_transcript(input)
     if transcript:
         summary = summarize_text(transcript)
@@ -54,3 +54,8 @@ 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)