app.py

import streamlit as st
from ai_helper import AIHelper

ai = AIHelper()

def main():
    st.title("AI Prompt App")

    # Get user input
    user_prompt = st.text_area("Enter your prompt here:")

    # Create a button
    if st.button("Generate AI Response"):
        if user_prompt:
            try:
                # Generate a response from the AI model
                ai.generate_ai_response(user_prompt)
                
            except Exception as e:
                st.error("An error occurred: " + str(e))
        else:
            st.warning("Please enter a prompt.")

if __name__ == "__main__":
    main()