Update app.py

app.py

@@ -7,9 +7,14 @@ from selenium.webdriver.chrome.options import Options
 import openai
 import gradio as gr
 import time
+from dotenv import load_dotenv
 
-# Set up the YouTube Data API key
-YOUTUBE_API_KEY = 'AIzaSyB18WjBnkOYmTO3xiWlRqcVgwMAlPtLE0w'
+# Load environment variables from .env file
+load_dotenv()
+
+# Get API keys from environment variables
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+YOUTUBE_API_KEY = os.getenv("YOUTUBE_API_KEY")
 
 def extract_video_id(url):
     if "youtu.be" in url:  # Mobile YouTube URL
@@ -47,7 +52,6 @@ def get_redirected_url_video_id(mobile_url):
     video_id = re.search(r"v=([^&]+)", current_url).group(1)
     return video_id
 
-
 def get_youtube_comments(video_id):
     youtube = build('youtube', 'v3', developerKey=YOUTUBE_API_KEY)
     comments = []
@@ -91,7 +95,6 @@ def get_youtube_comments(video_id):
     comments = sorted(comments, key=lambda x: x['like_count'], reverse=True)[:20]
     return [comment['text'] for comment in comments]
 
-
 def generate_story(comments, temperature=0.7):
     words = []
     word_count = 0
@@ -104,25 +107,27 @@ def generate_story(comments, temperature=0.7):
         word_count += len(comment_words)
 
     comments_text = " ".join(words)
-    client = openai
-    completion = client.ChatCompletion.create(
+    
+    openai.api_key = OPENAI_API_KEY
+    completion = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
-        messages=[{"role": "system", "content": f"""
-                    
-                        Read all the comments, understand the emotions people are feeling and pick any random emotion 
-                        and create a story in first person (the person can be randomly young or old and the story can be
-                        based in past or future) based on that emotion picking a random character keep the words 
-                        simple and a bit profound but overall simple words. Give more weight to the comments that 
-                        come earlier in sequence. The comments are given here: {comments_text}"""},
-                  {"role": "user", "content": """
-                        ignore the comment which has lyrics of the song, 
-                  ignore all comments similar to 'anyone in 2024', Keep the story randomly between 50-120 words. 
-                  don't mention the emotion chosen just start the story. 
-                  sometimes include bits where you tell how this song makes you feel. be very nostalgic about a feeling or a place this
-                  takes you to. half the times end the story with a hopeful future or a dark end or humorous.. choose randomly"""}]
-    ,temperature=temperature)
-    return completion.choices[0].message.content
-
+        messages=[
+            {"role": "system", "content": f"""
+            Read all the comments, understand the emotions people are feeling and pick any random emotion 
+            and create a story in first person (the person can be randomly young or old and the story can be
+            based in past or future) based on that emotion picking a random character keep the words 
+            simple and a bit profound but overall simple words. Give more weight to the comments that 
+            come earlier in sequence. The comments are given here: {comments_text}"""},
+            {"role": "user", "content": """
+            ignore the comment which has lyrics of the song, 
+            ignore all comments similar to 'anyone in 2024', Keep the story randomly between 50-120 words. 
+            don't mention the emotion chosen just start the story. 
+            sometimes include bits where you tell how this song makes you feel. be very nostalgic about a feeling or a place this
+            takes you to. half the times end the story with a hopeful future or a dark end or humorous.. choose randomly"""}
+        ],
+        temperature=temperature
+    )
+    return completion.choices[0].message['content']
 
 # Main function to execute the process
 def main(youtube_url, temperature):
@@ -131,7 +136,6 @@ def main(youtube_url, temperature):
     story = generate_story(comments, temperature)
     return story
 
-
 # Create Gradio interface
 youtube_url_input = gr.Textbox(label="YouTube URL")
 temperature_input = gr.Slider(minimum=0.0, maximum=2.0, value=0.7, label="Temperature (creativity)")
@@ -145,4 +149,4 @@ iface = gr.Interface(
 )
 
 # Launch the interface
-iface.launch()
+iface.launch(share=True)