Update app.py

app.py

@@ -1,135 +1,99 @@
+"""
+主程序入口点 - 用于 Hugging Face Spaces
+"""
 import os
 import sys
-import subprocess
-import pkg_resources
-import importlib.metadata
+import gradio as gr
 
-def install_package(package, options=None):
-    """安装指定的包并返回安装结果"""
-    cmd = [sys.executable, "-m", "pip", "install"]
-    if options:
-        cmd.extend(options)
-    cmd.append(package)
+# 确保能找到 install_pysrt 包
+current_dir = os.path.dirname(os.path.abspath(__file__))
+if current_dir not in sys.path:
+    sys.path.append(current_dir)
+
+# 尝试导入并安装 pysrt
+try:
+    import install_pysrt
+    installed = install_pysrt.is_pysrt_installed()
+    version = install_pysrt.get_pysrt_version() if installed else "未安装"
     
-    print(f"尝试安装 {package} 使用命令: {' '.join(cmd)}")
-    try:
-        subprocess.check_call(cmd)
-        return True
-    except subprocess.CalledProcessError as e:
-        print(f"安装失败: {e}")
-        return False
+    if not installed:
+        print("尝试安装 pysrt...")
+        success = install_pysrt.installer.install_pysrt()
+        if success:
+            version = install_pysrt.get_pysrt_version()
+            print(f"pysrt {version} 已成功安装")
+        else:
+            print("安装 pysrt 失败")
+    else:
+        print(f"pysrt {version} 已安装")
+        
+    import_success = True
+except Exception as e:
+    print(f"导入或安装 install_pysrt 时出错: {e}")
+    import_success = False
+    version = "错误"
 
-def get_package_version(package_name):
-    """获取已安装包的版本"""
-    try:
-        # 使用 importlib.metadata 获取版本 (Python 3.8+)
-        return importlib.metadata.version(package_name)
-    except importlib.metadata.PackageNotFoundError:
-        try:
-            # 备选方法
-            return pkg_resources.get_distribution(package_name).version
-        except pkg_resources.DistributionNotFound:
-            return None
 
-def main():
-    """主函数：尝试多种方法安装 pysrt"""
-    print("="*50)
-    print("开始诊断 HF 环境中的 pip 安装问题 (修改版)")
-    print("="*50)
-    
-    # 显示 Python 版本
-    print(f"Python 版本: {sys.version}")
-    
-    # 显示 pip 版本
+def create_subtitle(start_time, end_time, text):
+    """创建字幕并返回结果"""
     try:
-        pip_version = subprocess.check_output([sys.executable, "-m", "pip", "--version"]).decode().strip()
-        print(f"Pip 版本: {pip_version}")
-    except:
-        print("无法获取 pip 版本")
+        import pysrt
+        from pysrt import SubRipFile, SubRipItem, SubRipTime
+        
+        # 解析时间
+        start_parts = [int(x) for x in start_time.split(':')]
+        end_parts = [int(x) for x in end_time.split(':')]
+        
+        if len(start_parts) == 3:
+            start = SubRipTime(hours=start_parts[0], minutes=start_parts[1], seconds=start_parts[2])
+        else:
+            start = SubRipTime(minutes=start_parts[0], seconds=start_parts[1])
+            
+        if len(end_parts) == 3:
+            end = SubRipTime(hours=end_parts[0], minutes=end_parts[1], seconds=end_parts[2])
+        else:
+            end = SubRipTime(minutes=end_parts[0], seconds=end_parts[1])
+        
+        # 创建字幕文件
+        subs = SubRipFile()
+        item = SubRipItem(index=1, start=start, end=end, text=text)
+        subs.append(item)
+        
+        # 转换为字符串
+        return str(subs), f"字幕已成功创建! pysrt 版本: {install_pysrt.get_pysrt_version()}"
+    except Exception as e:
+        return "", f"创建字幕时出错: {e}"
+
+# 创建 UI
+with gr.Blocks(title="pysrt 演示") as demo:
+    gr.Markdown(f"""
+    # pysrt 字幕演示程序
     
-    # 安装最新版本的 setuptools
-    print("尝试更新 setuptools...")
-    install_package("setuptools", ["--upgrade"])
+    当前 pysrt 状态: {"已安装 ✅" if import_success else "未安装 ❌"}
     
-    # 方法 1: 使用 --use-pep517
-    print("\n方法 1: 使用 --use-pep517 安装")
-    if install_package("pysrt", ["--use-pep517"]):
-        print("成功使用 --use-pep517 安装 pysrt!")
-    else:
-        # 方法 2: 使用 egg 安装
-        print("\n方法 2: 尝试使用 egg 安装")
-        if install_package("pysrt", ["--egg"]):
-            print("成功使用 egg 安装 pysrt!")
-        else:
-            # 方法 3: 直接从 PyPI 获取源代码并安装
-            print("\n方法 3: 直接从源代码安装")
-            try:
-                import tempfile
-                import shutil
-                from urllib.request import urlretrieve
-                import tarfile
-                
-                # 创建临时目录
-                temp_dir = tempfile.mkdtemp()
-                try:
-                    # 下载源代码
-                    tar_path = os.path.join(temp_dir, "pysrt.tar.gz")
-                    urlretrieve("https://files.pythonhosted.org/packages/86/97/5ec8ae1f3e93414fb9c143c3b55a6b7c67caef9c89052c45b3222f2c6a36/pysrt-1.1.2.tar.gz", tar_path)
-                    
-                    # 解压
-                    with tarfile.open(tar_path) as tar:
-                        tar.extractall(temp_dir)
-                    
-                    # 找到 setup.py 所在的目录
-                    setup_dir = os.path.join(temp_dir, "pysrt-1.1.2")
-                    
-                    # 直接运行 setup.py install
-                    subprocess.check_call([sys.executable, "setup.py", "install"], cwd=setup_dir)
-                    print("成功从源代码安装 pysrt!")
-                    
-                except Exception as e:
-                    print(f"从源代码安装失败: {e}")
-                finally:
-                    # 清理临时目录
-                    shutil.rmtree(temp_dir)
-            except Exception as e:
-                print(f"方法 3 失败: {e}")
-                
-                # 方法 4: 使用 pip install -e
-                print("\n方法 4: 尝试使用 pip install -e")
-                if install_package("git+https://github.com/byroot/pysrt.git", ["-e"]):
-                    print("成功使用 -e 从 GitHub 安装 pysrt!")
+    版本: {version}
+    """)
     
-    # 检查是否真的安装成功并获取版本
-    version = get_package_version("pysrt")
-    if version:
-        print(f"\npysrt 版本: {version}")
+    with gr.Row():
+        with gr.Column():
+            start_time = gr.Textbox(label="开始时间 (分:秒 或 时:分:秒)", value="00:00")
+            end_time = gr.Textbox(label="结束时间 (分:秒 或 时:分:秒)", value="00:10")
+            text = gr.Textbox(label="字幕文本", value="这是一个测试字幕")
+            create_btn = gr.Button("创建字幕")
         
-        # 验证是否可以导入
-        try:
-            import pysrt
-            print("成功导入 pysrt 模块")
-            
-            # 创建一个简单的字幕对象以验证功能
-            try:
-                from pysrt import SubRipFile, SubRipItem, SubRipTime
-                subs = SubRipFile()
-                item = SubRipItem(index=1, 
-                                  start=SubRipTime(0, 0, 0), 
-                                  end=SubRipTime(0, 0, 10), 
-                                  text="测试字幕")
-                subs.append(item)
-                print("成功创建字幕对象，pysrt 功能正常")
-            except Exception as e:
-                print(f"创建字幕对象失败: {e}")
-        except ImportError as e:
-            print(f"导入 pysrt 失败: {e}")
-    else:
-        print("\n无法获取 pysrt 版本，安装可能失败")
+        with gr.Column():
+            result = gr.Textbox(label="生成的 SRT 内容", lines=10)
+            status = gr.Textbox(label="状态")
     
-    print("="*50)
-    print("诊断完成")
-    print("="*50)
+    create_btn.click(fn=create_subtitle, inputs=[start_time, end_time, text], outputs=[result, status])
 
 if __name__ == "__main__":
-    main()
+    # 显示 pysrt 安装状态
+    print("-" * 50)
+    print(f"pysrt 安装状态: {'成功' if import_success else '失败'}")
+    if import_success:
+        print(f"pysrt 版本: {version}")
+    print("-" * 50)
+    
+    # 启动 UI
+    demo.launch()