Upload streamlit_app.py

streamlit_app.py

@@ -0,0 +1,74 @@
+import streamlit as st
+import sqlite3
+
+def search_db(query):
+    # 连接SQLite数据库
+    conn = sqlite3.connect('rarbg_db.sqlite')
+    cursor = conn.cursor()
+
+    # 拆分查询以便在标题中使用多个关键字
+    include_keywords = []
+    exclude_keywords = []
+    for keyword in query.split():
+        if keyword.startswith('-'):
+            exclude_keywords.append(keyword[1:])
+        else:
+            include_keywords.append(keyword)
+
+    # 构建查询字符串
+    query_string = "SELECT title, hash, size, dt as time FROM items WHERE "
+    query_string += " AND ".join([f"title LIKE ?" for _ in include_keywords])
+    query_string += "".join([f" AND title NOT LIKE ?" for _ in exclude_keywords])
+
+    # 将关键字添加为通配符参数
+    params = [f"%{keyword}%" for keyword in include_keywords + exclude_keywords]
+    cursor.execute(query_string, params)
+
+    results = cursor.fetchall()
+    conn.close()
+
+    return results
+
+
+
+# 创建Streamlit app
+st.title("RARBG Database Search")
+search_query = st.text_input("Enter your search keywords:")
+
+if st.button("Search"):
+    results = search_db(search_query)
+    if results:
+        # 显示结果
+        st.write("Results:")
+
+        # 创建HTML表格
+        html_table = '<table><tr><th>No.</th><th>Title with Magnet Link</th><th>Size (GB)</th><th>Time</th></tr>'
+        all_links = []
+
+        for i, (title, hash_value, size, time) in enumerate(results, start=1):
+            magnet_link = f'magnet:?xt=urn:btih:{hash_value}'
+            all_links.append(magnet_link)
+
+            # 检查 size 是否为 None
+            if size is not None:
+                size_gb = size / (1024 ** 3)
+                size_gb_str = f"{size_gb:.2f} GB"
+            else:
+                size_gb_str = "Unknown"
+
+            html_table += f'<tr><td>{i}</td><td><a href="{magnet_link}" target="_blank">{title}</a></td><td>{size_gb_str}</td><td>{time}</td></tr>'
+
+        html_table += '</table>'
+
+        # 使用markdown来渲染HTML表格
+        st.markdown(html_table, unsafe_allow_html=True)
+
+        # 创建一个包含所有链接的文本区域
+        links_str = "\n".join(all_links)
+        copy_html = f'<textarea id="copy_area" readonly style="width: 100%; height: 100px">{links_str}</textarea>'
+        st.markdown(copy_html, unsafe_allow_html=True)
+
+    else:
+        st.write("No results found.")
+
+