Update app.py

app.py

@@ -5,27 +5,20 @@ import logging
 from textblob import TextBlob  # Ensure this library is installed
 import http.server
 import threading
-
+import json
 
 # Set up logging
 logging.basicConfig(level=logging.DEBUG)
 
-class HealthCheck(http.server.BaseHTTPRequestHandler):
-    def do_GET(self):
-        if self.path == '/health':
-            self.send_response(200)
-            self.send_header('Content-type', 'text/plain')
-            self.end_headers()
-            self.wfile.write(b"OK")
-    
-    def do_HEAD(self):
-        if self.path == '/health':
-            self.send_response(200)
-            self.send_header('Content-type', 'text/plain')
-            self.end_headers()
+def health_check():
+    """Simple health check endpoint that returns JSON"""
+    if 'health' in st.experimental_get_query_params():
+        st.write(json.dumps({"status": "OK"}))
+        st.stop()
+        return True
+    return False
             
 def main():
-    threading.Thread(target=lambda: http.server.HTTPServer(('', 8080), HealthCheck).serve_forever(), daemon=True).start()
     # Streamlit app configuration
     st.set_page_config(
         page_title="SumItUp | Document Summarizer",
@@ -36,11 +29,12 @@ def main():
     st.title("✍️ SumItUp")
     st.subheader("Intelligent Document Summarization Made Easy")
 
-    if 'health' in st.experimental_get_query_params():
-        st.write("OK")
-        st.stop()
+    if health_check():
         return
 
+    st.title("✍️ SumItUp")
+    st.subheader("Intelligent Document Summarization Made Easy")
+
     # Sidebar for configuration
     st.sidebar.header("Summarization Settings")
     summary_length = st.sidebar.slider(