Create app.py

app.py

@@ -0,0 +1,55 @@
+import openai
+import gradio as gr
+import pandas as pd
+import os  # Importing os to access environment variables
+from datasets import load_dataset
+
+# Load the dataset from Hugging Face
+dataset = load_dataset('dindizz/musicacademyarchives')
+
+# Access the OpenAI API key from environment variables (Hugging Face secret)
+openai.api_key = os.getenv('OPENAI_API_KEY')
+
+def extract_info(query):
+    """
+    This function interacts with OpenAI GPT-3.5 Turbo to extract information from the dataset based on the user's query.
+    """
+    # Extracting the text content from the dataset to pass as context
+    all_souvenirs = []
+    for item in dataset['train']:
+        souvenir_text = item['text']  # Assuming the column name is 'text' containing the content
+        all_souvenirs.append(souvenir_text)
+    
+    # Combine the content into a single string (you can adjust based on the size of the dataset)
+    combined_souvenir_text = "\n".join(all_souvenirs)
+
+    # Prompt OpenAI GPT-3.5 with the user's query and the combined text
+    prompt = f"Extract relevant information based on the following query: '{query}' from the Madras Music Academy Souvenir archives: {combined_souvenir_text[:2000]}"  # limiting the length for performance
+    
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",  # Updated model
+        messages=[
+            {"role": "system", "content": "You are an assistant that extracts information from the Madras Music Academy Souvenir dataset and present in a friendly tone ."},
+            {"role": "user", "content": prompt}
+        ],
+        max_tokens=300
+    )
+    
+    # Returning the answer from OpenAI GPT-3.5 Turbo
+    answer = response['choices'][0]['message']['content']
+    return answer.strip()
+
+# Define the Gradio interface
+def gradio_interface(query):
+    return extract_info(query)
+
+# Launch the Gradio app
+iface = gr.Interface(
+    fn=gradio_interface, 
+    inputs="text", 
+    outputs="text",
+    title="Sabha Scholar - Madras Music Academy AI Explorer",
+    description="Ask questions about the Madras Music Academy Souvenirs and extract information using OpenAI GPT-3.5 Turbo."
+)
+
+iface.launch()