Update utils.py

utils.py

@@ -318,7 +318,8 @@ def fetch_and_save_news(company_name):
     if not company_name:
         print("❌ Error: Company name is required")
         return None
-    file_name = f"{company_name}_news.txt"
+    
+    file_name = f"{company_name}_news.json"
     articles = []
     article_count = 0
     article_limit = 10
@@ -374,36 +375,13 @@ def fetch_and_save_news(company_name):
         },
         "Final Sentiment Analysis": analysis_result['Final Sentiment Analysis']
     }
+    # Write output data as valid JSON
     with open(file_name, "w", encoding="utf-8") as file:
-        file.write(f'"Company": "{output_data["Company"]}",\n')
-        file.write('"Articles": [\n')
-        for i, article in enumerate(output_data["Articles"]):
-            file.write('{\n')
-            file.write(f'"Title": "{article["Title"]}",\n')
-            file.write(f'"Summary": "{article["Summary"]}",\n')
-            file.write(f'"Sentiment": "{article["Sentiment"]}",\n')
-            file.write(f'"Topics": {article["Topics"]}\n')
-            file.write('}' + (',\n' if i < len(output_data["Articles"]) - 1 else '\n'))
-        file.write('],\n')
-        file.write('"Comparative Sentiment Score": {\n')
-        file.write('"Sentiment Distribution": {\n')
-        for i, (sentiment, count) in enumerate(output_data["Comparative Sentiment Score"]["Sentiment Distribution"].items()):
-            file.write(f'"{sentiment}": {count}' + (',' if i < 2 else '') + '\n')
-        file.write('}\n')
-        file.write('},\n')
-        file.write(f'{output_data["Coverage Differences"]},\n')
-        file.write('"Topic Overlap": {\n')
-        file.write(f'"Common Topics": {output_data["Topic Overlap"]["Common Topics"]},\n')
-        for i, (key, value) in enumerate([(k, v) for k, v in output_data["Topic Overlap"].items() if k != "Common Topics"]):
-            file.write(f'"{key}": {value}' + (',\n' if i < len(output_data["Topic Overlap"]) - 2 else '\n'))
-        file.write('},\n')
-        file.write(f'"Final Sentiment Analysis": "{output_data["Final Sentiment Analysis"]}"\n')
-    print("\nOutput format:")
-    with open(file_name, "r", encoding="utf-8") as file:
-        print(file.read())
-    print("✅ File saved successfully!")
+        json.dump(output_data, file, indent=4, ensure_ascii=False)
+    print("✅ File saved successfully as JSON!")
     return file_name
 
+    
 if __name__ == "__main__":
     company_name = input("Enter company name to search for (e.g., Tesla): ")
     fetch_and_save_news(company_name)