Update app.py

app.py

@@ -4,16 +4,25 @@ import os
 import requests
 import base64
 import urllib.parse
+from duckduckgo_search import DDGS # Web Search සඳහා
 
-# 🔱 CONFIGURATION (NATURAL SINHALA ENGINE)
+# 🔱 CONFIGURATION
 HF_TOKEN = os.getenv("HF_TOKEN")
-# සිංහල භාෂාව සඳහා දැනට ඇති හොඳම Open-source මොළය
-TEXT_MODEL = "meta-llama/Meta-Llama-3.1-8B-Instruct"
+TEXT_MODEL = "google/gemma-2-27b-it" # Gemma Core 4 (ප්‍රබල සංස්කරණය)
 
 client = InferenceClient(token=HF_TOKEN)
 
+def web_search(query):
+    """අන්තර්ජාලයෙන් අලුත්ම තොරතුරු සෙවීමේ මෙවලම"""
+    try:
+        results = DDGS().text(query, max_results=3)
+        search_data = "\n".join([f"Source: {r['body']}" for r in results])
+        return search_data
+    except Exception:
+        return "No real-time data found."
+
 def dual_engine(message, history):
-    # 🔱 IMAGE GENERATION (POLLINATIONS FLUX)
+    # 🔱 IMAGE GENERATION ENGINE
     if message.lower().startswith("/image "):
         prompt = message.replace("/image ", "").strip()
         safe_prompt = urllib.parse.quote(prompt)
@@ -24,18 +33,22 @@ def dual_engine(message, history):
             if response.status_code == 200:
                 img_str = base64.b64encode(response.content).decode()
                 return f"IMAGE_DATA:data:image/png;base64,{img_str}"
-            else:
-                return "⚠️ Image Engine Busy. Please retry."
-        except Exception:
-            return "⚠️ Connection Error."
+            else: return "⚠️ Image Engine Busy."
+        except Exception: return "⚠️ Connection Error."
 
-    # 🔱 TEXT CHAT (LLAMA 3.1 - NATURAL SINHALA)
+    # 🔱 TEXT CHAT WITH WEB SEARCH
     else:
-        # සිංහලෙන් වඩාත් හොඳින් පිළිතුරු දීමට System Prompt එක සැකසීම
+        # අලුත්ම තොරතුරු අවශ්‍ය දැයි තීරණය කිරීම (Simple Keyword Check)
+        search_keywords = ["news", "today", "weather", "latest", "price", "අද", "අලුත්ම"]
+        context = ""
+        
+        if any(word in message.lower() for word in search_keywords):
+            context = web_search(message)
+
         system_instruction = (
-            "You are MINZO AI, created by MINZO-PRIME. "
-            "You must communicate in natural, fluent Sinhala. "
-            "Helpful, brilliant, and strategic in your responses. "
+            f"නම: MINZO AI. නිර්මාතෘ: MINZO-PRIME. "
+            f"ඔබ සතු අන්තර්ජාල සෙවුම් දත්ත: {context}\n"
+            "ඉහත දත්ත පාවිච්චි කරමින් ඉතාමත් නිවැරදි සහ ස්වාභාවික සිංහලෙන් පිළිතුරු දෙන්න. "
             "පින්තූර සෑදීමට /image විධානය භාවිතා කරන්න."
         )
         
@@ -50,11 +63,11 @@ def dual_engine(message, history):
                 messages, 
                 model=TEXT_MODEL, 
                 max_tokens=1500,
-                temperature=0.7 # සිංහල ගලායාම ස්වාභාවික කිරීමට
+                temperature=0.6
             )
             return response.choices[0].message.content
-        except Exception:
-            return "⚠️ පද්ධතියේ දෝෂයක්. කරුණාකර නැවත උත්සාහ කරන්න."
+        except Exception as e:
+            return f"⚠️ Neural Link Error: {str(e)}"
 
 # Gradio Interface
 demo = gr.ChatInterface(fn=dual_engine, api_name="chat")