Update app.py

app.py

@@ -9,16 +9,16 @@ def text_to_srt(text):
             continue
         try:
             times, content = line.split(']', 1)
-            # Splitting times by ' -> ' to get start and end times
             start, end = times[1:].split(' -> ')
-            # Replacing '.' with ',' for milliseconds and ensuring the time format is correct
-            start_formatted = start.replace('.', ',')
-            end_formatted = end.replace('.', ',')
-            # Adding the subtitle entry to the SRT content
-            srt_content += f"{i+1}\n{start_formatted} --> {end_formatted}\n{content.strip()}\n\n"
+            # Check if the timestamp includes hours; if not, prepend "00:"
+            if start.count(":") == 1:
+                start = "00:" + start
+            if end.count(":") == 1:
+                end = "00:" + end
+            # Replace '.' with ',' in timestamps for SRT format
+            srt_content += f"{i+1}\n{start.replace('.', ',')} --> {end.replace('.', ',')}\n{content.strip()}\n\n"
         except ValueError:
-            # Skip lines that don't match the expected format
-            continue
+            continue  # Skip lines that don't match the expected format
     # Save SRT content to a temporary file
     temp_file_path = '/tmp/output.srt'
     with open(temp_file_path, 'w', encoding='utf-8') as file:
@@ -26,9 +26,10 @@ def text_to_srt(text):
     return temp_file_path
 
 with gr.Blocks() as app:
-    gr.Markdown("# Text to SRT Converter")
+    gr.Markdown("### Text to SRT Converter")
     text_input = gr.TextArea(label="Enter text")
-    output = gr.File(label="Download SRT File", file_count="single")  # Change output to a File component
+    output = gr.File(label="Download SRT File", file_count="single")
     text_input.change(fn=text_to_srt, inputs=text_input, outputs=output)
 
-app.launch(share=True)
+if __name__ == "__main__":
+    app.launch()