Update app.py

app.py

@@ -1,18 +1,6 @@
-import spaces
-from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
-import gradio as gr
-import torch
-import logging
-import sys
 import os
-from accelerate import init_empty_weights
+import logging
 from typing import List, Dict
-from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain.embeddings import HuggingFaceEmbeddings
-from langchain.vectorstores import FAISS
-from langchain.chains import RetrievalQA
-from langchain.prompts import PromptTemplate
-from langchain_community.document_loaders import PyPDFLoader
 
 # Configure logging
 logging.basicConfig(
@@ -21,15 +9,38 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
-# Get HuggingFace token from environment variable
-hf_token = os.environ.get('HUGGINGFACE_TOKEN')
-if not hf_token:
-    logger.error("HUGGINGFACE_TOKEN environment variable not set")
-    raise ValueError("Please set the HUGGINGFACE_TOKEN environment variable")
+try:
+    # Basic imports first
+    import torch
+    import gradio as gr
+    from langchain.text_splitter import RecursiveCharacterTextSplitter
+    from langchain.embeddings import HuggingFaceEmbeddings
+    from langchain.vectorstores import FAISS
+    from langchain.chains import RetrievalQA
+    from langchain.prompts import PromptTemplate
+    from langchain.llms import HuggingFacePipeline
+    from langchain_community.document_loaders import PyPDFLoader
+
+    # Now try to import transformers components one by one
+    from transformers.pipelines import pipeline
+    from transformers import AutoTokenizer, AutoModelForCausalLM
+
+    # If all imports successful, proceed with spaces import
+    import spaces
+
+except ImportError as e:
+    logger.error(f"Error importing dependencies: {str(e)}")
+    logger.error("Trying to install missing packages...")
+    os.system('pip install -q transformers torch accelerate safetensors')
+    
+    # Try imports again after installation
+    from transformers.pipelines import pipeline
+    from transformers import AutoTokenizer, AutoModelForCausalLM
+    import spaces
 
 # Constants
 MODEL_NAME = "meta-llama/Llama-2-7b-chat-hf"
-KNOWLEDGE_BASE_DIR = "."  # Changed to root directory
+KNOWLEDGE_BASE_DIR = "."
 
 class DocumentLoader:
     """Class to manage PDF document loading."""