Update app.py

app.py

@@ -54,10 +54,14 @@ def analyze_attrition_with_llm(df_dict, hr_query):
         ],
         function_call="auto"
     )
+    print(response)  # Debugging: Print response
 
-    # Directly return the full response as a formatted string
-    return response.choices[0].message.content if response.choices[0].message.content else "🤖 No response generated."
-
+    # Handle function call response properly
+    if response.choices[0].message.function_call:
+        function_args = json.loads(response.choices[0].message.function_call.arguments)
+        return predict_attrition_risk(**function_args)
+    else:
+        return "🤖 No response generated."
 
 with gr.Blocks() as demo:
     gr.Markdown("<h1>AI-Driven Employee Attrition Risk Analysis</h1>")