Update app.py

app.py

@@ -14,131 +14,99 @@ def get_gemini_response(prompt):
     genai.configure(api_key=api_key)
     model = genai.GenerativeModel("gemini-2.0-flash")
     try:
-        response = model.generate_content(prompt)
+        response = model.generate_content(prompt + " Use plain text format without markdown, avoid symbols like **, #, or any special formatting.")
         return response.text
     except Exception as e:
         return f"❌ Gemini API Error: {str(e)}"
 
-# Remove or replace Unicode characters not supported by FPDF
+# Enhanced text sanitization
 def sanitize_text(text):
-    text = text.replace("–", "-").replace("’", "'").replace("“", '"').replace("”", '"')
-    text = re.sub(r"[^\x00-\x7F]+", "", text)  # remove other non-ASCII characters
-    return text
-
-# Custom PDF class with header and footer
+    # Replace problematic Unicode characters
+    replacements = {
+        "–": "-", "’": "'", "“": '"', "”": '"',
+        "•": "*", "●": "*", "": "*", "⋯": "..."
+    }
+    for k, v in replacements.items():
+        text = text.replace(k, v)
+    
+    # Remove markdown and special formatting
+    text = re.sub(r'\*\*|\*|__|_|~~|`|#+', '', text)  # Remove markdown symbols
+    text = re.sub(r'\[(.*?)\]\(.*?\)', r'\1', text)   # Remove links but keep text
+    text = re.sub(r'<.*?>', '', text)                 # Remove HTML tags
+    text = re.sub(r'\n{3,}', '\n\n', text)            # Reduce excessive newlines
+    text = re.sub(r'[^\x00-\x7F]+', ' ', text)        # Replace non-ASCII with space
+    return text.strip()
+
+# Improved PDF class with better formatting
 class PDF(FPDF):
     def header(self):
         self.set_font("Arial", "B", 12)
         self.set_text_color(40, 40, 40)
-        self.multi_cell(0, 10, '"Success is where preparation and opportunity meet." - Bobby Unser', align='C')
+        self.cell(0, 10, '"Success is where preparation and opportunity meet." - Bobby Unser', 0, 1, 'C')
         self.ln(5)
 
     def footer(self):
         self.set_y(-15)
         self.set_font("Arial", "I", 8)
         self.set_text_color(100, 100, 100)
-        # Added your contact information here
         footer_text = f"Generated on {datetime.now().strftime('%Y-%m-%d %H:%M:%S')} | Created by Mahar Affandi Noor Ghazi - Pace GK Academy | Contact: +92 311 750 5369"
-        self.cell(0, 10, footer_text, align='C')
+        self.cell(0, 10, footer_text, 0, 0, 'C')
 
 def create_pdf(interview_text, note_text, name):
     pdf = PDF()
     pdf.add_page()
-    pdf.set_font("Times", "B", 16)
-    pdf.cell(0, 10, f"Mock Interview: {name}", ln=True, align='C')
-    pdf.ln(10)
-
+    
+    # Main title
+    pdf.set_font("Times", "B", 18)
+    pdf.cell(0, 10, f"Mock Interview: {name}", ln=1, align='C')
+    pdf.ln(12)
+    
+    # Interview questions section
+    pdf.set_font("Times", "B", 14)
+    pdf.set_text_color(0, 0, 139)  # Dark blue
+    pdf.cell(0, 10, "Interview Questions", ln=1)
+    pdf.ln(5)
+    
+    # Process interview text
     pdf.set_font("Times", "", 12)
+    pdf.set_text_color(0, 0, 0)  # Black
     interview_text = sanitize_text(interview_text)
-    lines = interview_text.split("\n")
-    for line in lines:
-        pdf.multi_cell(0, 8, line)
-        pdf.ln(1)
-
-    pdf.ln(10)
+    
+    # Format numbered questions
+    for line in interview_text.split('\n'):
+        line = line.strip()
+        if re.match(r'^\d+[\.\)]', line):  # Detect numbered questions
+            pdf.set_font("Times", "B", 12)
+            pdf.multi_cell(0, 8, line)
+            pdf.set_font("Times", "", 12)
+        else:
+            pdf.multi_cell(0, 8, line)
+        pdf.ln(4)
+    
+    # Recommendations section
+    pdf.add_page()
     pdf.set_font("Times", "B", 14)
-    pdf.cell(0, 10, "Special Note & Recommendations", ln=True)
+    pdf.set_text_color(0, 0, 139)  # Dark blue
+    pdf.cell(0, 10, "Special Note & Recommendations", ln=1)
+    pdf.ln(8)
+    
+    # Process recommendations
     pdf.set_font("Times", "", 12)
-
+    pdf.set_text_color(0, 0, 0)  # Black
     note_text = sanitize_text(note_text)
-    note_lines = note_text.split("\n")
-    for line in note_lines:
-        pdf.multi_cell(0, 8, line)
-        pdf.ln(1)
-
-    # Added your credentials at the end of the document
-    pdf.ln(15)
-    pdf.set_font("Times", "I", 10)
-    pdf.set_text_color(0, 0, 128)  # Navy blue color
-    pdf.multi_cell(0, 6, "Prepared by:\nMahar Affandi Noor Ghazi\nPace GK Academy\nContact: +92 311 750 5369", align='C')
-
+    pdf.multi_cell(0, 8, note_text)
+    pdf.ln(10)
+    
+    # Creator credentials
+    pdf.set_font("Times", "I", 11)
+    pdf.set_text_color(0, 0, 128)  # Navy blue
+    pdf.multi_cell(0, 8, "Prepared by:\nMahar Affandi Noor Ghazi\nPace GK Academy\nContact: +92 311 750 5369", align='C')
+    
     filename = f"{name.replace(' ', '_')}_interview.pdf"
     pdf.output(filename)
     return filename
 
-# Rest of the Streamlit UI code remains the same...
-# (The Streamlit UI section below this point remains unchanged from your original code)
-
-# Streamlit UI
-st.set_page_config(page_title="Interview Generator", layout="centered")
-st.title("📚 Mock Interview Generator for 🇵🇰 Competitive Exams")
-
-with st.form("interview_form"):
-    name = st.text_input("Full Name")
-    father_name = st.text_input("Father's Name")
-    district = st.text_input("District of Domicile")
-    tehsil = st.text_input("Tehsil")
-    bachelors = st.text_input("Bachelor's Degree (e.g., BSc Physics)")
-    masters = st.text_input("Master's Degree (e.g., MSc Public Admin or 'None')")
-    department_post = st.text_input("Department and Post (e.g., Education Dept - Lecturer English)")
-    exam_type = st.selectbox("Exam Type", ["PPSC", "FPSC", "CSS", "PMS", "Other"])
-    hobby = st.text_input("Hobby")
-    fav_personality = st.text_input("Favorite Personality")
-    submitted = st.form_submit_button("Generate Interview")
-
-if submitted:
-    if all([name, father_name, district, tehsil, bachelors, masters, department_post, exam_type, hobby, fav_personality]):
-        # Main interview prompt
-        interview_prompt = f"""
-        Generate a detailed mock interview based on the following Pakistani candidate's profile:
-        - Name: {name}
-        - Father's Name: {father_name}
-        - District: {district}
-        - Tehsil: {tehsil}
-        - Bachelor's Degree: {bachelors}
-        - Master's Degree: {masters}
-        - Exam Type: {exam_type}
-        - Department and Post: {department_post}
-        - Hobby: {hobby}
-        - Favorite Personality: {fav_personality}
-        Include questions from:
-        - Name-based personalities
-        - District/Tehsil-specific geography, history, administration
-        - Educational background
-        - Post/Department-specific issues
-        - Pakistan's geography, national & international affairs, sports
-        - Government projects related to education
-        - Situational judgment and problem-solving
-        Format all in clear, numbered interview questions.
-        """
-
-        # Recommendation note prompt
-        note_prompt = f"""
-        Based on the profile of a candidate applying for the post of {department_post} through {exam_type},
-        give a special note with motivational advice and suggest specific areas the candidate should study.
-        Keep the tone encouraging and professional.
-        """
-
-        with st.spinner("Generating your personalized interview..."):
-            interview_text = get_gemini_response(interview_prompt)
-            note_text = get_gemini_response(note_prompt)
-
-        st.success("✅ Interview Generated Successfully!")
-        st.text_area("📋 Interview Questions", interview_text, height=400)
-        st.text_area("📝 Special Note", note_text, height=200)
+# Streamlit UI remains the same...
+# (Keep your existing Streamlit interface code here without changes)
 
-        file_path = create_pdf(interview_text, note_text, name)
-        with open(file_path, "rb") as f:
-            st.download_button("📄 Download PDF with Recommendations", f, file_name=file_path, mime="application/pdf")
-    else:
-        st.warning("⚠️ Please complete all fields to proceed.")
+# Rest of your Streamlit code remains identical...