Update response_generator.py

response_generator.py

@@ -1,5 +1,4 @@
-import requests
-import json
+from openai import OpenAI  # نستخدم نفس المكتبة!
 from utils import setup_logger
 from config import Config
 
@@ -7,50 +6,48 @@ logger = setup_logger('response_generator')
 
 class ResponseGenerator:
     def __init__(self):
-        logger.info("Response generator initialized (Google Direct REST API Mode)")
+        logger.info("Response generator initialized (Groq Llama-3 mode)")
         
-        if not Config.GOOGLE_API_KEY:
-            logger.error("GOOGLE_API_KEY is missing.")
-            self.api_key = None
+        if not Config.GROQ_API_KEY:
+            logger.error("GROQ_API_KEY is missing.")
+            self.client = None
         else:
-            self.api_key = Config.GOOGLE_API_KEY
-            # رابط الاتصال المباشر بجوجل
-            self.api_url = f"https://generativelanguage.googleapis.com/v1beta/models/{Config.GENAI_MODEL}:generateContent?key={self.api_key}"
+            # هنا السحر: نوجه العميل إلى رابط Groq
+            self.client = OpenAI(
+                api_key=Config.GROQ_API_KEY,
+                base_url="https://api.groq.com/openai/v1"
+            )
 
     def generate_response(self, query, relevant_docs):
         if len(relevant_docs) == 0:
             return "عذرًا، لم أجد أي معلومات ذات صلة في المستندات."
 
-        if not self.api_key:
-            return "عذرًا، لم يتم إعداد مفتاح API الخاص بجوجل."
+        if not self.client:
+            return "عذرًا، لم يتم إعداد مفتاح API."
 
         try:
-            # تجهيز السياق
             context_text = "\n\n".join(relevant_docs['content'].tolist())
 
-            # تجهيز الرسالة (JSON)
-            payload = {
-                "contents": [{
-                    "parts": [{
-                        "text": f"أنت مساعد لطلاب السنة الأولى المشتركة. جاوب بناء على السياق التالي:\n\n{context_text}\n\nالسؤال: {query}"
-                    }]
-                }]
-            }
-
-            headers = {'Content-Type': 'application/json'}
-
-            # إرسال الطلب مباشرة (POST request)
-            response = requests.post(self.api_url, headers=headers, data=json.dumps(payload))
-
-            # التحقق من نجاح الطلب
-            if response.status_code == 200:
-                result = response.json()
-                # استخراج النص من رد جوجل
-                return result['candidates'][0]['content']['parts'][0]['text']
-            else:
-                logger.error(f"Google API Error: {response.text}")
-                return f"حدث خطأ في الاتصال بجوجل: {response.status_code}"
+            system_instruction = (
+                "أنت مساعد ذكي تتحدث العربية بطلاقة. مهمتك الإجابة بدقة بناءً على السياق فقط."
+                "إذا لم تجد الإجابة في السياق، قل 'لا تتوفر معلومات'."
+                "اجعل إجابتك رسمية، قصيرة، ومباشرة."
+            )
+
+            user_message = f"السياق:\n{context_text}\n\nالسؤال: {query}"
+
+            response = self.client.chat.completions.create(
+                model=Config.LLM_MODEL,
+                messages=[
+                    {"role": "system", "content": system_instruction},
+                    {"role": "user", "content": user_message}
+                ],
+                temperature=0.3,
+                max_tokens=300
+            )
+
+            return response.choices[0].message.content.strip()
 
         except Exception as e:
             logger.error(f"Error generating response: {e}")
-            return "عذرًا، واجهت مشكلة أثناء صياغة الإجابة."
+            return "عذرًا، حدث خطأ أثناء المعالجة."