Update app.py

app.py

@@ -5,8 +5,6 @@ from gtts import gTTS
 import tempfile
 import re
 from deep_translator import GoogleTranslator
-import json
-import io
 
 st.title("News Summarization and Text-to-Speech Application")
 
@@ -15,99 +13,69 @@ company_name = st.text_input("Enter the company name:", "").strip().lower()
 
 if st.button("Fetch News"):
     if company_name:
-        with st.status("Fetching news...", expanded=True) as status:
-            st.write(f"Fetching news for **{company_name}**...")
+        # Run news extraction and analysis
+        st.write(f"Fetching news for **{company_name}**...")
+        
+        # Call the function from utils.py
+        file_name = utils.fetch_and_save_news(company_name)
+        
+        if os.path.exists(file_name):
+            st.success(f"Data saved in **{file_name}**")
             
-            try:
-                # Call the utils function directly
-                file_name = utils.fetch_and_save_news(company_name)
+            # Read the file to display content
+            with open(file_name, "r", encoding="utf-8") as file:
+                text_content = file.read()
+                st.text_area("News Analysis", text_content, height=400)
+            
+            # Provide a download button for text file
+            with open(file_name, "rb") as file:
+                st.download_button(
+                    label="Download Text File",
+                    data=file,
+                    file_name=file_name,
+                    mime="text/plain"
+                )
+            
+            # Extract only the Final Sentiment Analysis line
+            final_sentiment_line = ""
+            with open(file_name, "r", encoding="utf-8") as file:
+                content = file.read()
+                # Use regular expression to find the Final Sentiment Analysis line
+                match = re.search(r'"Final Sentiment Analysis": "([^"]+)"', content)
+                if match:
+                    final_sentiment_line = match.group(1)
+            
+            if final_sentiment_line:
+                st.subheader("Hindi Audio for Final Sentiment Analysis")
                 
-                if not file_name:
-                    status.update(label="No news found", state="error")
-                    st.warning(f"No news found for {company_name}")
-                else:
-                    # Read the saved JSON file and parse it
-                    with open(file_name, "r", encoding="utf-8") as file:
-                        news_data = json.load(file)
-                    
-                    status.update(label="News fetched successfully!", state="complete", expanded=False)
+                try:
+                    # First translate the English text to Hindi using deep_translator
+                    translator = GoogleTranslator(source='en', target='hi')
+                    hindi_text = translator.translate(final_sentiment_line)
                     
-                    # Display news data
-                    st.subheader(f"News Analysis for {news_data['Company']}")
+                    # Create Hindi audio from the translated text
+                    tts = gTTS(text=hindi_text, lang='hi', slow=False)
                     
-                    # Articles section
-                    st.subheader("Articles")
-                    for i, article in enumerate(news_data['Articles']):
-                        with st.expander(f"Article {i+1}: {article['Title']}", expanded=False):
-                            st.markdown(f"**Summary:** {article['Summary']}")
-                            st.markdown(f"**Sentiment:** {article['Sentiment']}")
-                            st.markdown(f"**Topics:** {', '.join(article['Topics'])}")
+                    # Save the audio in a temporary file
+                    temp_audio_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
+                    tts.save(temp_audio_file.name)
                     
-                    # Sentiment Distribution
-                    st.subheader("Sentiment Distribution")
-                    sentiment_data = news_data['Comparative Sentiment Score']['Sentiment Distribution']
-                    col1, col2, col3 = st.columns(3)
-                    col1.metric("Positive", sentiment_data['Positive'])
-                    col2.metric("Neutral", sentiment_data['Neutral'])
-                    col3.metric("Negative", sentiment_data['Negative'])
                     
-                    # Topic Analysis
-                    st.subheader("Topic Analysis")
-                    st.markdown("**Common Topics:**")
-                    st.write(", ".join(news_data['Topic Overlap']['Common Topics']))
-                    for key, value in news_data['Topic Overlap'].items():
-                        if key != "Common Topics":
-                            st.markdown(f"**{key}:**")
-                            st.write(", ".join(value))
                     
-                    # Coverage Differences
-                    st.subheader("Coverage Differences")
-                    coverage_diff = news_data['Coverage Differences']
-                    if isinstance(coverage_diff, str):
-                        st.write(coverage_diff)  # Fallback for error cases
-                    else:
-                        formatted_text = '"Coverage Differences": [\n'
-                        for i, item in enumerate(coverage_diff.get("Coverage Differences", [])):
-                            formatted_text += "{\n"
-                            formatted_text += f'    "Comparison": "{item["Comparison"]}",\n'
-                            formatted_text += f'    "Impact": "{item["Impact"]}"\n'
-                            formatted_text += "}" + (",\n" if i < len(coverage_diff["Coverage Differences"]) - 1 else "\n")
-                        formatted_text += "]"
-                        st.code(formatted_text, language="json")
-                    
-                    # Final Sentiment Analysis
-                    st.subheader("Final Sentiment Analysis")
-                    st.info(news_data['Final Sentiment Analysis'])
-                    
-                    # Download JSON
-                    st.subheader("Download Data")
-                    st.download_button(
-                        label="Download JSON File",
-                        data=json.dumps(news_data, indent=4),
-                        file_name=f"{company_name}_news.json",
-                        mime="application/json"
-                    )
-                    
-                    # Hindi Audio
-                    st.subheader("Hindi Audio for Final Sentiment Analysis")
-                    try:
-                        hindi_text = translator.translate(news_data['Final Sentiment Analysis'])
-                        tts = gTTS(text=hindi_text, lang='hi')
-                        audio_io = io.BytesIO()
-                        tts.write_to_fp(audio_io)
-                        audio_io.seek(0)
-                        audio_bytes = audio_io.read()
+                    # Provide download button for the audio
+                    with open(temp_audio_file.name, "rb") as audio_file:
+                        audio_data = audio_file.read()
                         st.download_button(
                             label="Download Hindi Audio",
-                            data=audio_bytes,
+                            data=audio_data,
                             file_name=f"{company_name}_sentiment_hindi.mp3",
                             mime="audio/mp3"
                         )
-                    except Exception as e:
-                        st.error(f"Error generating Hindi audio: {str(e)}")
-            
-            except Exception as e:
-                status.update(label="Processing error", state="error")
-                st.error(f"Error processing news data: {str(e)}")
+                except Exception as e:
+                    st.error(f"Error generating Hindi audio: {str(e)}")
+            else:
+                st.warning("Could not find Final Sentiment Analysis in the text.")
+        else:
+            st.error("No relevant news articles found.")
     else:
-        st.warning("Please enter a company name.")
+        st.warning("Please enter a company name.")