Update app.py

app.py

@@ -1,17 +1,20 @@
-
-from flask import Flask, render_template, request,  url_for, session
+from flask import Flask, render_template, request, url_for, session
 import random
 import os
 from proof_generator import generate_fake_proof
 from datetime import datetime
 from googletrans import Translator
-
-
+from werkzeug.utils import secure_filename
+import uuid
 
 app = Flask(__name__)
 app.secret_key = "your-secret-key"
 translator = Translator()
 
+# Configure upload folder
+app.config['UPLOAD_FOLDER'] = 'static/uploads'
+if not os.path.exists(app.config['UPLOAD_FOLDER']):
+    os.makedirs(app.config['UPLOAD_FOLDER'])
 
 translations = {
     'en': {
@@ -26,8 +29,8 @@ translations = {
         'family': "Family",
         'history': "History",
         'favorites': "Favorites",
-         'your_excuse': "Your Excuse:",
-        'fake_proof': "Fake Proof:",
+        'your_excuse': "Your Excuse:",
+        'fake_proof': "Proof:",
         'top_excuses': "Top Excuses:",
         'prediction': "Prediction: Based on your history, you may need your next excuse on:",
         'play_excuse': "▶️ Play Excuse"
@@ -45,7 +48,7 @@ translations = {
         'history': "Historial",
         'favorites': "Favoritos",
         'your_excuse': "Tu excusa:",
-        'fake_proof': "Prueba falsa:",
+        'fake_proof': "Prueba:",
         'top_excuses': "Principales excusas:",
         'prediction': "Predicción: Según tu historial, podrías necesitar tu próxima excusa el:",
         'play_excuse': "▶️ Reproducir excusa"
@@ -62,8 +65,8 @@ translations = {
         'family': "Famille",
         'history': "Historique",
         'favorites': "Favoris",
-         'your_excuse': "Votre excuse:",
-        'fake_proof': "Fausse preuve:",
+        'your_excuse': "Votre excuse:",
+        'fake_proof': "Preuve:",
         'top_excuses': "Meilleures excuses:",
         'prediction': "Prévision : Basé sur votre historique, vous pourriez avoir besoin de votre prochaine excuse le:",
         'play_excuse': "▶️ Lire l'excuse"
@@ -80,8 +83,8 @@ translations = {
         'family': "परिवार",
         'history': "इतिहास",
         'favorites': "पसंदीदा",
-         'your_excuse': "आपका बहाना:",
-        'fake_proof': "नकली सबूत:",
+        'your_excuse': "आपका बहाना:",
+        'fake_proof': "सबूत:",
         'top_excuses': "शीर्ष बहाने:",
         'prediction': "भविष्यवाणी: आपके इतिहास के आधार पर, अगला बहाना आपको इस दिन ज़रूरत पड़ सकती है:",
         'play_excuse': "▶️ बहाना सुनें"
@@ -98,16 +101,14 @@ translations = {
         'family': "కుటుంబం",
         'history': "చరిత్ర",
         'favorites': "ఇష్టమైనవి",
-         'your_excuse': "మీ నెపం:",
-        'fake_proof': "నకిలీ ఆధారం:",
+        'your_excuse': "మీ నెపం:",
+        'fake_proof': "ఆధారం:",
         'top_excuses': "అగ్ర నెపాలు:",
         'prediction': "అంచనా: మీ చరిత్ర ఆధారంగా, మీకు వచ్చే నెపం అవసరం అయ్యే రోజు:",
         'play_excuse': "▶️ నెపం వినండి"
     }
 }
 
-
-# --- Excuse + Proof Mappings ---
 excuse_proofs = {
     "Work": [
         ("I'm feeling unwell and didn’t want to risk spreading anything.", "proof6.jpg"),
@@ -135,7 +136,6 @@ excuse_proofs = {
     ]
 }
 
-# --- Apology Templates ---
 apology_templates = {
     "Work": [
         "I'm truly sorry for missing work. It wasn’t my intention to let anyone down.",
@@ -158,12 +158,10 @@ apology_templates = {
         "Please know I regret not showing up — it wasn’t intentional."
     ]
 }
-top_excuses = []
-scheduled_excuses = []
 
-def rank_excuses(excuse_list):
+def rank_excuses(excuse_proof_list):
     ranked = []
-    for excuse in excuse_list:
+    for excuse, proof in excuse_proof_list:
         score = 0
         if "emergency" in excuse.lower():
             score += 5
@@ -175,11 +173,9 @@ def rank_excuses(excuse_list):
             score += 2
         if "family" in excuse.lower():
             score += 1
-        
-        ranked.append((score, excuse))
-
-    ranked.sort(reverse=True)
-    return [exc for score, exc in ranked]
+        ranked.append((score, (excuse, proof)))
+    ranked.sort(reverse=True, key=lambda x: x[0])
+    return [item[1] for item in ranked]
 
 def predict_next_excuse_time(history):
     from collections import Counter
@@ -189,118 +185,91 @@ def predict_next_excuse_time(history):
         return "No data to predict."
     
     valid_entries = [entry for entry in history if isinstance(entry, dict) and "timestamp" in entry]
-
     if not valid_entries:
         return "No valid history entries."
 
-    times = [datetime.strptime(entry["timestamp"], "%Y-%m-%d %H:%M:%S") for entry in history]
-    days_of_week = [time.weekday() for time in times]  # 0 = Monday
-
-    # Most common day
+    times = [datetime.strptime(entry["timestamp"], "%Y-%m-%d %H:%M:%S") for entry in valid_entries]
+    days_of_week = [time.weekday() for time in times]
     common_day = Counter(days_of_week).most_common(1)[0][0]
-
-    # Estimate next occurrence of that weekday
     today = datetime.now()
     days_until_next = (common_day - today.weekday()) % 7
     predicted_date = today + timedelta(days=days_until_next)
-
-    # Predict similar time based on average hour
     avg_hour = sum(time.hour for time in times) / len(times)
     predicted_time = predicted_date.replace(hour=int(avg_hour), minute=0, second=0, microsecond=0)
-
     return predicted_time.strftime("%A, %Y-%m-%d at %I:%M %p")
 
-
 @app.route('/', methods=['GET', 'POST'])
 def index():
-    
-    selected_language = request.form.get('language')
-    if not selected_language:
-        selected_language = request.args.get('lang')
-    if not selected_language:
-        selected_language = 'en-US'
-    lang_code = selected_language.split('-')[0]  # en-US -> en
+    selected_language = request.form.get('language') or request.args.get('lang') or 'en-US'
+    lang_code = selected_language.split('-')[0]
     language_dict = translations.get(lang_code, translations['en'])
 
-
     if "history" not in session:
         session["history"] = []
     if "favorites" not in session:
         session["favorites"] = []
 
-    session["history"] = []
-
     excuse = None
     proof = None
+    proof_label = ""
     top_3_excuses = []
-  
-    #selected_language = request.form.get('language', 'en-US') # default language
-
 
     if request.method == 'POST':
         scenario = request.form.get('scenario')
         urgency = request.form.get('urgency')
         selected_language = request.form.get('language', 'en-US')
-    
-        if scenario in excuse_proofs:
-    # Fetch all excuses for the selected scenario
-           all_excuses_proofs = excuse_proofs[scenario]
-
-    # Extract only excuses (texts) for ranking
-           all_excuses = [exc for exc, proof_file in all_excuses_proofs]
-
-    # Rank excuses
-           ranked_excuses = rank_excuses(all_excuses)
-
-    # Pick the best excuse (top-ranked)
-           best_excuse = ranked_excuses[0]
-           top_3_excuses = ranked_excuses[:3] 
 
-    # Find corresponding proof image
-           for exc, proof_filename in all_excuses_proofs:
-               if exc == best_excuse:
-                   proof = url_for('static', filename=f'proofs/{proof_filename}')
-                   break
-
-           excuse = best_excuse
-
-        else:
-            excuse = f"Sorry, I couldn't attend due to a {scenario} emergency."
-            proof = url_for('static', filename='proofs/default_proof.jpg')
-
-        # --- Translate excuse based on selected language ---
-        target_lang = selected_language.split('-')[0]  # en-US -> en
-        if target_lang != 'en':  # Only translate if not English
-            try:
-               translation = translator.translate(excuse, dest=target_lang)
-               excuse = translation.text
-            except Exception as e:
-               print("Translation error:", e)
-        # If translation fails, keep the original English excuse
-        
-         # Save excuse to history with timestamp
-        session["history"].append({
-            "excuse": excuse,
-            "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        })
-        session.modified = True
+        if scenario in excuse_proofs:
+            all_excuses_proofs = excuse_proofs[scenario]
+            ranked_excuses_proofs = rank_excuses(all_excuses_proofs)
+            top_3_excuses_proofs = ranked_excuses_proofs[:3]
+            if top_3_excuses_proofs:
+                selected_excuse, selected_proof_filename = random.choice(top_3_excuses_proofs)
+            else:
+                selected_excuse, selected_proof_filename = ranked_excuses_proofs[0]
+            excuse = selected_excuse
+            top_3_excuses = [exc for exc, proof in ranked_excuses_proofs[:3]]
+
+            # Handle proof upload
+            if 'proof_image' in request.files and request.files['proof_image'].filename != '':
+                file = request.files['proof_image']
+                ext = os.path.splitext(file.filename)[1]
+                filename = str(uuid.uuid4()) + ext
+                file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
+                proof = url_for('static', filename=f'uploads/{filename}')
+                is_fake = random.choice(["fake", "real"])
+                proof_label = f"This proof is likely {is_fake}"
+            else:
+                proof = url_for('static', filename=f'proofs/{selected_proof_filename}')
+                proof_label = ""
+
+            # Translate excuse if not English
+            target_lang = selected_language.split('-')[0]
+            if target_lang != 'en':
+                try:
+                    translation = translator.translate(excuse, dest=target_lang)
+                    excuse = translation.text
+                except Exception as e:
+                    print("Translation error:", e)
+
+            # Save to history
+            session["history"].append({
+                "excuse": excuse,
+                "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+            })
+            session.modified = True
 
-    language_dict = translations.get(selected_language[:2], translations['en'])
     predicted_schedule = predict_next_excuse_time(session.get("history", []))
-
-        
-    return render_template('index.html', excuse=excuse, proof=proof,selected_language=selected_language, language_dict=language_dict,top_excuses=top_3_excuses, predicted_schedule=predicted_schedule)
-
+    return render_template('index.html', excuse=excuse, proof=proof, proof_label=proof_label,
+                           selected_language=selected_language, language_dict=language_dict,
+                           top_excuses=top_3_excuses, predicted_schedule=predicted_schedule)
 
 @app.route("/history")
 def history():
     lang = request.args.get('lang', 'en')
     language_dict = translations.get(lang, translations['en'])
-
-    return render_template("history.html",
-                           history=session.get("history", []),
-                           favorites=session.get("favorites", []),
-                           language_dict=language_dict)
+    return render_template("history.html", history=session.get("history", []),
+                           favorites=session.get("favorites", []), language_dict=language_dict)
 
 @app.route("/favorite", methods=["POST"])
 def favorite():
@@ -308,8 +277,7 @@ def favorite():
     if excuse and excuse not in session["favorites"]:
         session["favorites"].append(excuse)
         session.modified = True
-    return (url_for("history"))
-
+    return url_for("history")
 
 @app.route('/apology', methods=['GET', 'POST'])
 def apology():
@@ -318,12 +286,10 @@ def apology():
     lang = request.args.get('lang', 'en')
     language_dict = translations.get(lang, translations['en'])
 
-
     if request.method == 'POST':
         scenario = request.form.get('scenario')
         urgency = request.form.get("urgency")
         tone = request.form.get('tone')
-
         if scenario in apology_templates:
             apology_message = random.choice(apology_templates[scenario])
             proof = generate_fake_proof(urgency)
@@ -331,4 +297,4 @@ def apology():
     return render_template('apology.html', apology=apology_message, proof=proof, language_dict=language_dict)
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860, debug=True)
+    app.run(host='0.0.0.0', port=7860, debug=True)