Update app.py

app.py

@@ -84,7 +84,7 @@ class EmotionalAnalyzer:
             plt.close()
             return path
         except Exception:
-            return None
+            return None  # Ensures that if there's an issue, we return None
 
 # --- Text Completion LLM ---
 tokenizer = AutoTokenizer.from_pretrained("diabolic6045/ELN-Llama-1B-base")
@@ -118,7 +118,7 @@ def emotion_aware_response(input_text):
     try:
         analyzer = EmotionalAnalyzer()
         results = analyzer.analyze(input_text)
-        image_path = analyzer.plot_emotions()
+        image_path = analyzer.plot_emotions()  # This could return None if plotting fails
 
         prompt = (
             f"Input: {input_text}\n"
@@ -146,7 +146,7 @@ def emotion_aware_response(input_text):
             f"TextBlob: {results['textblob']}\n\n"
             f"LLM Response:\n{response}"
         )
-        return summary, image_path
+        return summary, image_path if image_path else None
     except Exception:
         return "Error processing emotion-aware response", None