Update app.py

app.py

@@ -1,33 +1,30 @@
 import os
 import streamlit as st
+from langchain_community.llms import HuggingFaceHub
 from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
-from langchain_community.llms import HuggingFaceHub
 import matplotlib.pyplot as plt
-from io import BytesIO
-from reportlab.pdfgen import canvas
-from reportlab.lib.pagesizes import letter
+from fpdf import FPDF
+import tempfile
 
-# Load Hugging Face token
+# Set Hugging Face token from secrets
 os.environ["HUGGINGFACEHUB_API_TOKEN"] = st.secrets["HF_TOKEN"]
 
-# Initialize LLM
+# Load LLM from Hugging Face
 llm = HuggingFaceHub(
     repo_id="google/flan-t5-xl",
-    model_kwargs={"temperature": 0.5, "max_length": 1024}
+    model_kwargs={"temperature": 0.7, "max_length": 1024}
 )
 
 # Prompt Template
-template = """
-Compare the following two products or services:
+template = """Compare the following two products/services:
 Product 1: {product1}
 Product 2: {product2}
 
-1. Feature-by-feature comparison.
-2. SWOT analysis for each.
-3. A comparative summary highlighting key differentiators.
-
-Respond in structured format with headings.
+Provide:
+1. A feature-by-feature comparison (up to 5 features).
+2. SWOT analysis for both.
+3. A comparative summary highlighting key differences and similarities.
 """
 
 prompt = PromptTemplate(
@@ -35,61 +32,56 @@ prompt = PromptTemplate(
     template=template,
 )
 
+# LLM Chain
 comparison_chain = LLMChain(llm=llm, prompt=prompt)
 
 # Streamlit UI
-st.title("🔍 Competitive Analysis Tool (Hugging Face Version)")
-
-product1 = st.text_area("Enter details for Product/Service 1")
-product2 = st.text_area("Enter details for Product/Service 2")
+st.title("🔍 AI Competitive Analysis Tool")
+product1 = st.text_area("Enter Product 1 description")
+product2 = st.text_area("Enter Product 2 description")
+
+def generate_pdf(text, filename):
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_auto_page_break(auto=True, margin=15)
+    pdf.set_font("Arial", size=12)
+    for line in text.split('\n'):
+        pdf.multi_cell(0, 10, line)
+    pdf.output(filename)
 
 if st.button("Compare"):
-    with st.spinner("Generating analysis..."):
-        try:
+    if not product1 or not product2:
+        st.warning("Please enter both products.")
+    else:
+        with st.spinner("Analyzing..."):
             result = comparison_chain.run(product1=product1, product2=product2)
-            st.subheader("📊 Analysis Result")
+            st.success("Comparison complete!")
+
+            st.subheader("📋 Analysis Report")
             st.markdown(result)
 
-            # Optional: Simple chart visualization (Dummy feature scores)
-            features = ["Usability", "Performance", "Support", "Integration"]
-            p1_scores = [8, 7, 6, 9]
-            p2_scores = [7, 8, 8, 7]
+            # Optional Chart (dummy data, just for visual comparison)
+            st.subheader("📊 Feature Comparison Chart")
+            features = ["Ease of Use", "Performance", "Cost", "Support", "Scalability"]
+            values1 = [3, 4, 2, 3, 4]  # Dummy data
+            values2 = [4, 3, 3, 4, 3]  # Dummy data
 
             fig, ax = plt.subplots()
-            bar_width = 0.35
-            index = range(len(features))
-
-            ax.bar(index, p1_scores, bar_width, label="Product 1")
-            ax.bar([i + bar_width for i in index], p2_scores, bar_width, label="Product 2")
-
-            ax.set_xlabel('Features')
-            ax.set_ylabel('Scores')
-            ax.set_title('Feature Comparison')
-            ax.set_xticks([i + bar_width / 2 for i in index])
+            x = range(len(features))
+            ax.bar(x, values1, width=0.4, label="Product 1", align='center')
+            ax.bar([p + 0.4 for p in x], values2, width=0.4, label="Product 2", align='center')
+            ax.set_xticks([p + 0.2 for p in x])
             ax.set_xticklabels(features)
             ax.legend()
-
             st.pyplot(fig)
 
             # PDF Download
-            def create_pdf(text):
-                buffer = BytesIO()
-                c = canvas.Canvas(buffer, pagesize=letter)
-                width, height = letter
-                y = height - 40
-
-                for line in text.split('\n'):
-                    c.drawString(30, y, line)
-                    y -= 15
-                    if y < 50:
-                        c.showPage()
-                        y = height - 40
-                c.save()
-                buffer.seek(0)
-                return buffer
-
-            pdf_data = create_pdf(result)
-            st.download_button("📄 Download PDF Report", data=pdf_data, file_name="analysis_report.pdf")
-
-        except Exception as e:
-            st.error(f"Something went wrong: {e}")
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_file:
+                generate_pdf(result, tmp_file.name)
+                with open(tmp_file.name, "rb") as f:
+                    st.download_button(
+                        label="📄 Download PDF Report",
+                        data=f,
+                        file_name="comparison_report.pdf",
+                        mime="application/pdf"
+                    )