Enhanced frontend

app.py

@@ -147,19 +147,19 @@ def main():
                 # In the main() function:
 
                 # Display metrics in columns
-                col1, col2, col3 = st.columns(3)
+                col1, col2 = st.columns(2)
 
                 with col1:
-                    st.subheader("ML Model Analysis")
+                    st.subheader("ML Model and Knowedge Graph Analysis")
                     st.metric("Prediction", ml_prediction)
                     st.metric("Confidence", f"{ml_confidence:.2f}%")
 
-                with col2:
-                    st.subheader("Knowledge Graph Analysis")
-                    st.metric("Prediction", kg_prediction)
-                    st.metric("Confidence", f"{kg_confidence:.2f}%")
+                # with col2:
+                #     st.subheader("Knowledge Graph Analysis")
+                #     st.metric("Prediction", kg_prediction)
+                #     st.metric("Confidence", f"{kg_confidence:.2f}%")
 
-                with col3:
+                with col2:
                     st.subheader("Gemini Analysis")
                     gemini_pred = gemini_result["gemini_analysis"]["predicted_classification"]
                     gemini_conf = gemini_result["gemini_analysis"]["confidence_score"]
@@ -211,17 +211,6 @@ def main():
                     except Exception as e:
                         st.error("Error processing analysis results")
 
-
-
-                with st.expander("Named Entities"):
-                    entities = extract_entities(news_text, nlp)
-                    df = pd.DataFrame(entities, columns=["Entity", "Type"])
-                    st.dataframe(df)
-
-                 # Generate and display knowledge graph
-                fig = generate_knowledge_graph_viz(news_text, nlp, tokenizer, model)
-                st.plotly_chart(fig, use_container_width=True)
-
         else:
             st.warning("Please enter some text to analyze")