Update app.py

app.py

@@ -431,6 +431,63 @@ with st.sidebar:
                 st.error(f"❌ Generation failed: {str(e)}")
                 st.write("💡 Try adjusting your parameters or check your API key")
     
+    # Data Management
+    st.markdown("---")
+    st.subheader("📁 Data Management")
+    
+    # Import functionality
+    with st.expander("📥 Import Scripts from JSONL", expanded=False):
+        st.info("💡 Upload a JSONL file to import scripts as reference data for the RAG system")
+        
+        uploaded_file = st.file_uploader(
+            "Choose a JSONL file", 
+            type=['jsonl'],
+            help="Upload your script data file (e.g., anya_scripts.jsonl)"
+        )
+        
+        if uploaded_file is not None:
+            # Save uploaded file temporarily
+            import tempfile
+            with tempfile.NamedTemporaryFile(delete=False, suffix='.jsonl') as tmp_file:
+                tmp_file.write(uploaded_file.getvalue())
+                tmp_path = tmp_file.name
+            
+            col1, col2 = st.columns(2)
+            with col1:
+                if st.button("📥 Import Scripts", type="primary", use_container_width=True):
+                    with st.spinner("Importing scripts..."):
+                        try:
+                            from db import import_jsonl
+                            count = import_jsonl(tmp_path)
+                            st.success(f"✅ Successfully imported {count} scripts!")
+                            st.balloons()
+                            time.sleep(1)
+                            st.rerun()
+                        except Exception as e:
+                            st.error(f"❌ Import failed: {str(e)}")
+                        finally:
+                            # Clean up temp file
+                            import os
+                            try:
+                                os.unlink(tmp_path)
+                            except:
+                                pass
+            
+            with col2:
+                if st.button("📊 Preview Data", use_container_width=True):
+                    try:
+                        import json
+                        # Show first few lines as preview
+                        content = uploaded_file.getvalue().decode('utf-8')
+                        lines = content.strip().split('\n')[:3]
+                        st.write("**Preview (first 3 scripts):**")
+                        for i, line in enumerate(lines, 1):
+                            if line.strip():
+                                data = json.loads(line)
+                                st.write(f"**Script {i}:** {data.get('theme', 'No title')} - {data.get('model_name', 'Unknown creator')}")
+                    except Exception as e:
+                        st.error(f"Preview failed: {str(e)}")
+    
     # Quick Actions
     st.markdown("---")
     st.subheader("⚡ Quick Actions")