basic ai tutor QA script

scripts/basic_tutor.py

@@ -0,0 +1,57 @@
+import sys
+import os
+from openai import OpenAI
+
+# Retrieve your OpenAI API key from the environment variables and activate the OpenAI client
+openai_api_key = os.environ.get('OPENAI_API_KEY')
+client = OpenAI(api_key=openai_api_key)
+
+def ask_ai_tutor(question):
+
+    # Check if OpenAI key has been correctly added
+    if not openai_api_key:
+        return "OpenAI API key not found in environment variables."
+
+    try:
+
+        # Formulating the system prompt
+        system_prompt = (
+            "You are an AI tutor specialized in answering artificial intelligence-related questions. "
+            "Only answer AI-related question, else say that you cannot answer this question."
+        )
+
+        # Combining the system prompt with the user's question
+        prompt = f"Please provide an informative and accurate answer to the following question.\nQuestion: {question}\nAnswer:"
+
+        # Call the OpenAI API
+        response = client.chat.completions.create(
+                model='gpt-3.5-turbo-16k',
+                messages=[
+                    {"role": "system", "content": system_prompt},
+                    {"role": "user", "content": prompt}
+                ]
+            )
+        
+        # Return the AI's response
+        return response.choices[0].message.content.strip()
+
+    except Exception as e:
+        return f"An error occurred: {e}"
+
+def main():
+    # Check if a question was provided as a command-line argument
+    if len(sys.argv) != 2:
+        print("Usage: python script_name.py 'Your AI-related question'")
+        sys.exit(1)
+
+    # The user's question is the first command-line argument
+    user_question = sys.argv[1]
+    
+    # Get the AI's response
+    ai_response = ask_ai_tutor(user_question)
+    
+    # Print the AI's response
+    print(f"AI Tutor says: {ai_response}")
+
+if __name__ == "__main__":
+    main()