added ai for eval and sh

DragMusic/plugins/tools/dev.py

@@ -14,6 +14,7 @@ from DragMusic import app
 from config import OWNER_ID
 from DragMusic.core.mongo import mongodb
 from DragMusic import userbot  # Ensure userbot is imported
+from DragMusic.utils.gemini import get_gemini_fix
 
 async def aexec(code, client, message):
     afkdb = mongodb.afk
@@ -59,17 +60,32 @@ async def executor(client: app, message: Message):
         redirected_output = sys.stdout = StringIO()
         redirected_error = sys.stderr = StringIO()
         stdout, stderr, exc = None, None, None
-        
+        gemini_suggestion = None
+        gemini_result = None
         try:
             await aexec(cmd, client, message)
         except Exception:
             exc = traceback.format_exc()
-        
+            # Try Gemini fix
+            gemini_suggestion = await get_gemini_fix(exc, cmd, mode="python")
+            if gemini_suggestion and gemini_suggestion.strip() != cmd.strip():
+                # Try to run Gemini's suggestion
+                try:
+                    redirected_output = sys.stdout = StringIO()
+                    redirected_error = sys.stderr = StringIO()
+                    await aexec(gemini_suggestion, client, message)
+                    gemini_result = redirected_output.getvalue() or "Success"
+                except Exception as e2:
+                    gemini_result = f"Gemini fix also failed: {traceback.format_exc()}"
         stdout, stderr = redirected_output.getvalue(), redirected_error.getvalue()
         sys.stdout, sys.stderr = old_stdout, old_stderr
         
         evaluation = exc or stderr or stdout or "Success"
         final_output = f"<b>⥤ ʀᴇsᴜʟᴛ :</b>\n<pre language='python'>{evaluation}</pre>"
+        if gemini_suggestion:
+            final_output += f"\n\n<b>Gemini Suggestion:</b>\n<pre language='python'>{gemini_suggestion}</pre>"
+            if gemini_result:
+                final_output += f"\n<b>Gemini Result:</b>\n<pre language='python'>{gemini_result}</pre>"
         
         t2 = time()
         keyboard = InlineKeyboardMarkup(
@@ -89,17 +105,28 @@ async def executor(client: app, message: Message):
             return await edit_or_reply(message, text="<b>ᴇxᴀᴍᴩʟᴇ :</b>\nsh git pull")
         
         text = message.text.split(None, 1)[1]
-        shell = re.split(r""" (?=(?:[^'"]|'[^']*'|"[^"]*")*$)""", text)
-        
+        shell = re.split(r""" (?=(?:[^'\"]|'[^']*'|\"[^\"]*\")*$)""", text)
+        t1 = time()  # Ensure t1 is defined before usage
+        output = None
+        exc = None
+        gemini_suggestion = None
+        gemini_result = None
         try:
-            t1 = time()  # Ensure t1 is defined before usage
             process = subprocess.Popen(shell, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
             output = process.stdout.read().decode("utf-8").strip() or "None"
         except Exception as err:
             exc_type, exc_obj, exc_tb = sys.exc_info()
             errors = traceback.format_exception(etype=exc_type, value=exc_obj, tb=exc_tb)
-            return await edit_or_reply(message, text=f"<b>ERROR :</b>\n<pre>{''.join(errors)}</pre>")
-        
+            exc = ''.join(errors)
+            # Try Gemini fix
+            gemini_suggestion = await get_gemini_fix(exc, text, mode="shell")
+            if gemini_suggestion and gemini_suggestion.strip() != text.strip():
+                shell2 = re.split(r""" (?=(?:[^'\"]|'[^']*'|\"[^\"]*\")*$)""", gemini_suggestion)
+                try:
+                    process2 = subprocess.Popen(shell2, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+                    gemini_result = process2.stdout.read().decode("utf-8").strip() or "None"
+                except Exception as e2:
+                    gemini_result = f"Gemini fix also failed: {traceback.format_exc()}"
         t2 = time()
         keyboard = InlineKeyboardMarkup(
             [
@@ -109,8 +136,15 @@ async def executor(client: app, message: Message):
                 ]
             ]
         )
-
-        await edit_or_reply(message, text=f"<b>OUTPUT :</b>\n<pre>{output}</pre>", reply_markup=keyboard)
+        if exc:
+            final_output = f"<b>ERROR :</b>\n<pre>{exc}</pre>"
+            if gemini_suggestion:
+                final_output += f"\n\n<b>Gemini Suggestion:</b>\n<pre>{gemini_suggestion}</pre>"
+                if gemini_result:
+                    final_output += f"\n<b>Gemini Result:</b>\n<pre>{gemini_result}</pre>"
+        else:
+            final_output = f"<b>OUTPUT :</b>\n<pre>{output}</pre>"
+        await edit_or_reply(message, text=final_output, reply_markup=keyboard)
 
     await message.stop_propagation()
 

DragMusic/utils/gemini.py

@@ -0,0 +1,20 @@
+import google.generativeai as genai
+import os
+
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+
+genai.configure(api_key=GEMINI_API_KEY)
+
+async def get_gemini_fix(error_message, code, mode="python"):
+    prompt = (
+        f"I tried to run the following {mode} code/command and got this error.\n"
+        f"Code/Command:\n{code}\n\n"
+        f"Error:\n{error_message}\n\n"
+        "Please provide a corrected version of the code/command only. Do not explain, just output the fixed code/command."
+    )
+    try:
+        model = genai.GenerativeModel("gemini-pro")
+        response = model.generate_content(prompt)
+        return response.text.strip()
+    except Exception as e:
+        return None 

requirements.txt

@@ -47,3 +47,4 @@ pillow
 imagehash
 # ffmpeg is installed at the system level, but if you want a Python wrapper, uncomment the next line:
 # ffmpeg-python
+google-generativeai