Update response_generator.py

response_generator.py

@@ -1,4 +1,5 @@
-import google.generativeai as genai
+import requests
+import json
 from utils import setup_logger
 from config import Config
 
@@ -6,41 +7,50 @@ logger = setup_logger('response_generator')
 
 class ResponseGenerator:
     def __init__(self):
-        logger.info("Response generator initialized (Google Gemini Flash mode)")
+        logger.info("Response generator initialized (Google Direct REST API Mode)")
         
         if not Config.GOOGLE_API_KEY:
             logger.error("GOOGLE_API_KEY is missing.")
-            self.model = None
+            self.api_key = None
         else:
-            # إعداد مكتبة جوجل
-            genai.configure(api_key=Config.GOOGLE_API_KEY)
-            self.model = genai.GenerativeModel(Config.GENAI_MODEL)
+            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}"
 
     def generate_response(self, query, relevant_docs):
         if len(relevant_docs) == 0:
             return "عذرًا، لم أجد أي معلومات ذات صلة في المستندات."
 
-        if not self.model:
+        if not self.api_key:
             return "عذرًا، لم يتم إعداد مفتاح API الخاص بجوجل."
 
         try:
-            # دمج النصوص المسترجعة
+            # تجهيز السياق
             context_text = "\n\n".join(relevant_docs['content'].tolist())
 
-            # صياغة البرومبت (التعليمات)
-            prompt = (
-                f"أنت مساعد ذكي لطلاب السنة الأولى المشتركة. "
-                f"استخدم المعلومات التالية للإجابة على سؤال الطالب بدقة ومباشرة.\n"
-                f"معلومات من اللائحة:\n{context_text}\n\n"
-                f"سؤال الطالب: {query}\n\n"
-                f"الإجابة (تحدث بصيغة رسمية وودودة، وإذا لم تكن المعلومة موجودة قل لا أعلم):"
-            )
-
-            # إرسال الطلب
-            response = self.model.generate_content(prompt)
-            
-            return response.text.strip()
+            # تجهيز الرسالة (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}"
 
         except Exception as e:
-            logger.error(f"Error generating Gemini response: {e}")
+            logger.error(f"Error generating response: {e}")
             return "عذرًا، واجهت مشكلة أثناء صياغة الإجابة."