Update app.py

app.py

@@ -17,11 +17,20 @@ def generate_knowledge_graph(api_key, user_input):
             }
         ]
     )
-    response_data = completion.choices[0].message.to_dict()
-    response_data = json.loads(response_data['content'])
-
-    print("Dados da resposta:")
-    print(response_data)
+    raw_response = completion.choices[0].message.to_dict()
+    print("Resposta bruta da API:")
+    print(raw_response)
+
+    # Verificar se a resposta contém conteúdo
+    if 'content' in raw_response and raw_response['content']:
+        try:
+            response_data = json.loads(raw_response['content'])
+        except json.JSONDecodeError:
+            print("Erro ao decodificar o JSON.")
+            return "Erro ao decodificar o JSON."
+    else:
+        print("Resposta da API vazia ou inválida.")
+        return "Resposta da API vazia ou inválida."
 
     # Visualizar o conhecimento usando Graphviz
     print("Gerando o conhecimento usando Graphviz...")