Update app.py

app.py

@@ -427,8 +427,8 @@ def generate_code_from_pseudo(pseudo_code, max_length, temperature, top_k, top_p
             try:
                 # Calculate BLEU scores
                 bleu_1, bleu_2, bleu_3, bleu_4 = calculate_bleu_score(reference_code, primary_code)
-            
-            bleu_output = f"""📊 BLEU Scores:
+                
+                bleu_output = f"""📊 BLEU Scores:
 ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
 • BLEU-1 (Unigram): {bleu_1:.4f} ({bleu_1*100:.2f}%)
 • BLEU-2 (Bigram):  {bleu_2:.4f} ({bleu_2*100:.2f}%)
@@ -442,11 +442,11 @@ def generate_code_from_pseudo(pseudo_code, max_length, temperature, top_k, top_p
 • BLEU 0.2-0.3: Fair match
 • BLEU < 0.2: Poor match
 """
-            
-            # Calculate additional metrics
-            code_metrics = calculate_code_metrics(reference_code, primary_code)
-            
-            metrics_output = f"""📈 Additional Metrics:
+                
+                # Calculate additional metrics
+                code_metrics = calculate_code_metrics(reference_code, primary_code)
+                
+                metrics_output = f"""📈 Additional Metrics:
 ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
 • Length Ratio: {code_metrics['length_ratio']:.3f}
 • Precision: {code_metrics['precision']:.4f} ({code_metrics['precision']*100:.2f}%)
@@ -477,18 +477,23 @@ def generate_code_from_pseudo(pseudo_code, max_length, temperature, top_k, top_p
         
         # Format alternative sequences
         alternatives = ""
-        if num_sequences > 1:
+        if num_sequences > 1 and len(generated_codes) > 1:
             alternatives = "🔄 Alternative Generations:\n" + "━"*50 + "\n\n"
             for i, code in enumerate(generated_codes[1:], 2):
-                alternatives += f"Variation {i}:\n```python\n{code}\n```\n\n"
+                # Skip error messages in alternatives
+                if not code.startswith('#'):
+                    alternatives += f"Variation {i}:\n```python\n{code}\n```\n\n"
+                else:
+                    alternatives += f"Variation {i}: {code}\n\n"
         
-        # Add to history
-        generation_history.append({
-            'pseudo': pseudo_code,
-            'generated': primary_code,
-            'bleu_4': bleu_4 if reference_code else None,
-            'time': generation_time
-        })
+        # Add to history (only if primary code is not an error message)
+        if not primary_code.startswith('#'):
+            generation_history.append({
+                'pseudo': pseudo_code,
+                'generated': primary_code,
+                'bleu_4': bleu_4 if reference_code and not primary_code.startswith('#') else None,
+                'time': generation_time
+            })
         
         return primary_code, metrics_output, bleu_output, alternatives