update

app.py

@@ -630,30 +630,54 @@ def create_app():
                     print("About to process visualization components")
                     viz_components = process_and_visualize_sentiment_analysis(roberta_results)
                     print(f"Visualization components generated: {len(viz_components)}")
-
-                    # Process the visualization
-                    viz_components = process_and_visualize_sentiment_analysis(roberta_results)
                     
                     print("Starting HTML conversion of visualization components")
-                    
-                    # Convert the visualization components to HTML
+
+                    # Convert the visualization components to HTML - OPTIMIZED VERSION
+                    print("Starting HTML conversion of visualization components")
                     html_content = "<div class='sentiment-visualization'>"
                     
-                    for component in viz_components:
-                        if isinstance(component, gr.Markdown):
-                            # Access the Markdown content correctly
-                            html_content += f"<div class='markdown-content'>{str(component)}</div>"
-                        elif isinstance(component, gr.Plot):
-                            # For Plot components, add a placeholder message
-                            html_content += "<div class='plot-placeholder'>Sentiment plot will appear here</div>"
-                        elif isinstance(component, gr.HTML):
-                            # Access HTML content correctly
-                            html_content += f"<div class='html-content'>{str(component)}</div>"
-                        else:
-                            # Handle any other component type
-                            html_content += f"<div class='component'>{str(component)}</div>"
+                    # Use a more efficient and safer approach for component conversion
+                    for i, component in enumerate(viz_components):
+                        print(f"Converting component {i+1}/{len(viz_components)}")
+                        
+                        try:
+                            if isinstance(component, gr.Markdown):
+                                # Simple string representation for Markdown
+                                md_content = str(component).replace('Markdown(value=', '').rstrip(')')
+                                if md_content.startswith("'") and md_content.endswith("'"):
+                                    md_content = md_content[1:-1]
+                                html_content += f"<div class='markdown-content'>{md_content}</div>"
+                            
+                            elif isinstance(component, gr.Plot):
+                                # Simple placeholder for plots
+                                html_content += "<div class='plot-placeholder'>Sentiment visualization chart would appear here</div>"
+                            
+                            elif isinstance(component, gr.HTML):
+                                # Extract HTML content safely
+                                if hasattr(component, 'value'):
+                                    html_content += component.value
+                                else:
+                                    html_str = str(component)
+                                    # Try to extract HTML content from string representation
+                                    if 'value=' in html_str:
+                                        html_value = html_str.split('value=', 1)[1].split(',', 1)[0].strip()
+                                        if html_value.startswith("'") and html_value.endswith("'"):
+                                            html_value = html_value[1:-1]
+                                        html_content += html_value
+                                    else:
+                                        html_content += "<div>HTML component (content unavailable)</div>"
+                            
+                            else:
+                                # Generic fallback for any other component type
+                                html_content += f"<div class='component'>Component: {component.__class__.__name__}</div>"
+                        
+                        except Exception as e:
+                            print(f"Error converting component {i+1}: {str(e)}")
+                            html_content += "<div class='error'>Error rendering component</div>"
                     
                     html_content += "</div>"
+                    print("HTML conversion completed")
                     
                     # Return updated values
                     return (
@@ -663,7 +687,7 @@ def create_app():
                         gr.update(visible=True),   # roberta_viz_title (show title)
                         gr.update(visible=True, value=html_content)  # roberta_viz_content (show content)
                     )
-                    
+                            
                 except Exception as e:
                     import traceback
                     error_msg = f"Error in RoBERTa analysis: {str(e)}\n{traceback.format_exc()}"