Update app.py

app.py

@@ -76,16 +76,23 @@ def parse_fasta(content: str) -> list:
         logger.error(f"Error parsing FASTA: {str(e)}")
         raise
 
-def predict_sequence(file_obj) -> str:
+def predict_sequence(fasta_file) -> str:
     """Process FASTA input and return formatted predictions"""
     try:
         logger.info("Starting prediction process")
         
-        if file_obj is None:
+        if fasta_file is None:
             return "Please upload a FASTA file"
             
-        # Read the file content
-        content = file_obj.read().decode('utf-8')
+        # Get file content - handle both string and file inputs
+        try:
+            if isinstance(fasta_file, str):
+                content = fasta_file
+            else:
+                content = fasta_file.name  # For Gradio file upload
+        except Exception as e:
+            logger.error(f"Error reading file: {str(e)}")
+            return f"Error reading file: {str(e)}"
         
         device = 'cuda' if torch.cuda.is_available() else 'cpu'
         logger.info(f"Using device: {device}")
@@ -147,14 +154,22 @@ Non-human probability: {float(probs[0][0]):0.4f}
         logger.error(f"Unexpected error in predict_sequence: {str(e)}")
         return f"An unexpected error occurred: {str(e)}"
 
-# Create Gradio interface
+# Create Gradio interface with both file upload and text input
 iface = gr.Interface(
     fn=predict_sequence,
-    inputs=gr.File(label="Upload FASTA file", file_types=[".fasta", ".fa", ".txt"]),
+    inputs=[
+        gr.File(label="Upload FASTA file", file_types=[".fasta", ".fa", ".txt"]),
+        # Adding a text input as an alternative
+        gr.Textbox(label="Or paste FASTA sequence here", lines=10)
+    ],
     outputs=gr.Textbox(label="Prediction Results", lines=10),
     title="Virus Host Classifier",
-    description="Upload a FASTA file to predict whether a virus sequence is likely to infect human or non-human hosts.",
-    examples=[["example.fasta"]],
+    description="""Upload a FASTA file or paste your sequence to predict whether a virus sequence is likely to infect human or non-human hosts.
+    
+Example format:
+>sequence_name
+ATCGATCGATCG...""",
+    examples=[["example.fasta", None]],
     cache_examples=True
 )