Upload app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask,make_response,render_template, request, redirect, url_for, session, jsonify, flash
+from flask import Flask,make_response,render_template, request, redirect, url_for, session, jsonify, flash, Response
 from datetime import datetime, timedelta
 import sqlite3
 from urllib.parse import urlparse
@@ -9,7 +9,6 @@ import time
 import tempfile
 import logging
 import re
-import psycopg2
 from markupsafe import Markup
 from gradio_client import Client,handle_file
 from functools import wraps
@@ -18,8 +17,7 @@ import google.generativeai as genai
 import textwrap
 import PIL.Image
 from werkzeug.utils import secure_filename
-from pytube import YouTube
-from psycopg2.extras import RealDictCursor
+
 import redis 
 
 
@@ -28,7 +26,7 @@ logging.basicConfig(level=logging.INFO)
 
 DATABASE_URL= os.environ.get("DATABASE")
 
-GOOGLE_API_KEY = os.environ.get("TOKEN")
+GOOGLE_API_KEY = "AIzaSyC_zxN9IHjEAxIoshWPzMfgb9qwMsu5t5Y"
 
 
 
@@ -57,10 +55,8 @@ safety_settings = [
   },
 ]
 
-
 genai.configure(api_key=GOOGLE_API_KEY)
 
-
 MAX_CONTENT_LENGTH = 760 * 1024 * 1024  # 16MB max file size
 
 def create_connection():
@@ -119,19 +115,17 @@ def get_philosophy_course(course_id):
         print(f"Erreur lors de la récupération du cours : {e}")
         return jsonify({"error": str(e)}), 500
 
-# Route Flask mise à jour
-@app.route('/submit_philo', methods=['POST'])
+# Route Flask mise à jour pour le streaming
+@app.route('/submit_philo', methods=['GET', 'POST'])  # Accepte GET et POST
 def submit_philo():
-    data = request.json
-    phi_prompt = data.get('question', '').strip()
-    phi_type = data.get('type', '1')  # Type 1 par défaut
-    course_id = data.get('courseId')
-
-    logging.info(f"Le sujet est: {phi_prompt}")
-    logging.info(f"Le type est: {phi_type}")
-    
-    if not phi_prompt:
-        return jsonify({"error": "Veuillez saisir un sujet."}), 400
+    if request.method == 'POST':
+        data = request.json
+    else:  # request.method == 'GET'
+        data_string = request.args.get('data')
+        if data_string:
+            data = json.loads(data_string)
+        else:
+            return jsonify({"error": "No data provided"}), 400
 
     try:
         # Récupération du contenu du cours si un cours est sélectionné
@@ -196,7 +190,6 @@ def submit_philo():
             Je veux un travail bien détaillé et complet avec un français raffiné et soutenu.
           """
 
-
         
 
             
@@ -389,7 +382,6 @@ Prise de position :
 
 La conclusion répond à la question critique posée en introduction, en s'appuyant sur l'ensemble de la réflexion développée.
 
-
             """
 
         # Ajout du contenu du cours à la prompt si disponible
@@ -398,15 +390,19 @@ La conclusion répond à la question critique posée en introduction, en s'appuy
 
         prompt = methodologie
 
-        try:
-            model = genai.GenerativeModel(model_name="models/gemini-2.0-flash-exp", safety_settings=safety_settings)
-            response = model.generate_content(prompt, request_options={"timeout": 1100})
-            
-            dissertation = response.text
-            return jsonify({"response": dissertation}), 200
-        except Exception as e:
-            print(f"Erreur dans la génération de contenu : {e}")
-            return jsonify({"error": "error"}), 500
+        def generate_stream():
+            try:
+                model = genai.GenerativeModel(model_name="models/gemini-2.0-flash-exp", safety_settings=safety_settings)
+                response = model.generate_content(prompt, stream=True, request_options={"timeout": 1100})
+
+                for chunk in response:
+                  yield f"data: {json.dumps({'response': chunk.text})}\n\n"
+                  time.sleep(0.05)  # Ajustez le délai selon les besoins pour la fluidité du streaming
+            except Exception as e:
+                print(f"Erreur dans la génération de contenu : {e}")
+                yield f"data: {json.dumps({'error': 'Erreur lors de la génération'})}\n\n"
+
+        return Response(generate_stream(), mimetype='text/event-stream')
 
     except Exception as e:
         print(f"Erreur dans submit_philo : {e}")
@@ -460,6 +456,8 @@ def manage_philosophy_courses():
                 return redirect(url_for('manage_philosophy_courses'))
             except Exception as e:
                 flash(f'Erreur lors de la suppression du cours : {e}', 'danger')
-                return redirect(url_for('manage_philosophy_courses'))        
-        
-
+                return redirect(url_for('manage_philosophy_courses'))  
+                
+                      
+if __name__ == '__main__':
+    app.run(debug=True)