ilit.py

import streamlit as st
import os
from langchain_community.document_loaders import PyPDFLoader
from langchain_community.vectorstores import FAISS
from langchain_community.embeddings import CohereEmbeddings
from langchain_google_genai import ChatGoogleGenerativeAI
from langchain_core.messages import HumanMessage, SystemMessage

# Set environment variables for API keys
os.environ["COHERE_API_KEY"] = "############################"
if "GOOGLE_API_KEY" not in os.environ:
    os.environ["GOOGLE_API_KEY"] = "########################"

# Load PDF document
def load_document(file_path):
    loader = PyPDFLoader(file_path)
    pages = loader.load()
    return pages

# Create FAISS index
def create_faiss_index(pages):
    faiss_index = FAISS.from_documents(pages, CohereEmbeddings())
    return faiss_index

# Streamlit app
def main():
    st.title("Dream Interpretation with Langchain")
    st.write("Enter your dream in Arabic and let Langchain interpret it!")
    # Load PDF document
    faiss_index = FAISS.load_local("faiss_index", CohereEmbeddings())
    # Get user input
    user_ask = st.text_input("ما هو حلمك؟")
    
    if st.button("Interpret Dream"):
        text_contents = []

        # Similarity search
        docs = faiss_index.similarity_search(user_ask, k=25)

        # Extract text content from each document
        for doc in docs:
            text_content = doc.page_content[:]
            text_contents.append(text_content)

        # Concatenate text contents
        concatenated_texts = [text_content + "رد بمساعده المعلومات المعطاه في تفسير الحلم ودائما انهى الرد بكلمة والله اعلم وفسرالاحداث كلها وأربط بينهم" for text_content in text_contents]

        # Initialize Google GenAI model
        model = ChatGoogleGenerativeAI(model="gemini-pro", convert_system_message_to_human=True)

        # Invoke the model
        res = model.invoke([
            SystemMessage(content=concatenated_texts),
            HumanMessage(content=user_ask)
        ])

        # Display result
        st.write(eval(f'f"""{res}"""'))

if __name__ == "__main__":
    main()