Upload 4 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+faiss_index_sysml/index.faiss filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,71 @@
+import os
+import warnings
+import gradio as gr
+from dotenv import load_dotenv
+
+from langchain.chains import ConversationalRetrievalChain
+from langchain_community.vectorstores import FAISS
+from langchain_openai import ChatOpenAI
+from langchain_community.embeddings import OpenAIEmbeddings
+
+# Patch Gradio bug (schema parsing issue)
+import gradio_client.utils
+gradio_client.utils.json_schema_to_python_type = lambda schema, defs=None: "string"
+
+
+
+# Load environment variables
+load_dotenv()
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY_P") #or os.getenv("OPENAI_API_KEY_ROBO") 
+if not OPENAI_API_KEY:
+    raise ValueError("Missing OPENAI_API_KEY. Please set it in your environment variables.")
+
+
+# Suppress warnings
+warnings.filterwarnings("ignore")
+
+# Initialize embedding model
+embeddings = OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY)
+
+# Load FAISS vector store
+vectorstore = FAISS.load_local(
+    "faiss_index_sysml", embeddings, allow_dangerous_deserialization=True
+)
+
+# Load ChatOpenAI model
+llm = ChatOpenAI(
+    model_name="gpt-4", 
+    temperature=0.5,
+    openai_api_key=OPENAI_API_KEY
+)
+
+# Build conversational chain with history
+qa = ConversationalRetrievalChain.from_llm(
+    llm=llm,
+    retriever=vectorstore.as_retriever(),
+    return_source_documents=False
+)
+
+history = []
+
+# Chatbot logic
+def sysml_chatbot(message, history):
+    result = qa({"question": message, "chat_history": history})
+    answer = result["answer"]
+    history.append((message, answer))
+    return "", history
+
+# Gradio UI
+with gr.Blocks() as demo:
+    gr.Markdown("## SysML Chatbot")
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(placeholder="Ask me about SysML diagrams or concepts...")
+    clear = gr.Button("Clear")
+
+    state = gr.State(history)
+
+    msg.submit(sysml_chatbot, [msg, state], [msg, chatbot])
+    clear.click(lambda: ([], ""), None, [chatbot, msg])
+
+if __name__ == "__main__":
+    demo.launch()

faiss_index_sysml/index.faiss

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8e6cf999585beebe259f8ffc3fae13d8015eb193459dcf610bf710b95f009c6c
+size 2291757

faiss_index_sysml/index.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e034f5a0c34cbf3d0ab09b6411579a47e9dd5d9ac371a0af1ac6e07c1b0309a7
+size 761219

requirements.txt

@@ -0,0 +1,12 @@
+langchain
+langchain-community
+langchain-core
+langchain-openai
+openai
+faiss-cpu
+python-dotenv
+transformers
+sentence-transformers
+gradio==4.15.0
+gradio_client==0.8.1
+huggingface_hub >= 0.19.3