cp

app.py

@@ -1,25 +1,42 @@
-import streamlit as st
-from transformers import pipeline
+# import streamlit as st
+# from transformers import pipeline
 
-# Load the SQLCoder model
-sql_generator = pipeline('text-generation', model='defog/sqlcoder')
+# # Load the SQLCoder model
+# sql_generator = pipeline('text-generation', model='defog/sqlcoder')
 
-st.title('SQL Table Extractor')
+# st.title('SQL Table Extractor')
 
-# Text input for SQL query
-user_sql = st.text_input("Enter your SQL statement", "SELECT * FROM my_table WHERE condition;")
+# # Text input for SQL query
+# user_sql = st.text_input("Enter your SQL statement", "SELECT * FROM my_table WHERE condition;")
 
-# Button to parse SQL
-if st.button('Extract Tables'):
-    # Generate SQL or parse directly
-    results = sql_generator(user_sql)
-    # Assuming results contain SQL, extract table names (this part may require custom logic based on output)
-    tables = extract_tables_from_sql(results)
+# # Button to parse SQL
+# if st.button('Extract Tables'):
+#     # Generate SQL or parse directly
+#     results = sql_generator(user_sql)
+#     # Assuming results contain SQL, extract table names (this part may require custom logic based on output)
+#     tables = extract_tables_from_sql(results)
     
-    # Display extracted table names
-    st.write('Extracted Tables:', tables)
+#     # Display extracted table names
+#     st.write('Extracted Tables:', tables)
+
+# def extract_tables_from_sql(sql):
+#     # Dummy function: Implement logic to parse table names from SQL
+#     return ["my_table"]  # Example output
+
+import streamlit as st
+from transformers import pipeline
+
+# Load the NER model
+ner = pipeline("ner", model="dbmdz/bert-large-cased-finetuned-conll03-english", grouped_entities=True)
+
+st.title('Hello World NER Parser')
 
-def extract_tables_from_sql(sql):
-    # Dummy function: Implement logic to parse table names from SQL
-    return ["my_table"]  # Example output
+# User input for text
+user_input = st.text_area("Enter a sentence to parse for named entities:", "John Smith lives in San Francisco.")
 
+# Parse entities
+if st.button('Parse'):
+    entities = ner(user_input)
+    # Display extracted entities
+    for entity in entities:
+        st.write(f"Entity: {entity['word']}, Entity Type: {entity['entity_group']}")