Update process_interview.py

process_interview.py

@@ -25,7 +25,7 @@ from reportlab.lib.units import inch
 from reportlab.lib import colors
 import matplotlib.pyplot as plt
 import matplotlib
-matplotlib.use('Agg') # --- FIX: تحديد backend لـ matplotlib ---
+matplotlib.use('Agg')
 from reportlab.platypus import Image
 import io # --- FIX: إضافة import io لـ BytesIO ---
 # --- End Imports for enhanced PDF ---
@@ -273,7 +273,7 @@ def identify_speakers(transcript: Dict, wav_file: str) -> List[Dict]:
 
 def train_role_classifier(utterances: List[Dict]):
     try:
-        texts = [u['text'] for u in utterances]
+        texts = [u['text'] for u u in utterances]
         vectorizer = TfidfVectorizer(max_features=500, ngram_range=(1, 2))
         X_text = vectorizer.fit_transform(texts)
 
@@ -779,18 +779,16 @@ def create_pdf_report(analysis_data: Dict, output_path: str, gemini_report_text:
 
             # --- Charts ---
             story.append(Paragraph("Score Visualization:", h3))
-            chart_path = os.path.join(OUTPUT_DIR, f"anxiety_confidence_{uuid.uuid4().hex[:8]}.png")
-            # --- FIX: Call generate_anxiety_confidence_chart if it is defined and imports are correct ---
+            # chart_path = os.path.join(OUTPUT_DIR, f"anxiety_confidence_{uuid.uuid4().hex[:8]}.png") # Removed from here
+            # --- FIX: Generate chart in memory (BytesIO) ---
+            chart_buffer = io.BytesIO() # Create in-memory buffer
             try:
-                # The generate_anxiety_confidence_chart function is now expected to be defined.
-                # It relies on matplotlib and Image (from reportlab.platypus)
-                generate_anxiety_confidence_chart(voice_analysis['composite_scores'], chart_path)
-                if os.path.exists(chart_path):
-                    img = Image(chart_path, width=3.5*inch, height=2.0*inch)
-                    story.append(img)
-                    story.append(Spacer(1, 0.1 * inch))
-                    os.remove(chart_path)
-            except NameError: # Catch NameError if function is truly not defined
+                generate_anxiety_confidence_chart(voice_analysis['composite_scores'], chart_buffer) # Pass buffer instead of path
+                chart_buffer.seek(0) # Rewind the buffer to the beginning
+                img = Image(chart_buffer, width=3.5*inch, height=2.0*inch) # Load image from buffer
+                story.append(img)
+                story.append(Spacer(1, 0.1 * inch))
+            except NameError:
                 logger.warning("Chart generation function 'generate_anxiety_confidence_chart' is not defined. Skipping chart.")
             except Exception as chart_e:
                 logger.warning(f"Could not add chart image to PDF: {chart_e}. Please check matplotlib installation.")