sameerrawat07/yt-transcript

1from flask import Flask, request, jsonify2import os3from dotenv import load_dotenv4load_dotenv() 5app = Flask(__name__)6 7@app.route('/summarize', methods=['POST'])8def summarize():9    print("flask running1")10    data = request.json11    video_transcript = data.get('transcript')12    13    try:14        import google.generativeai as genai15        # Define the video ID16        API_KEY=os.getenv("API_KEY")17        genai.configure(api_key=API_KEY)18 19        # Combine all the transcript text into a single string20        transcript_text = " ".join([entry['text'] for entry in video_transcript])21        model = genai.GenerativeModel("gemini-1.5-flash")22        print("flask running2")23        response = model.generate_content(f"Summarize the youtube content given below(Just use plain text)\n{transcript_text}")24        print("flask running3")25        return jsonify({'summary': response.text})26    except Exception as e:27        return jsonify({'error': str(e)}), 50028 29if __name__ == '__main__':30    app.run(debug=True)31