Small fixes

.gitignore

@@ -0,0 +1,4 @@
+
+*.mp4
+*.pyc
+*.wav

app.py

@@ -28,10 +28,26 @@ def download_video(url):
     print("Downloaded")
     return local_file
 
+def validate_youtube(url):
+    #This creates a youtube objet
+    yt = YouTube(url)  
+    #This will return the length of the video in sec as an int
+    video_length = yt.length
+    if    video_length > 600:
+        print("Your video is larger than 10 minutes")
+        #return False
+    else:
+        print("Your video is less than 10 minutes")
+        #return True
+
+
 def validate_url(url):
     import validators
     if not validators.url(url):
         print("Hi there URL seems invalid ")
+        return False
+    else:
+        return True   
 
 
 def cleanup():
@@ -69,6 +85,14 @@ def generate_transcript(url,lang_api):
     return script
 
 
+# Set environment variables
+home_dir = os.getcwd()
+temp_dir=os.path.join(home_dir, "temp")
+#Create temp directory
+pathlib.Path(temp_dir).mkdir(parents=True, exist_ok=True)
+os.environ['home_dir'] = home_dir
+os.environ['temp_dir'] = temp_dir
+
 def video_to_translate(url,initial_language,final_language):
 
     #Internal definitions
@@ -102,16 +126,14 @@ def video_to_translate(url,initial_language,final_language):
         lang='de'
     elif final_language == "Japanese":
         lang='ja'        
-
     # Initial directory
-    home_dir = os.getcwd()
+    home_dir= os.getenv('home_dir')
+    temp_dir = os.getenv('temp_dir')
+    os.chdir(home_dir)
     print('Initial directory:',home_dir)
     cleanup()
     # Temporal directory
-    temp_dir=os.path.join(home_dir, "temp")
     print('Temporal directory:',temp_dir)
-    #Create temp directory
-    pathlib.Path(temp_dir).mkdir(parents=True, exist_ok=True)
     # Go to temp directory
     os.chdir(temp_dir)
     print('Changing temporal directory',os.getcwd())
@@ -159,7 +181,15 @@ def video_to_translate(url,initial_language,final_language):
                 print(type(text))
                 
             else:
-                text = r.recognize_google(audio_data, language = lang_in)
+                try:
+                        text = r.recognize_google(audio_data, language = lang_in)
+                except Exception:
+                        print("This video cannot be recognized")
+                        cleanup()
+                        # Return back to main directory
+                        os.chdir(home_dir)
+                        return "./demo/tryagain.mp4"
+                
         #print(text)
     print("Destination language ",lang)
 
@@ -210,11 +240,11 @@ gr.Interface(fn = video_to_translate,
             outputs = 'video', 
             verbose = True,
             title = 'Video Youtube Translator',
-            description = 'A simple application that translates Youtube videos from English, Italian, Japanese, Russian, Spanish, and German  to  Italian, Spanish, Russian, English and Japanese.  Wait one minute to process.',
+            description = 'A simple application that translates Youtube small videos from English, Italian, Japanese, Russian, Spanish, and German  to  Italian, Spanish, Russian, English and Japanese.  Wait one minute to process.',
             article = 
                         '''<div>
-                            <p style="text-align: center"> All you need to do is to paste the Youtube link  and hit submit, then wait for compiling. After that click on Play/Pause for listing to the video. The video is saved in an mp4 format.
-                            For more information visit <a href="https://ruslanmv.com/">ruslanmv.com</a>
+                            <p style="text-align: center"> All you need to do is to paste the Youtube link and hit submit,, then wait for compiling. After that click on Play/Pause for listing to the video. The video is saved in an mp4 format.
+                            The lenght video limit is 10 minutes. For more information visit <a href="https://ruslanmv.com/">ruslanmv.com</a>
                             </p>
                         </div>''',
 
@@ -222,8 +252,9 @@ gr.Interface(fn = video_to_translate,
                         ["https://www.youtube.com/watch?v=Cu3R5it4cQs&list", "English","Italian"],
                         ["https://www.youtube.com/watch?v=fkGCLIQx1MI", "English","Spanish"],
                         ["https://www.youtube.com/watch?v=fkGCLIQx1MI", "English","Russian"],
-                        ["https://www.youtube.com/watch?v=_5YeX8eCLgA&ab_channel=TheTelegraph", "Russian","English"],
+                        ["https://www.youtube.com/watch?v=QbkhvLrlex4", "Russian","English"],
                         ["https://www.youtube.com/watch?v=qzzweIQoIOU", "Japanese","English"],
-                        ["https://www.youtube.com/watch?v=eo17uDr2_XA", "German","Spanish"]
+                        ["https://www.youtube.com/watch?v=nOGZvu6tJFE", "German","Spanish"]
+
                         ]           
             ).launch()