Update app.py

app.py

@@ -1,10 +1,5 @@
 import gradio as gr
-from transformers import pipeline, AutoTokenizer, AutoModelForMaskedLM, AutoModelForSeq2SeqLM
-
-# Load DNA Analysis Model
-dna_tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t6_8M_UR50D")
-dna_model = AutoModelForMaskedLM.from_pretrained("facebook/esm2_t6_8M_UR50D")
-dna_pipeline = pipeline("fill-mask", model=dna_model, tokenizer=dna_tokenizer)
+from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 
 # Load Ethical Inquiry and Learning Support Model
 ethics_tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
@@ -13,10 +8,8 @@ ethics_pipeline = pipeline("text2text-generation", model=ethics_model, tokenizer
 
 # Query Classification
 def classify_query(query):
-    """Classify the query into DNA Analysis, Ethical Inquiry, or Learning Support."""
-    if "DNA" in query or "sequence" in query:
-        return "dna_analysis"
-    elif "ethics" in query or "privacy" in query:
+    """Classify the query into Ethical Inquiry or Learning Support."""
+    if "ethics" in query or "privacy" in query:
         return "ethical_inquiry"
     else:
         return "learning_support"
@@ -26,16 +19,7 @@ def handle_query(query):
     """Route the query to the appropriate model and generate a response."""
     task = classify_query(query)
 
-    if task == "dna_analysis":
-        try:
-            # Example DNA sequence processing: Replace X or any part of the sequence with [MASK]
-            masked_sequence = query.replace("X", "[MASK]")
-            output = dna_pipeline(masked_sequence)
-            return f"DNA Analysis Result: {output}"
-        except Exception as e:
-            return f"Error in DNA Analysis: {e}"
-
-    elif task == "ethical_inquiry":
+    if task == "ethical_inquiry":
         try:
             # Ethical guidance response
             response = ethics_pipeline(query)
@@ -61,7 +45,7 @@ interface = gr.Interface(
     inputs="text",
     outputs="text",
     title="BioSphere AI Chatbot",
-    description="A chatbot for DNA Analysis, Ethical Guidance, and Learning Support in Biotech.",
+    description="A chatbot for Ethical Guidance and Learning Support in Biotech.",
 )
 
 # Add Gemmini API Key Integration