LegendaryToe/SqlParser

1# import streamlit as st2# from transformers import pipeline3 4# # Load the SQLCoder model5# sql_generator = pipeline('text-generation', model='defog/sqlcoder')6 7# st.title('SQL Table Extractor')8 9# # Text input for SQL query10# user_sql = st.text_input("Enter your SQL statement", "SELECT * FROM my_table WHERE condition;")11 12# # Button to parse SQL13# if st.button('Extract Tables'):14#     # Generate SQL or parse directly15#     results = sql_generator(user_sql)16#     # Assuming results contain SQL, extract table names (this part may require custom logic based on output)17#     tables = extract_tables_from_sql(results)18    19#     # Display extracted table names20#     st.write('Extracted Tables:', tables)21 22# def extract_tables_from_sql(sql):23#     # Dummy function: Implement logic to parse table names from SQL24#     return ["my_table"]  # Example output25 26# import streamlit as st27# from transformers import pipeline28 29# # Load the NER model30# ner = pipeline("ner", model="dbmdz/bert-large-cased-finetuned-conll03-english", grouped_entities=True)31 32# st.title('Hello World NER Parser')33 34# # User input for text35# user_input = st.text_area("Enter a sentence to parse for named entities:", "John Smith lives in San Francisco.")36 37# # Parse entities38# if st.button('Parse'):39#     entities = ner(user_input)40#     # Display extracted entities41#     for entity in entities:42#         st.write(f"Entity: {entity['word']}, Entity Type: {entity['entity_group']}")43 44import streamlit as st45from transformers import pipeline46 47# Load a smaller LLaMA model with permission to run custom code48text_generator = pipeline("text-generation", model="microsoft/Phi-3-mini-128k-instruct", trust_remote_code=True)49 50st.title('General Query Answerer')51 52# User input for a general question53user_query = st.text_area("Enter your question:", "Name all 50 US states.")54 55# Generate answer56if st.button('Answer Question'):57    answer = text_generator(user_query, max_length=150)[0]['generated_text']58    # Display the answer59    st.write('Answer:', answer)60 61 62 63 64