Update app.py

app.py

@@ -8,10 +8,10 @@ import re
 from youtube_transcript_api import YouTubeTranscriptApi
 
 app = Flask(__name__)
-CORS(app)
+CORS(app)  # Enable CORS for all routes
 logging.basicConfig(level=logging.DEBUG)
 
-# Initialize the client
+# Initialize the client (this loads the API)
 CLIENT_URL = os.getenv("CLIENT_URL")
 MODEL = os.getenv("MODEL")
 
@@ -21,7 +21,7 @@ if not MODEL:
     raise ValueError("MODEL environment variable must be set")
 
 app.logger.info(f"Initializing client with URL: {CLIENT_URL}")
-client = Client('Nymbo/Groq-Playground-Master')
+client = Client(CLIENT_URL)
 
 # System message
 system_message = """You are Echo 1.5, an AI assistant developed by Abhinav. Your purpose is to assist users with their queries and provide helpful information. You should always be polite, respectful, and informative in your responses.
@@ -31,7 +31,7 @@ Remember:
 2. Be helpful and informative in your responses.
 3. If relevant, you can mention IamWomen and their work for women's welfare.
 4. Always maintain a respectful and professional tone.
-5. If the user query requires you to search the internet, you should say "I'm sorry, but the web search would be added soon."
+5. If the user's query requires you to search the internet, you should said "I'm sorry, but the web search would be added soon."
 Links:
 Echo: echo.iamwomen.cloudns.be
 Iamwomen: iamwomen.cloudns.be
@@ -39,42 +39,36 @@ Iamwomen_Instagram: https://www.instagram.com/i_am_woman.24"""
 
 context = [{"role": "system", "content": system_message}]
 
-def extract_youtube_id(url):
-    # This regex pattern matches various forms of YouTube URLs
-    pattern = r'(?:https?:\/\/)?(?:www\.)?(?:youtube\.com|youtu\.be)\/(?:watch\?v=)?(?:embed\/)?(?:v\/)?(?:shorts\/)?(?:live\/)?(?:\S+)'
-    match = re.search(pattern, url)
-    if match:
-        video_id = match.group().split('/')[-1].split('v=')[-1].split('&')[0]
-        return video_id
-    return None
+# Regex to detect YouTube URLs
+youtube_url_pattern = re.compile(
+    r'(https?://)?(www\.)?(youtube|youtu|youtube-nocookie)\.(com|be)/.+')
 
 def get_youtube_transcript(video_id):
     try:
         transcript = YouTubeTranscriptApi.get_transcript(video_id)
-        return ' '.join([entry['text'] for entry in transcript])
+        transcript_text = " ".join([item['text'] for item in transcript])
+        return transcript_text
     except Exception as e:
-        app.logger.error(f"Error fetching transcript: {str(e)}")
+        app.logger.error(f"Error fetching YouTube transcript: {str(e)}")
         return None
 
-def analyze_input(user_input):
-    video_id = extract_youtube_id(user_input)
-    if video_id:
-        transcript = get_youtube_transcript(video_id)
-        user_request = re.sub(r'(?:https?:\/\/)?(?:www\.)?(?:youtube\.com|youtu\.be)\/\S+', '', user_input).strip()
-        return video_id, user_request, transcript
-    return None, user_input, None
+def extract_youtube_id(url):
+    # Extract video ID from URL
+    video_id_match = re.search(r"(?:v=|\/)([0-9A-Za-z_-]{11}).*", url)
+    return video_id_match.group(1) if video_id_match else None
 
 def chat_with_ai(message, transcript=None):
     app.logger.debug(f"Received message: {message}")
     
     if transcript:
-        context.append({"role": "system", "content": f"Video Transcript: {transcript}"})
+        context.append({"role": "context", "content": transcript})
     
     context.append({"role": "user", "content": message})
 
+    # Prepare the full prompt
     full_prompt = "\n".join([f"{msg['role']}: {msg['content']}" for msg in context])
 
-    app.logger.debug(f"Making API call with prompt: {full_prompt[:100]}...")
+    app.logger.debug(f"Making API call with prompt: {full_prompt[:100]}...")  # Log first 100 chars of prompt
     try:
         result = client.predict(
             message=full_prompt,
@@ -85,12 +79,13 @@ def chat_with_ai(message, transcript=None):
             param_6=0,
             api_name="/chat"
         )
-        app.logger.debug(f"Received result: {result[:100]}...")
+        app.logger.debug(f"Received result: {result[:100]}...")  # Log first 100 chars of result
     except Exception as e:
         app.logger.error(f"Error in API call: {str(e)}")
         app.logger.error(traceback.format_exc())
         raise
 
+    # Add AI response to context
     context.append({"role": "assistant", "content": result})
 
     return result
@@ -108,11 +103,20 @@ def chat():
         return jsonify({"error": "No user input provided"}), 400
     
     app.logger.debug(f"Received request: {data}")
-    
+
+    youtube_match = youtube_url_pattern.search(user_input)
+    transcript = None
+
+    if youtube_match:
+        youtube_url = youtube_match.group(0)
+        video_id = extract_youtube_id(youtube_url)
+        if video_id:
+            transcript = get_youtube_transcript(video_id)
+            user_input = youtube_url_pattern.sub('', user_input).strip()
+
     try:
-        video_id, user_request, transcript = analyze_input(user_input)
-        response = chat_with_ai(user_request, transcript)
-        app.logger.debug(f"API response content: {response[:100]}...")
+        response = chat_with_ai(user_input, transcript)
+        app.logger.debug(f"API response content: {response[:100]}...")  # Log first 100 chars of response
         return jsonify({"response": response})
     except Exception as e:
         app.logger.error(f"Unexpected error: {str(e)}")