Update app.py

app.py

@@ -23,23 +23,31 @@ except Exception as e:
 
 def get_smart_fallback(diff_text, filename):
     """
-    If AI fails, look at the code to see WHICH function was touched.
+    Analyzes the CODE itself to generate a specific message 
+    when the AI model fails or hallucinates.
     """
-    # Look for function definitions or modifications
-    # Regex matches: function name(), const name =, class Name
-    patterns = [
-        r'function\s+([a-zA-Z0-9_]+)',
-        r'const\s+([a-zA-Z0-9_]+)\s*=',
-        r'let\s+([a-zA-Z0-9_]+)\s*=',
-        r'class\s+([a-zA-Z0-9_]+)',
-        r'def\s+([a-zA-Z0-9_]+)'
-    ]
-    
+    # 1. Detect Logging / Debugging
+    if "console.log" in diff_text or "outputChannel.append" in diff_text or "print(" in diff_text:
+        return f"Add debug logging to {filename}"
+
+    # 2. Detect Error Handling
+    if "try" in diff_text and "catch" in diff_text:
+        return f"Improve error handling in {filename}"
+
+    # 3. Detect Timing / Async Logic
+    if "setTimeout" in diff_text or "debounce" in diff_text or "await" in diff_text:
+        return f"Update async/debounce logic in {filename}"
+
+    # 4. Detect Import Changes
+    if "import " in diff_text or "require(" in diff_text:
+        return f"Update imports in {filename}"
+
+    # 5. Last Resort: Find the function name
+    patterns = [r'function\s+([a-zA-Z0-9_]+)', r'const\s+([a-zA-Z0-9_]+)\s*=']
     for pattern in patterns:
         match = re.search(pattern, diff_text)
         if match:
-            func_name = match.group(1)
-            return f"Refactor '{func_name}' in {filename}"
+            return f"Refactor '{match.group(1)}' logic"
             
     return f"Update logic in {filename}"
 
@@ -50,8 +58,8 @@ def preprocess_diff(diff_text):
     for line in lines:
         if (line.startswith('+') or line.startswith('-')):
             if line.startswith('+++') or line.startswith('---'): continue
-            if "import " in line or "require(" in line: continue
-            if len(line.strip()) < 5: continue 
+            # Don't strip imports here, we might need them for context
+            if len(line.strip()) < 4: continue 
             cleaned_lines.append(line.strip())
     return "\n".join(cleaned_lines)
 
@@ -65,14 +73,13 @@ def sanitize_summary(summary, diff_text, filename):
     if match:
         ticket = match.group()
         if ticket not in diff_text:
-            print(f"⚠️ Hallucination Killed: '{ticket}' -> Switching to Smart Fallback")
+            print(f"⚠️ Hallucination Killed: '{ticket}' -> Using Smart Fallback")
             return get_smart_fallback(diff_text, filename)
 
     # 2. Catch Linguistic Nonsense
     forbidden_words = ["transitive verb", "intransitive", "adjective"]
     for word in forbidden_words:
         if word in summary_clean.lower():
-            print(f"⚠️ Nonsense Killed: '{word}' -> Switching to Smart Fallback")
             return get_smart_fallback(diff_text, filename)
 
     return summary_clean