Update app.py

app.py

@@ -98,13 +98,26 @@ def extract_frames(text):
         logging.error(f"Groq API error: {e}")
         return extract_frames_fallback(text)
 
-# Fallback method for frame extraction
+# Fallback method for frame extraction (with categorization of Major, Significant, Minor)
 def extract_frames_fallback(text):
     detected_frames = set()
+    frame_focus = {"Major Focus": [], "Significant Focus": [], "Minor Mention": []}
     text_lower = text.lower()
+    
     for category, keywords in frame_categories.items():
-        if any(word in text_lower for word in keywords):
-            detected_frames.add(category)
+        keyword_count = sum(word in text_lower for word in keywords)
+        if keyword_count > 5:
+            frame_focus["Major Focus"].append(category)
+        elif keyword_count > 2:
+            frame_focus["Significant Focus"].append(category)
+        elif keyword_count > 0:
+            frame_focus["Minor Mention"].append(category)
+    
+    # Return categorized frames
+    for focus, categories in frame_focus.items():
+        for category in categories:
+            detected_frames.add(f"{focus}: {category}")
+    
     return list(detected_frames)
 
 # Extract captions from DOCX
@@ -136,6 +149,7 @@ def extract_metadata_from_excel(excel_file):
             "Number of Audios": row.get("Number of Audios", 0),
             "Likes": row.get("Likes", 0),
             "Comments": row.get("Comments", 0),
+            "Tagged Audience": row.get("Tagged Audience", "No"),
         }
         extracted_data.append(post_data)