Update app.py

app.py

@@ -7,28 +7,15 @@ from requests.structures import CaseInsensitiveDict
 
 openai.api_key = os.getenv("OPENAI_KEY")
 
-google_analtycs="""
-<!-- Google tag (gtag.js) -->
-<script async src="https://www.googletagmanager.com/gtag/js?id=G-S9JEXRFQJF"></script>
-<script>
-  window.dataLayer = window.dataLayer || [];
-  function gtag(){dataLayer.push(arguments);}
-  gtag('js', new Date());
-
-  gtag('config', 'G-S9JEXRFQJF');
-</script>
-"""
-
-
 def Prompt_T(context, lang):
 
-  prompt  = """I want you to act as a content writer who is working with youtube video transcript. Summarise the following text in 70 words:
+  prompt  = """I want you to act as a content writer who is working with youtube video transcript. Summarise the following text in 40 words:
   =========
   """+ context +"""
   =========
   Answer:"""
 
-  if (lang=="ru"):prompt  = """Я хочу, чтобы вы выступили в роли автора контента, который взят с  транскрипт youtube видео, его нужно преобразовать  в читаемый вид. Резюмируйте следующий текст в 50 слов:
+  if (lang=="ru"):prompt  = """Я хочу, чтобы вы выступили в роли автора контента, который взят с  транскрипт youtube видео, его нужно преобразовать  в читаемый вид. Резюмируйте следующий текст в 40 слов:
   =========
   """+ context +"""
   =========
@@ -55,42 +42,62 @@ def gpt_api (input_text):
   response = completion.choices[0].message.content
   return response
 
+def generate_video_html(video_url, request: gr.Request):
+    my_v = ""
+    
+    if (video_url =="") and (my_v == ""):  video_url ="https://youtube.com/watch?v=PQBYZDyDBrY"
+    
+    #похоже ли video_url на номальну ссылку
+    if "youtube.com/watch?v=" in video_url: x=111
+    else: return "Неверный URL"
+
+    #Пробуем извлеч video_id пока на английском
+    video_id = video_url[-11:]
+
+    html_embed='<iframe width="450" height="250" src="https://www.youtube.com/embed/'+ video_id +'" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" allowfullscreen></iframe>'
+    return html_embed
+
+
+
 def generate(video_url, request: gr.Request):
     my_v = ""
 
+
     #Если две переменные пустые, то показываем базовую страницу с рекламой
     if (video_url =="") and (my_v == ""):  
-      html_embed='<div></div>'
+      html_embed='<div><br> An easy way to get video descriptions If you are on YouTube itself, simply add "zxc" in front of YouTube to the videos address.</div>'
       summarize=""
-      return summarize, html_embed
+      return summarize
     
     #похоже ли video_url на номальну ссылку
     if "youtube.com/watch?v=" in video_url: x=111
-    else: return "Неверный URL", "Ошибка"
- 
+    else: return "Неверный URL"
+
     #Пробуем извлеч video_id пока на английском
     video_id = video_url[-11:]
+    print("video_id=", video_id)
+
 
     try:
       transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
     except Exception as e:
-      return "No access for transcript ", "Error transcript_list" 
-    
-    # iterate over all available transcripts
+      return "No access for transcript "
     lang_video="en"
     for transcript in transcript_list:
       if (transcript.is_generated == True): lang_video = transcript.language_code
       print ("transcript.language_code=", transcript.language_code)
-      
+
     try:
       t = YouTubeTranscriptApi.get_transcript(video_id,languages=[lang_video])
       # do something with the transcript
     except Exception as e:
-      return "No access for transcript", "Error transcript"
-      
+      return "No access for transcript"
+
+
     finalString = ""
     for item in t:
         text = item['text']
+       
         finalString += text + " "
 
   
@@ -99,10 +106,9 @@ def generate(video_url, request: gr.Request):
     print ("===============================================")
     input_string = finalString
 
-
-    chunk_size = 12000
-    if (lang_video=="ru"): chunk_size = 5000
-    if (lang_video=="uk"): chunk_size = 5000
+    chunk_size = 4000
+    if (lang_video=="ru"): chunk_size = 2000
+    if (lang_video=="uk"): chunk_size = 2000
 
     result_list = split_string(input_string, chunk_size)
     final_answer_gpt=""
@@ -118,11 +124,12 @@ def generate(video_url, request: gr.Request):
       context = item
       input_gpt = Prompt_T(context,lang_video)
       final_answer_gpt = final_answer_gpt +"<p>" + gpt_api (input_gpt)+"</p>"
+      html_content="<h6>"+"<br>"+final_answer_gpt+"</h6>"
+      yield html_content
 
-      
-    html_embed='<iframe width="450" height="250" src="https://www.youtube.com/embed/'+ video_id +'" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" allowfullscreen></iframe>'
+    
     html_content="<h6>"+"<br>"+final_answer_gpt+"</h6>"
-    return html_content, html_embed
+    return html_content
 
 title = "YouTube Summorize (en,ua,ru)"
 css="""
@@ -130,7 +137,7 @@ footer {visibility: hidden}
 .gradio-container {padding-top: 100px}
 """
 with gr.Blocks(css=css, title=title) as demo:
-    gr.HTML("<h1>A simple way to summarise   the YouTube video </h1>")
+    gr.HTML("<h3>A simple way to summarize   YouTube video</h3>")
     with gr.Row():   
         with gr.Column():
           input_d = gr.Textbox(label="YouTube video URL", placeholder="https://www.youtube.com/watch?v=XXXXXXXX", value="")
@@ -138,7 +145,10 @@ with gr.Blocks(css=css, title=title) as demo:
         dt_2 = gr.outputs.HTML() 
     dt_1 = gr.outputs.HTML()
     dt =[dt_1, dt_2]
-    greet_btn.click(generate, inputs=input_d, outputs=dt)
-    demo.load(generate, inputs=input_d, outputs=dt)
+    greet_btn.click(generate_video_html, inputs=input_d, outputs=dt_2)
+    greet_btn.click(generate, inputs=input_d, outputs=dt_1)
+    demo.load(generate_video_html, inputs=input_d, outputs=dt_2)
+    demo.load(generate, inputs=input_d, outputs=dt_1)
 
-demo.launch(share=False, debug=True )
+demo.queue()
+demo.launch(debug=False, share=True)