|
|
from flask import Flask, render_template, request, jsonify |
|
|
import google.generativeai as genai |
|
|
import os |
|
|
from dotenv import load_dotenv |
|
|
import http.client |
|
|
import json |
|
|
from werkzeug.utils import secure_filename |
|
|
|
|
|
app = Flask(__name__) |
|
|
app.config['UPLOAD_FOLDER'] = 'temp' |
|
|
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024 |
|
|
|
|
|
load_dotenv() |
|
|
|
|
|
|
|
|
genai.configure(api_key=os.getenv("GOOGLE_API_KEY")) |
|
|
|
|
|
safety_settings = [ |
|
|
{"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"}, |
|
|
{"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"}, |
|
|
{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"}, |
|
|
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"}, |
|
|
] |
|
|
|
|
|
model = genai.GenerativeModel('gemini-2.0-flash-exp', |
|
|
tools='code_execution', |
|
|
safety_settings=safety_settings, |
|
|
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") |
|
|
|
|
|
def perform_web_search(query): |
|
|
conn = http.client.HTTPSConnection("google.serper.dev") |
|
|
payload = json.dumps({"q": query}) |
|
|
headers = { |
|
|
'X-API-KEY': '9b90a274d9e704ff5b21c0367f9ae1161779b573', |
|
|
'Content-Type': 'application/json' |
|
|
} |
|
|
try: |
|
|
conn.request("POST", "/search", payload, headers) |
|
|
res = conn.getresponse() |
|
|
data = json.loads(res.read().decode("utf-8")) |
|
|
return data |
|
|
except Exception as e: |
|
|
return {"error": str(e)} |
|
|
finally: |
|
|
conn.close() |
|
|
|
|
|
def format_search_results(data): |
|
|
if not data: |
|
|
return "Aucun résultat trouvé" |
|
|
|
|
|
result = "" |
|
|
|
|
|
if 'knowledgeGraph' in data: |
|
|
kg = data['knowledgeGraph'] |
|
|
result += f"### {kg.get('title', '')}\n" |
|
|
result += f"*{kg.get('type', '')}*\n\n" |
|
|
result += f"{kg.get('description', '')}\n\n" |
|
|
|
|
|
if 'organic' in data: |
|
|
result += "### Résultats principaux:\n" |
|
|
for item in data['organic'][:3]: |
|
|
result += f"- **{item['title']}**\n" |
|
|
result += f" {item['snippet']}\n" |
|
|
result += f" [Lien]({item['link']})\n\n" |
|
|
|
|
|
return result |
|
|
|
|
|
@app.route('/') |
|
|
def home(): |
|
|
return render_template('index.html') |
|
|
|
|
|
@app.route('/chat', methods=['POST']) |
|
|
def chat(): |
|
|
data = request.json |
|
|
prompt = data.get('message') |
|
|
web_search_enabled = data.get('web_search', False) |
|
|
|
|
|
try: |
|
|
web_results = None |
|
|
if web_search_enabled: |
|
|
web_results = perform_web_search(prompt) |
|
|
if web_results and 'error' not in web_results: |
|
|
formatted_results = format_search_results(web_results) |
|
|
prompt = f"""Question: {prompt}\n\nRésultats de recherche web:\n{formatted_results}\n\nPourrais-tu analyser ces informations et me donner une réponse complète?""" |
|
|
|
|
|
response = model.generate_content(prompt) |
|
|
return jsonify({"response": response.text}) |
|
|
|
|
|
except Exception as e: |
|
|
return jsonify({"error": str(e)}), 500 |
|
|
|
|
|
@app.route('/upload', methods=['POST']) |
|
|
def upload_file(): |
|
|
if 'file' not in request.files: |
|
|
return jsonify({"error": "No file part"}), 400 |
|
|
|
|
|
file = request.files['file'] |
|
|
if file.filename == '': |
|
|
return jsonify({"error": "No selected file"}), 400 |
|
|
|
|
|
if file: |
|
|
filename = secure_filename(file.filename) |
|
|
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename) |
|
|
file.save(filepath) |
|
|
|
|
|
try: |
|
|
gemini_file = genai.upload_file(filepath) |
|
|
return jsonify({"success": True, "filename": filename}) |
|
|
except Exception as e: |
|
|
return jsonify({"error": str(e)}), 500 |
|
|
|
|
|
if __name__ == '__main__': |
|
|
os.makedirs("temp", exist_ok=True) |
|
|
app.run(debug=True) |
