Update process.py

process.py

@@ -74,17 +74,23 @@ def analyze_post(post):
 
     result = agent.send_post(post)
     print(result)
-    data = json.loads(result)
-    final = {"J_P":0 ,"T_F":0,  "S_N":0, "E_I": 0 }
-    final['E_I'] = 100 - data["Extrovert-Introvert"]["score"]
-    final['S_N'] = 100 - data["Sensing-Intuition"]["score"]
-    final["T_F"] = 100 - data["Thinking-Feeling"]["score"]
-    final["J_P"] = 100 - data["Judging-Perceiving"]["score"]
+    try:
+        data = json.loads(result)
+        final = {"J_P":0 ,"T_F":0,  "S_N":0, "E_I": 0 }
+        final['E_I'] = 100 - data["Extrovert-Introvert"]["score"]
+        final['S_N'] = 100 - data["Sensing-Intuition"]["score"]
+        final["T_F"] = 100 - data["Thinking-Feeling"]["score"]
+        final["J_P"] = 100 - data["Judging-Perceiving"]["score"]
+    except:
+        final = {}
     return final
     
 
 
 def generate_image(final):
+    if final == {}:
+        st.write("I have no idea about it. Please provide more information")
+        return
     # 初始化画布和坐标轴
     fig, ax = plt.subplots(figsize=(6, 6))