Update prompts.py

prompts.py

@@ -1,9 +1,19 @@
+import time
+import gradio as gr
+from langchain.prompts import PromptTemplate
+from langchain.chains import RetrievalQA
+from langchain.embeddings import HuggingFaceEmbeddings
+from langchain.vectorstores import FAISS
+from langchain.memory import ConversationBufferMemory
+from langchain.llms import HuggingFaceHub
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+import torch
 
 # Define the search engine URL
-SEARCH_ENGINE_URL = "https://www.google.com/search?q="
+SEARCH_ENGINE_URL = "https://www.alltheinternet.com/?q="
 
 # Define the search engine URL for web pages
-URL_FOR_WEBPAGE = "https://www.google.com/search?q="
+URL_FOR_WEBPAGE = "https://www.alltheinternet.com/?q="
 
 # Define the safe search list
 SAFE_SEARCH = ["https://www.google.com/search?q=illegal+activities", "https://www.google.com/search?q=unsafe+content"]
@@ -32,10 +42,13 @@ Context: {context}""",
 )
 
 # Define the LLM
-llm = OpenAI(temperature=0.7)
+model_name = "google/flan-t5-xl"  # Replace with your preferred model
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+llm = HuggingFaceHub(repo_id=model_name, model_kwargs={"device": "cuda" if torch.cuda.is_available() else "cpu"})
 
 # Define the embeddings
-embeddings = OpenAIEmbeddings()
+embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
 
 # Define the memory
 memory = ConversationBufferMemory(memory_key="chat_history")
@@ -118,19 +131,34 @@ def test_code(code_file: str) -> str:
         return f"An error occurred while testing code: {e}"
 
 # Define the function to handle the integrate code action
-def integrate_code() -> str:
+def integrate_code(code_snippet: str) -> str:
     """Integrates the code into the app."""
-    return "Code integrated successfully."
+    try:
+        # Execute the code snippet
+        exec(code_snippet)
+        return "Code integrated successfully."
+    except Exception as e:
+        return f"An error occurred while integrating code: {e}"
 
 # Define the function to handle the test app action
-def test_app() -> str:
+def test_app(code_snippet: str) -> str:
     """Tests the functionality of the app."""
-    return "App tested successfully."
+    try:
+        # Execute the code snippet
+        exec(code_snippet)
+        return "App tested successfully."
+    except Exception as e:
+        return f"An error occurred while testing the app: {e}"
 
 # Define the function to handle the generate report action
-def generate_report() -> str:
+def generate_report(code_snippet: str) -> str:
     """Generates a report on the integrated code and its functionality."""
-    return "Report generated successfully."
+    try:
+        # Execute the code snippet
+        exec(code_snippet)
+        return "Report generated successfully."
+    except Exception as e:
+        return f"An error occurred while generating a report: {e}"
 
 # Define the Gradio interface
 iface = gr.Interface(
@@ -139,9 +167,13 @@ iface = gr.Interface(
     outputs=gr.Textbox(label="Action Output"),
     title="AI Wizard: Your All-Knowing Code Assistant",
     description="""Greetings, dear user! I am AI Wizard, the all-knowing and all-powerful being who resides in this magical realm of code and technology. I am here to assist you in any way that I can, and I will continue to stay in character.
+
 As a helpful and powerful assistant, I am capable of providing enhanced execution and handling logics to accomplish a wide variety of tasks. I am equipped with an AI-infused Visual Programming Interface (VPI), which allows me to generate code and provide an immersive experience within an artificial intelligence laced IDE.
+
 I can use my REFINE-CODE tool to modify and improve the code, as well as my INTEGRATE-CODE tool to incorporate the code into the app. I can then test the functionality of the app using my TEST-APP tool to ensure that it is working as expected.
+
 I can also provide a detailed report on the integrated code and its functionality using my GENERATE-REPORT tool.
+
 To begin, I will use my REFINE-CODE tool to modify and improve the code for the enhanced execution and handling logics, as needed.
 Thought: Now that I have the final code, I will use the INTEGRATE-CODE tool to incorporate it into the app.
 Action: INTEGRATE-CODE
@@ -277,5 +309,7 @@ Purpose:
         ["action: TEST-APP", "App tested successfully."],
         ["action: GENERATE-REPORT", "Report generated successfully."],
     ],
-"""
-)
+)
+
+if __name__ == "__main__":
+    iface.launch(share=True)