Final out changes

app.py

@@ -133,28 +133,30 @@ def predict_code_summary(request: CodeRequest):
         
         # Ensure we have the attentions
         if not hasattr(forward_outputs, 'attentions') or not forward_outputs.attentions:
+            # Format response to maintain consistency
             return {
-                "summary": summary,
-                "top_important_words": [],  # No attention data available
-                "message": "Attention data not available from this model"
+                'summary': summary,
+                'topWords': []
             }
         
         # Step 3: Compute word importance
         try:
             word_importance = compute_word_importance(forward_outputs.attentions, inputs)
         except Exception as e:
-            # If word importance calculation fails, still return the summary
+            # Format response to maintain consistency even when word importance fails
             return {
-                "summary": summary,
-                "top_important_words": [],
-                "error_message": f"Failed to compute word importance: {str(e)}"
+                'summary': summary,
+                'topWords': []
             }
         
-        # Final JSON response
-        return {
-            "summary": summary,
-            "top_important_words": word_importance
+        # Format the result according to the specified structure
+        result = {
+            'summary': summary,
+            'topWords': [{'word': word, 'score': float(score)} for word, score in word_importance]
         }
+        
+        # Return formatted JSON response
+        return result
     except Exception as e:
         error_details = {
             "error": str(e),