Update app.py

app.py

@@ -263,6 +263,7 @@ def handle_conversation(chat_id, text):
         else:
             user_inputs[current_step] = text.strip()
             logging.info(f"Received {current_step}: {text.strip()}")
+
         idx = conversation_fields.index(current_step)
         if idx < len(conversation_fields) - 1:
             current_step = conversation_fields[idx + 1]
@@ -298,14 +299,17 @@ def stream_to_youtube(input_url, quality_settings, video_codec, audio_codec, out
     try:
         streaming_state = "streaming"
         reset_statistics()
+
         input_stream = av.open(input_url)
         output_stream = av.open(output_url, mode='w', format='flv')
+
         # Configure video stream
         video_stream = output_stream.add_stream(video_codec, rate=30)
         video_stream.width = input_stream.streams.video[0].width
         video_stream.height = input_stream.streams.video[0].height
         video_stream.pix_fmt = input_stream.streams.video[0].format.name
         video_stream.codec_context.options.update({'g': '30'})
+
         if quality_settings.lower() == "high":
             video_stream.bit_rate = 3000000
             video_stream.bit_rate_tolerance = 1000000
@@ -315,12 +319,16 @@ def stream_to_youtube(input_url, quality_settings, video_codec, audio_codec, out
         elif quality_settings.lower() == "low":
             video_stream.bit_rate = 800000
             video_stream.bit_rate_tolerance = 200000
+
         # Configure audio stream
         audio_stream_in = input_stream.streams.audio[0]
         out_audio_stream = output_stream.add_stream(audio_codec, rate=audio_stream_in.rate)
         out_audio_stream.layout = "stereo"
+
         video_stream.codec_context.time_base = fractions.Fraction(1, video_stream.rate)
+
         logging.info("Streaming started successfully.")
+
         # Start the live log updater in a background thread if not already running.
         global live_log_thread
         if live_log_thread is None or not live_log_thread.is_alive():
@@ -328,6 +336,7 @@ def stream_to_youtube(input_url, quality_settings, video_codec, audio_codec, out
             live_log_thread.daemon = True
             live_log_thread.start()
             logging.info("Live log updater thread started.")
+
         # Stream loop: process packets until state changes
         while streaming_state in ["streaming", "paused"]:
             for packet in input_stream.demux():
@@ -352,12 +361,17 @@ def stream_to_youtube(input_url, quality_settings, video_codec, audio_codec, out
                             output_stream.mux(out_packet)
                             if hasattr(out_packet, "size"):
                                 bytes_sent += out_packet.size
+
+            # Flush remaining packets
             for out_packet in video_stream.encode():
                 output_stream.mux(out_packet)
             for out_packet in out_audio_stream.encode():
                 output_stream.mux(out_packet)
+
             if streaming_state == "paused":
                 time.sleep(1)
+
+        # Clean up resources
         try:
             video_stream.close()
             out_audio_stream.close()
@@ -365,6 +379,7 @@ def stream_to_youtube(input_url, quality_settings, video_codec, audio_codec, out
             input_stream.close()
         except Exception as cleanup_error:
             logging.error(f"Error during cleanup: {cleanup_error}")
+
         logging.info("Streaming complete, resources cleaned up.")
         streaming_state = "idle"
     except Exception as e:
@@ -378,6 +393,7 @@ def start_streaming(chat_id):
     valid, msg = validate_inputs()
     if not valid:
         return send_guide_message(chat_id, f"Validation error: {msg}")
+
     stream_chat_id = chat_id
     try:
         stream_thread = threading.Thread(
@@ -472,9 +488,11 @@ async def telegram_webhook(request: Request):
     update = await request.json()
     logging.debug(f"Received update: {update}")
 
+    # Process messages from users
     if "message" in update:
         chat_id = update["message"]["chat"]["id"]
         text = update["message"].get("text", "").strip()
+
         if text.startswith("/setting"):
             return handle_setting(chat_id)
         elif text.startswith("/start"):
@@ -493,10 +511,13 @@ async def telegram_webhook(request: Request):
             return stream_status(chat_id)
         else:
             return handle_conversation(chat_id, text)
+
+    # Process inline keyboard callback queries
     elif "callback_query" in update:
         callback_data = update["callback_query"]["data"]
         chat_id = update["callback_query"]["message"]["chat"]["id"]
         message_id = update["callback_query"]["message"]["message_id"]
+
         if callback_data == "pause":
             response = pause_stream(chat_id)
         elif callback_data == "resume":
@@ -509,12 +530,16 @@ async def telegram_webhook(request: Request):
             response = start_streaming(chat_id)
         else:
             response = send_guide_message(chat_id, "❓ Unknown callback command.")
+
+        # For streaming control callbacks, update the message with the latest live logs
         if callback_data in ["pause", "resume", "abort", "status", "start_stream"]:
             response["method"] = "editMessageText"
             response["message_id"] = message_id
             response["text"] = compose_streaming_message()
             response["parse_mode"] = "HTML"
+            # Always include the inline keyboard if streaming is active.
             if streaming_state in ["streaming", "paused"]:
                 response["reply_markup"] = get_inline_keyboard_for_stream()
         return response
+
     return {"status": "ok"}