update

app.py

@@ -132,8 +132,11 @@ async def generate_text(request: RequestModel):
             generated_text += chunk.choices[0].delta.content
 
     return {"summary_text_2": generated_text}
-@app.post("/generate2/")
-async def generate_text(file: UploadFile = File(...)):
+@app.post("/analyse/")
+async def generate_text(
+    period: str = Query(..., description="Choisissez entre 'semaine' ou 'mois'"),
+    file: UploadFile = File(...)):
+    
     # Check file size
     contents = await file.read()
     file_size = len(contents)
@@ -161,11 +164,19 @@ async def generate_text(file: UploadFile = File(...)):
     if len(text_to_generate.encode('utf-8')) > 5_000_000:
         return {"error": "Generated text exceeds size limit even after sampling. Please reduce the data further."}
 
+    # Define the appropriate prompt based on the period (week or month)
+    if period == "semaine":
+        prompt = "Analyse des plaintes par semaine :\n\n"
+    elif period == "mois":
+        prompt = "Analyse des plaintes par mois :\n\n"
+    else:
+        return {"error": "Invalid period. Choose either 'semaine' or 'mois'."}
+
     # Create the request for the API
     try:
         completion = client.chat.completions.create(
             model="meta/llama-3.1-8b-instruct", 
-            messages=[{"role": "user", "content": prompt1 + text_to_generate}],
+            messages=[{"role": "user", "content": prompt + text_to_generate}],
             temperature=0.2,
             top_p=0.9,
             stream=True