Enhanced AraGPT2 response with better cleaning and prompts

app.py

@@ -90,30 +90,34 @@ async def get_car(car_id: int):
 @app.get("/car_info/")
 async def get_car_info(model: str):
     try:
-        cars = db.get_all_cars()  # استخدمنا db
+        cars = db.get_all_cars()
         available_cars = [car for car in cars if car['model'].lower() == model.lower()]
         
         if not available_cars:
-            prompt = f"ما اعرف هذا الموديل '{model}'، ممكن تكلي شنو تعرف عنه؟"
+            prompt = f"ما اعرف هذا الموديل '{model}'، اعطيني معلومات عنه بلهجة عراقية طبيعية."
             try:
                 gen = load_generator()
-                response = gen(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
-                response = response.split("؟")[0].strip() + "."
+                response = gen(prompt, max_length=80, num_return_sequences=1, do_sample=True, temperature=0.7)[0]['generated_text']
+                response = response.split(".")[0].split("؟")[0].strip() + "."
+                if len(response) < 20:  # إذا الرد قصير جدًا، نرجع رد افتراضي
+                    response = f"والله ما عندي معلومات كافية عن '{model}'، جربي موديل ثاني!"
                 return {"message": response}
             except Exception as e:
                 logger.error(f"Error generating response for unknown model: {str(e)}")
-                return {"message": "عذرًا، حدث خطأ أثناء توليد الرد."}
+                return {"message": "عذرًا، خطأ أثناء توليد الرد."}
         
         car = available_cars[0]
-        prompt = f"زبون يريد يعرف عن سيارة {car['model']}، سعرها {car['price']} ووضعها {car['status']}، كله بلهجة عراقية طبيعية."
+        prompt = f"اعطيني معلومات عن سيارة {car['model']}، سعرها {car['price']} دولار ووضعها {car['status']}، بلهجة عراقية طبيعية."
         try:
             gen = load_generator()
-            response = gen(prompt, max_length=150, num_return_sequences=1)[0]['generated_text']
-            response = response.split("؟")[0].strip() + "."
+            response = gen(prompt, max_length=100, num_return_sequences=1, do_sample=True, temperature=0.7)[0]['generated_text']
+            response = response.split(".")[0].split("؟")[0].strip() + "."
+            if len(response) < 20:  # إذا الرد قصير جدًا
+                response = f"سيارة {car['model']} سعرها {car['price']} دولار، ووضعها {car['status']}، حلوة ومناسبة للعراق!"
             return {"message": response}
         except Exception as e:
             logger.error(f"Error generating response for car info: {str(e)}")
-            return {"message": "عذرًا، حدث خطأ أثناء توليد الرد."}
+            return {"message": "عذرًا، خطأ أثناء توليد الرد."}
     except Exception as e:
         logger.error(f"Error processing car info for model {model}: {str(e)}")
         raise HTTPException(status_code=500, detail="خطأ أثناء معالجة طلب المعلومات")