Upload main.py

main.py

@@ -0,0 +1,46 @@
+import gradio as gr
+from langchain.document_loaders.generic import GenericLoader
+from langchain.document_loaders.parsers import OpenAIWhisperParser
+from langchain.document_loaders.blob_loaders.youtube_audio import YoutubeAudioLoader
+import openai
+import os
+import platform
+
+# Set OpenAI API Key
+os.environ['OPENAI_API_KEY'] = "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"
+openai.api_key = os.environ['OPENAI_API_KEY']
+
+def add_link(url):
+    # Detect the running platform (laptop or mobile)
+    running_platform = platform.system().lower()
+
+    # Define saving directory based on the platform
+    if running_platform == 'linux' or running_platform == 'darwin':
+        save_dir = "docs/youtube/laptop/"
+    elif running_platform == 'windows':
+        save_dir = "docs/youtube/desktop/"
+    else:
+        save_dir = "docs/youtube/unknown_platform/"
+
+    # Ensure the directory exists
+    os.makedirs(save_dir, exist_ok=True)
+
+    # Create loader and parser
+    loader = GenericLoader(
+        YoutubeAudioLoader([url], save_dir),
+        OpenAIWhisperParser()
+    )
+
+    # Load documents
+    docs = loader.load()
+
+    # Show a message indicating the downloading process is complete
+    completion_message = f"Downloading process for {url} is complete. Saved in {save_dir}"
+    return completion_message
+
+# Create Gradio interface
+demo = gr.Interface(fn=add_link, inputs='text', outputs='text')
+
+# Launch the interface
+if __name__ == "__main__":
+    demo.launch(debug=False, share=True)