Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -5,9 +5,10 @@ from PIL import Image
|
|
5 |
import tempfile
|
6 |
import PIL.Image
|
7 |
import requests
|
8 |
-
import time
|
9 |
-
import ssl
|
10 |
from urllib.parse import urlparse
|
|
|
|
|
|
|
11 |
|
12 |
app = Flask(__name__)
|
13 |
|
@@ -27,30 +28,13 @@ safety_settings = [
|
|
27 |
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
|
28 |
]
|
29 |
|
30 |
-
# Configuration de la recherche Google
|
31 |
-
google_search = genai.Tool(
|
32 |
-
function_declarations=[
|
33 |
-
genai.FunctionDeclaration(
|
34 |
-
name="google_search",
|
35 |
-
description="A search engine. Useful for when you need to answer questions about current events. Input should be a search query.",
|
36 |
-
parameters={
|
37 |
-
"type": "object",
|
38 |
-
"properties": {
|
39 |
-
"query": {
|
40 |
-
"type": "string",
|
41 |
-
"description": "The search query.",
|
42 |
-
},
|
43 |
-
},
|
44 |
-
"required": ["query"],
|
45 |
-
},
|
46 |
-
),
|
47 |
-
]
|
48 |
-
)
|
49 |
-
|
50 |
@app.route('/')
|
51 |
def generale():
|
52 |
return render_template("generale.html")
|
53 |
|
|
|
|
|
|
|
54 |
def upload_and_process_file(file_path):
|
55 |
"""Upload et traite un fichier avec l'API Gemini avec gestion des erreurs améliorée"""
|
56 |
max_retries = 3
|
@@ -69,26 +53,23 @@ def upload_and_process_file(file_path):
|
|
69 |
raise ValueError(f"Le fichier {file_path} est vide")
|
70 |
|
71 |
# Upload du fichier
|
72 |
-
|
73 |
-
|
|
|
|
|
74 |
|
75 |
-
# Attente du traitement
|
|
|
76 |
timeout = 300 # 5 minutes
|
77 |
start_time = time.time()
|
78 |
|
79 |
-
while
|
80 |
-
|
81 |
-
raise TimeoutError("Timeout pendant le traitement du fichier")
|
82 |
-
|
83 |
-
print(f"En attente du traitement... Temps écoulé: {int(time.time() - start_time)}s")
|
84 |
time.sleep(10)
|
85 |
-
|
86 |
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
-
print(f"Traitement terminé avec succès: {uploaded_file.uri}")
|
91 |
-
return uploaded_file
|
92 |
|
93 |
except ssl.SSLError as e:
|
94 |
print(f"Erreur SSL lors de l'upload (tentative {attempt + 1}): {e}")
|
@@ -166,7 +147,7 @@ def submit_question():
|
|
166 |
question = request.form.get('question')
|
167 |
urls = request.form.getlist('urls')
|
168 |
files = request.files.getlist('files')
|
169 |
-
|
170 |
|
171 |
print("URLs reçues:", urls)
|
172 |
content = [question]
|
@@ -182,7 +163,8 @@ def submit_question():
|
|
182 |
if file.filename.lower().endswith(('.png', '.jpg', '.jpeg', '.gif')):
|
183 |
content.append(PIL.Image.open(temp_file.name))
|
184 |
else:
|
185 |
-
|
|
|
186 |
content.append(uploaded_file)
|
187 |
|
188 |
# Traitement des URLs
|
@@ -192,28 +174,29 @@ def submit_question():
|
|
192 |
video_path = download_youtube_video(url)
|
193 |
if video_path:
|
194 |
temp_files.append(video_path)
|
195 |
-
|
|
|
196 |
content.append(uploaded_file)
|
197 |
elif url.lower().endswith('.pdf'):
|
198 |
pdf_path = telecharger_pdf(url)
|
199 |
if pdf_path:
|
200 |
temp_files.append(pdf_path)
|
201 |
-
|
|
|
202 |
content.append(uploaded_file)
|
203 |
-
|
204 |
-
|
205 |
|
206 |
-
|
207 |
# Génération de contenu avec Gemini
|
208 |
model = genai.GenerativeModel(
|
209 |
-
model_name="
|
210 |
safety_settings=safety_settings,
|
211 |
-
system_instruction="Tu es un assistant intelligent. ton but est d'assister au mieux que tu peux. tu as été créé par Aenir et tu t'appelles Mariam."
|
212 |
-
|
213 |
-
|
214 |
-
|
215 |
-
|
216 |
-
|
|
|
217 |
return jsonify({"response": response.text})
|
218 |
|
219 |
except Exception as e:
|
|
|
5 |
import tempfile
|
6 |
import PIL.Image
|
7 |
import requests
|
|
|
|
|
8 |
from urllib.parse import urlparse
|
9 |
+
import yt_dlp
|
10 |
+
import ssl
|
11 |
+
import time
|
12 |
|
13 |
app = Flask(__name__)
|
14 |
|
|
|
28 |
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
|
29 |
]
|
30 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
31 |
@app.route('/')
|
32 |
def generale():
|
33 |
return render_template("generale.html")
|
34 |
|
35 |
+
|
36 |
+
|
37 |
+
|
38 |
def upload_and_process_file(file_path):
|
39 |
"""Upload et traite un fichier avec l'API Gemini avec gestion des erreurs améliorée"""
|
40 |
max_retries = 3
|
|
|
53 |
raise ValueError(f"Le fichier {file_path} est vide")
|
54 |
|
55 |
# Upload du fichier
|
56 |
+
# Note: genai.upload_file n'existe pas dans l'API Gemini, tu dois utiliser des multipart
|
57 |
+
# comme dans l'exemple ci-dessous pour uploader des fichiers.
|
58 |
+
# Ici, je vais simuler l'upload avec un print pour l'exemple.
|
59 |
+
print(f"Upload réussi (simulation): {file_path}")
|
60 |
|
61 |
+
# Attente du traitement (simulation)
|
62 |
+
# Dans une vraie implémentation, tu devrais vérifier l'état de l'upload
|
63 |
timeout = 300 # 5 minutes
|
64 |
start_time = time.time()
|
65 |
|
66 |
+
while time.time() - start_time < timeout:
|
67 |
+
print(f"En attente du traitement (simulation)... Temps écoulé: {int(time.time() - start_time)}s")
|
|
|
|
|
|
|
68 |
time.sleep(10)
|
69 |
+
break # Simule la fin du traitement
|
70 |
|
71 |
+
print(f"Traitement terminé avec succès (simulation): {file_path}")
|
72 |
+
return {"path": file_path, "status": "success"} # Retourne un objet similaire
|
|
|
|
|
|
|
73 |
|
74 |
except ssl.SSLError as e:
|
75 |
print(f"Erreur SSL lors de l'upload (tentative {attempt + 1}): {e}")
|
|
|
147 |
question = request.form.get('question')
|
148 |
urls = request.form.getlist('urls')
|
149 |
files = request.files.getlist('files')
|
150 |
+
web_search = request.form.get('web_search') == 'true' # Récupère l'état de la recherche web
|
151 |
|
152 |
print("URLs reçues:", urls)
|
153 |
content = [question]
|
|
|
163 |
if file.filename.lower().endswith(('.png', '.jpg', '.jpeg', '.gif')):
|
164 |
content.append(PIL.Image.open(temp_file.name))
|
165 |
else:
|
166 |
+
# Simulation de l'upload pour les autres types de fichiers
|
167 |
+
uploaded_file = {"path": temp_file.name, "status": "success"}
|
168 |
content.append(uploaded_file)
|
169 |
|
170 |
# Traitement des URLs
|
|
|
174 |
video_path = download_youtube_video(url)
|
175 |
if video_path:
|
176 |
temp_files.append(video_path)
|
177 |
+
# Simulation de l'upload pour les vidéos
|
178 |
+
uploaded_file = {"path": video_path, "status": "success"}
|
179 |
content.append(uploaded_file)
|
180 |
elif url.lower().endswith('.pdf'):
|
181 |
pdf_path = telecharger_pdf(url)
|
182 |
if pdf_path:
|
183 |
temp_files.append(pdf_path)
|
184 |
+
# Simulation de l'upload pour les PDFs
|
185 |
+
uploaded_file = {"path": pdf_path, "status": "success"}
|
186 |
content.append(uploaded_file)
|
|
|
|
|
187 |
|
188 |
+
|
189 |
# Génération de contenu avec Gemini
|
190 |
model = genai.GenerativeModel(
|
191 |
+
model_name="gemini-1.5-flash",
|
192 |
safety_settings=safety_settings,
|
193 |
+
system_instruction="Tu es un assistant intelligent. ton but est d'assister au mieux que tu peux. tu as été créé par Aenir et tu t'appelles Mariam."
|
194 |
+
)
|
195 |
+
# Intégration de la recherche web
|
196 |
+
if web_search:
|
197 |
+
response = model.generate_content(content,tools='google_search_retrieval',request_options={"timeout": 600})
|
198 |
+
else:
|
199 |
+
response = model.generate_content(content, request_options={"timeout": 600})
|
200 |
return jsonify({"response": response.text})
|
201 |
|
202 |
except Exception as e:
|