Update app.py

app.py

@@ -1,35 +1,30 @@
 from youtube_transcript_api import YouTubeTranscriptApi
 from openai import OpenAI
 import gradio as gr
-import os
-from dotenv import load_dotenv
-
-load_dotenv()
-
-api_key= os.getenv("api_key")
 
 def youtube_url(url):
-    
-    """
-    Function to retrieve the transcript of a YouTube video based on the provided URL.
-    
-    Parameters:
-    url (str): The URL of the YouTube video.
-    
-    Returns:
-    str: The transcript of the video.
-    """
+  
+  """
+  Function to retrieve the transcript of a YouTube video based on the provided URL.
+  
+  Parameters:
+  url (str): The URL of the YouTube video.
+  
+  Returns:
+  str: The transcript of the video.
+  """
 
-    transcript = ""
-    list=url.split("=")
-    video_id = list[1]
-    list = YouTubeTranscriptApi.get_transcript(video_id, languages=['tr', 'en',"de"])
-    for dict in list:
-        transcript += dict["text"] + "\n"
-    return transcript
+  transcript = ""
+  list=url.split("=")
+  video_id = list[1]
+  list = YouTubeTranscriptApi.get_transcript(video_id, languages=['tr', 'en',"de"])
+  for dict in list:
+      transcript += dict["text"] + "\n"
+  return transcript
 
 
-def chat(prompt):
+
+def summarizer(prompt, base_url, model, api_key):
 
   """
   This function takes a prompt as input and uses the OpenAI API to generate a chat completion based on the prompt. It returns the summary of the chat completion.
@@ -37,21 +32,20 @@ def chat(prompt):
 
   system_msg = "you are a youtube transcript summarizer."
 
-  client = OpenAI(api_key = api_key, base_url="https://api.deepseek.com/v1")
+  client = OpenAI(api_key = api_key, base_url=base_url)
 
   completion = client.chat.completions.create(
-  model="deepseek-chat",
-  messages=[
-      {"role": "system", "content": system_msg},
-      {"role": "user", "content": prompt}
-  ]
+    model=model,
+    messages=[
+        {"role": "system", "content": system_msg},
+        {"role": "user", "content": prompt}
+    ]
   )
   summary = completion.choices[0].message.content
   return summary
 
 
-def main(url):
-
+def main(url, model, api_key):     
   """
   Function to summarize the transcript of a YouTube video using the provided URL.
   
@@ -61,16 +55,26 @@ def main(url):
   Returns:
   str: The summary of the video transcript.
   """
+
+  if model == "deepseek-chat":
+    base_url = "https://api.deepseek.com/v1"
+  
+  else:
+    base_url = None
   
   transcript = youtube_url(url)
   query=f"summarize this {transcript} transcript"
-  summary = chat(query)
+  summary = summarizer(query, base_url, model, api_key)
   return summary
 
 
 iface = gr.Interface(
     fn=main,
-    inputs="text",
+    inputs=[
+        gr.Textbox(label="YouTube URL", type="text"),
+        gr.Dropdown(label="Model Seçimi", choices = ["deepseek-chat", "gpt-3 turbo", "gpt-3.5-turbo"]),
+        gr.Textbox(label="Api Key", type="text")
+    ],
     outputs="text",
     
     title="Video Özeti Oluşturucu",